rserv/misc/rserv_test.pl

#!/usr/bin/perl


use Pg;
use Getopt::Long;
use POSIX ":sys_wait_h";


$| = 1;
$control = 0;


$result = GetOptions("numprocs=i", "numstmts=i", "help");
if (defined($opt_help)) {
    print "Usage: $0 --numprocs=number_of_processes --numstmts=number_of_statements\n";
    print "\n";
    exit(0);
}


my $numprocs = $opt_numprocs || 4;
my $numstmts = $opt_numstmts || 100;

print STDERR "Running $numprocs threads, $numstmts inserts each\n";

my $info = "dbname=immanis host=kelly user=root";
#my $info = "dbname=replicate host=alicia user=nelio";

my @pids = ();
for ($i=0; $i < $numprocs; $i++) {
    if (($pid = fork()) == 0) {
        doInserts($i+1);
        exit;
    } elsif ($pid != undef) {
        push @pids, $pid;
    }
}

foreach $pid (@pids) {
    my $x = -1;
    do {
        sleep(1);
        $x = waitpid($pid, 0);
    } until $x == $pid;
}


#########################

sub doInserts {
    ($pid) = @_;
    print "<$pid> Running...\n";

    $conn = Pg::connectdb($info);
    if ($conn->status != PGRES_CONNECTION_OK) {
        print "<$pid> Failed opening $info\n";
        print "<$pid> Abort!\n";
        last;
    }
 
    $result = $conn->exec("BEGIN");
    if ($result->resultStatus ne PGRES_COMMAND_OK) {
        print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
        print "<$pid> Abort!\n";
        last;
    }
    
    print "<$pid> Inserting $numstmts records...\n";
    $sql = "INSERT INTO test (value) VALUES";
    for ($i = 0; $i < $numstmts; $i++) {
        $time = time;
        $q = "$sql ('test_${pid}_${i}_$time')";
        $result = $conn->exec($q);
        if ($result->resultStatus ne PGRES_COMMAND_OK) {
            print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
            $conn->exec("ROLLBACK");
            print "<$pid> Abort!\n";
            last;
        }
    }
    print "<$pid> done!\n";

    $result = $conn->exec("COMMIT");
    if ($result->resultStatus ne PGRES_COMMAND_OK) {
        print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
        $conn->exec("ROLLBACK");
        print "<$pid> Abort!\n";
        last;
    }
    
    print "<$pid> Finished.\n";
    $control++;
}
1	#!/usr/bin/perl
2
3
4	use Pg;
5	use Getopt::Long;
6	use POSIX ":sys_wait_h";
7
8
9	$\| = 1;
10	$control = 0;
11
12
13	$result = GetOptions("numprocs=i", "numstmts=i", "help");
14	if (defined($opt_help)) {
15	print "Usage: $0 --numprocs=number_of_processes --numstmts=number_of_statements\n";
16	print "\n";
17	exit(0);
18	}
19
20
21	my $numprocs = $opt_numprocs \|\| 4;
22	my $numstmts = $opt_numstmts \|\| 100;
23
24	print STDERR "Running $numprocs threads, $numstmts inserts each\n";
25
26	my $info = "dbname=immanis host=kelly user=root";
27	#my $info = "dbname=replicate host=alicia user=nelio";
28
29	my @pids = ();
30	for ($i=0; $i < $numprocs; $i++) {
31	if (($pid = fork()) == 0) {
32	doInserts($i+1);
33	exit;
34	} elsif ($pid != undef) {
35	push @pids, $pid;
36	}
37	}
38
39	foreach $pid (@pids) {
40	my $x = -1;
41	do {
42	sleep(1);
43	$x = waitpid($pid, 0);
44	} until $x == $pid;
45	}
46
47
48	#########################
49
50	sub doInserts {
51	($pid) = @_;
52	print "<$pid> Running...\n";
53
54	$conn = Pg::connectdb($info);
55	if ($conn->status != PGRES_CONNECTION_OK) {
56	print "<$pid> Failed opening $info\n";
57	print "<$pid> Abort!\n";
58	last;
59	}
60
61	$result = $conn->exec("BEGIN");
62	if ($result->resultStatus ne PGRES_COMMAND_OK) {
63	print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
64	print "<$pid> Abort!\n";
65	last;
66	}
67
68	print "<$pid> Inserting $numstmts records...\n";
69	$sql = "INSERT INTO test (value) VALUES";
70	for ($i = 0; $i < $numstmts; $i++) {
71	$time = time;
72	$q = "$sql ('test_${pid}_${i}_$time')";
73	$result = $conn->exec($q);
74	if ($result->resultStatus ne PGRES_COMMAND_OK) {
75	print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
76	$conn->exec("ROLLBACK");
77	print "<$pid> Abort!\n";
78	last;
79	}
80	}
81	print "<$pid> done!\n";
82
83	$result = $conn->exec("COMMIT");
84	if ($result->resultStatus ne PGRES_COMMAND_OK) {
85	print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
86	$conn->exec("ROLLBACK");
87	print "<$pid> Abort!\n";
88	last;
89	}
90
91	print "<$pid> Finished.\n";
92	$control++;
93	}