rserv/misc/rserv_test.pl

#!/usr/bin/perl -w

# original version by Nélio Alves Pereira Filho
# modification by Dobrica Pavlinusic

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