#!/usr/bin/perl -w

use strict;
use Benchmark;
use DBI;
use MLDBM qw(DB_File Storable);
use Storable;
use Fcntl;

my($i_pub, $i_size, $i_type, $i_ip);
my(%s_crr, $rs_tmp, %s_tmp);
my(%s_disp_hash, $r_hash);
my($r_flatfile);
my($dbh, $r_postgres, @row_ary);
my($dbh2, $r_mysql, @row_ary2);
my($r_storable, %s_disp_hash_storable);
my($r_mldbm, %s_disp_hash_mldbm);

my($i_count) = 20;
my($i_inner_loop) = 2;
my($s_delimiter) = '|';

%s_crr = ("0", 0,
		"1", 1,
		"2", 2,
		"3", 3,
		"4", 4,
		"5", 5,
		"6", 6,
		"7", 7,
		"8", 8,
		"9", 9 );

#--------------------------------------------------------------------------

$dbh = DBI->connect("dbi:Pg:dbname=benchmark", "drfrog",
		"", { RaiseError => 1, AutoCommit => 1 });	# Postgres
$dbh2 = DBI->connect("dbi:mysql:dbname=benchmark", "drfrog",
		"trans22", { RaiseError => 1, AutoCommit => 1 });	# mysql

foreach $i_pub (0..99) {

#	mkdir "flatfile/pub$i_pub", 0777;	# flat file	
	
	foreach $i_size (0..2) {
	
#		mkdir "flatfile/pub$i_pub/size$i_size", 0777;	# flat file
	
		foreach $i_type (0..3) {
		
#			mkdir "flatfile/pub$i_pub/size$i_size/type$i_type", 0777;

			foreach $i_ip (0..19) {

				#----- flat file -----

#				open(FH, ">flatfile/pub$i_pub/size$i_size/type$i_type/ip$i_ip");

#				print FH qq("0",0,"1",1,"2",2,"3",3,"4",4,"5",5,"6",6,"7",7,"8",8,"9",9);

#				close(FH);
			
				#----- hash -----
				
#				$s_disp_hash{join($s_delimiter,
#						"pub$i_pub","size$i_size","type$i_type","ip$i_ip")} =
#						\%s_crr;	

				#----- postgres -----
		
$dbh->do(qq{insert into benchmark values ('pub$i_pub',	'size$i_size', 'type$i_type', 'ip$i_ip',	'"0",0,"1",1,"2",2,"3",3,"4",4,"5",5,"6",6,"7",7,"8",8,"9",9')});

$dbh2->do(qq{insert into benchmark values ('pub$i_pub',	'size$i_size', 'type$i_type', 'ip$i_ip',	'"0",0,"1",1,"2",2,"3",3,"4",4,"5",5,"6",6,"7",7,"8",8,"9",9')});

						
			}
		}
	}
}

#----- storable -----

#store \%s_disp_hash, 'storable.dat';


#----- MLDBM -----

#tie %s_disp_hash, 'MLDBM', "mldbm.dat", O_CREAT|O_RDWR, 00644;
#untie %s_disp_hash;

#--------------------------------------------------------------------------

#----- r_hash -----

#$r_hash = sub {
#	foreach (1..$i_inner_loop) {
#		$i_pub = int(rand(100));
#		$i_size = int(rand(3));
#		$i_type = int(rand(4));
#		$i_ip = int(rand(20));

#		$rs_tmp = $s_disp_hash{join($s_delimiter,
#				"pub$i_pub","size$i_size","type$i_type","ip$i_ip")};
#	}
#};

#----- flat file -----

#$r_flatfile = sub {
#	foreach (1..$i_inner_loop) {
#		$i_pub = int(rand(100));
#		$i_size = int(rand(3));
#		$i_type = int(rand(4));
#		$i_ip = int(rand(20));

#		open(FH, "<flatfile/pub$i_pub/size$i_size/type$i_type/ip$i_ip");
#		%s_tmp = (eval(<FH>));
#		close(FH);
#	}
#};

#----- Postgres -----

$r_postgres = sub {
	foreach (1..$i_inner_loop) {
		$i_pub = int(rand(100));
		$i_size = int(rand(3));
		$i_type = int(rand(4));
		$i_ip = int(rand(20));

		@row_ary = $dbh->selectrow_array("select crr from benchmark where
				rtrim(pub) = 'pub$i_pub' and rtrim(size) = 'size$i_size' and
				rtrim(type) = 'type$i_type' and rtrim(ip) = 'ip$i_ip'");
	}
};
#----- MySQL -----
$r_mysql = sub {
	foreach (1..$i_inner_loop) {
		$i_pub = int(rand(100));
		$i_size = int(rand(3));
		$i_type = int(rand(4));
		$i_ip = int(rand(20));

		@row_ary2 = $dbh2->selectrow_array("select crr from benchmark where
				rtrim(pub) = 'pub$i_pub' and rtrim(size) = 'size$i_size' and
				rtrim(type) = 'type$i_type' and rtrim(ip) = 'ip$i_ip'");
	}
};
#----- Storable -----

#$r_storable = sub {
#	foreach (1..$i_inner_loop) {
#        $i_pub = int(rand(100));
#	    $i_size = int(rand(3));
#	    $i_type = int(rand(4));
#	    $i_ip = int(rand(20));

#		$rs_tmp = retrieve('storable.dat');
#		%s_disp_hash_storable = %{$rs_tmp}; 
		
#		$rs_tmp = $s_disp_hash_storable{join($s_delimiter,
#				"pub$i_pub","size$i_size","type$i_type","ip$i_ip")};
#	}	
#};

#----- MLDBM -----

#$r_mldbm = sub {
#	foreach (1..$i_inner_loop) {
#		$i_pub = int(rand(100));
#		$i_size = int(rand(3));
#		$i_type = int(rand(4));
#		$i_ip = int(rand(20));

#		tie(%s_disp_hash_mldbm, 'MLDBM', 'mldbm.dat', O_RDONLY,
#				00444);
				
#		$rs_tmp = $s_disp_hash_mldbm{join($s_delimiter,
#				"pub$i_pub","size$i_size","type$i_type","ip$i_ip")};

#		untie %s_disp_hash_mldbm;
#	}
#};

#--------------------------------------------------------------------------

foreach (1..2) {
#	print "hash\n";
#	print timestr(timeit($i_count, $r_hash)), "\n"; 
#	print "flatfile\n";
#	print timestr(timeit($i_count, $r_flatfile)), "\n";
	print "postgres\n";
	print timestr(timeit($i_count, $r_postgres)), "\n";
	print "mysql\n";
	print timestr(timeit($i_count, $r_mysql)), "\n";
#	print "storable\n";
#	print timestr(timeit($i_count, $r_storable)), "\n";
#	print "mldbm\n";
#	print timestr(timeit($i_count, $r_mldbm)), "\n";
}

$dbh->disconnect();
$dbh2->disconnect();