rserv/misc/rserv_test.pl

#!/usr/bin/perl -w

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


$| = 1;
my $control = 0;

my ($numprocs,$numstmts) = (4,100);
my ($debug,$verbose) = (0,0);
my ($help,$host,$port,$user,$password);

my $result = GetOptions(
        "debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help,
        "host=s" => \$host, "port=i" => \$port,
        "user=s" => \$user, "password=s" => \$password,
        "numprocs=i" => \$numprocs, "numstmts=i" => \$numstmts,
        );

if (defined($help) || (scalar(@ARGV) < 3)) {
    print "Usage: $0 [options] db table column
Options:
        --host=hostname --port=port
        --user=username --password=string
        --numprocs=4 --numstmts=100
";
    exit ((scalar(@ARGV) < 3)? 1:0);
}

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

my $db = $ARGV[0] || "master";
my $table = $ARGV[1];
my $col = $ARGV[2];

my $info = "dbname=$db";
$info = "$info host=$host" if (defined($host));
$info = "$info port=$port" if (defined($port));
$info = "$info user=$user" if (defined($user));
$info = "$info password=$password" if (defined($password));

my @pids = ();
my $q;

for (my $i=0; $i < $numprocs; $i++) {
    my $pid = fork();
    if (! defined($pid)) {
        print "Can't fork...\n";
    } elsif ($pid == 0) {
        doInserts($i+1);
        exit;
    } elsif ($pid != undef) {
        push @pids, $pid;
    }
}

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


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

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

    my $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";
    my $sql = "INSERT INTO $table ($col) VALUES";
    for (my $i = 0; $i < $numstmts; $i++) {
        my $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	dpavlin	1.2	#!/usr/bin/perl -w
2	dpavlin	1.1
3	dpavlin	1.2	use strict;
4	dpavlin	1.1	use Pg;
5			use Getopt::Long;
6			use POSIX ":sys_wait_h";
7
8
9			$\| = 1;
10	dpavlin	1.2	my $control = 0;
11	dpavlin	1.1
12	dpavlin	1.2	my ($numprocs,$numstmts) = (4,100);
13			my ($debug,$verbose) = (0,0);
14			my ($help,$host,$port,$user,$password);
15
16			my $result = GetOptions(
17			"debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help,
18			"host=s" => \$host, "port=i" => \$port,
19			"user=s" => \$user, "password=s" => \$password,
20			"numprocs=i" => \$numprocs, "numstmts=i" => \$numstmts,
21			);
22
23			if (defined($help) \|\| (scalar(@ARGV) < 3)) {
24			print "Usage: $0 [options] db table column
25			Options:
26			--host=hostname --port=port
27			--user=username --password=string
28			--numprocs=4 --numstmts=100
29			";
30			exit ((scalar(@ARGV) < 3)? 1:0);
31	dpavlin	1.1	}
32
33			print STDERR "Running $numprocs threads, $numstmts inserts each\n";
34
35	dpavlin	1.2	my $db = $ARGV[0] \|\| "master";
36			my $table = $ARGV[1];
37			my $col = $ARGV[2];
38
39			my $info = "dbname=$db";
40			$info = "$info host=$host" if (defined($host));
41			$info = "$info port=$port" if (defined($port));
42			$info = "$info user=$user" if (defined($user));
43			$info = "$info password=$password" if (defined($password));
44	dpavlin	1.1
45			my @pids = ();
46	dpavlin	1.2	my $q;
47
48			for (my $i=0; $i < $numprocs; $i++) {
49			my $pid = fork();
50			if (! defined($pid)) {
51			print "Can't fork...\n";
52			} elsif ($pid == 0) {
53	dpavlin	1.1	doInserts($i+1);
54			exit;
55			} elsif ($pid != undef) {
56			push @pids, $pid;
57			}
58			}
59
60	dpavlin	1.2	foreach my $pid (@pids) {
61	dpavlin	1.1	my $x = -1;
62			do {
63			sleep(1);
64			$x = waitpid($pid, 0);
65			} until $x == $pid;
66			}
67
68
69			#########################
70
71			sub doInserts {
72	dpavlin	1.2	my ($pid) = @_;
73	dpavlin	1.1	print "<$pid> Running...\n";
74
75	dpavlin	1.2	my $conn = Pg::connectdb($info);
76	dpavlin	1.1	if ($conn->status != PGRES_CONNECTION_OK) {
77			print "<$pid> Failed opening $info\n";
78			print "<$pid> Abort!\n";
79			last;
80			}
81
82			$result = $conn->exec("BEGIN");
83			if ($result->resultStatus ne PGRES_COMMAND_OK) {
84			print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
85			print "<$pid> Abort!\n";
86			last;
87			}
88
89			print "<$pid> Inserting $numstmts records...\n";
90	dpavlin	1.2	my $sql = "INSERT INTO $table ($col) VALUES";
91			for (my $i = 0; $i < $numstmts; $i++) {
92			my $time = time;
93	dpavlin	1.1	$q = "$sql ('test_${pid}_${i}_$time')";
94			$result = $conn->exec($q);
95			if ($result->resultStatus ne PGRES_COMMAND_OK) {
96			print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
97			$conn->exec("ROLLBACK");
98			print "<$pid> Abort!\n";
99			last;
100			}
101			}
102			print "<$pid> done!\n";
103
104			$result = $conn->exec("COMMIT");
105			if ($result->resultStatus ne PGRES_COMMAND_OK) {
106			print "<$pid> Error in query '$q': ". $conn->errorMessage."\n";
107			$conn->exec("ROLLBACK");
108			print "<$pid> Abort!\n";
109			last;
110			}
111
112			print "<$pid> Finished.\n";
113			$control++;
114			}