Contents of /bin/MasterInit

#!/usr/bin/perl -w
# MasterInit
# Vadim Mikheev, (c) 2000, PostgreSQL Inc.

eval '(exit $?0)' && eval 'exec perl -S $0 ${1+"$@"}'
    & eval 'exec perl -S $0 $argv:q'
    if 0;

use strict;
use Pg;
use Getopt::Long;

$| = 1;

my ($debug,$verbose) = (0,0);
my ($help,$masterhost,$masterport,$masteruser,$masterpassword);
my $lib;

my $result = GetOptions(
        "debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help,
        "masterhost=s" => \$masterhost, "masterport=i" => \$masterport,
        "masteruser=s" => \$masteruser, "masterpassword=s" => \$masterpassword,
        "lib=s" => \$lib);

if (defined($help) || (scalar(@ARGV) < 1)) {
    print "Usage: $0 [options] masterdb
Options:
        --masterhost=hostname --masterport=port
        --masteruser=username --masterpassword=string
        --lib=libpath
";
    exit ((scalar(@ARGV) < 1)? 1:0);
}

my $master = $ARGV[0] || "master";

my $minfo = "dbname=$master";
$minfo = "$minfo host=$masterhost" if (defined($masterhost));
$minfo = "$minfo port=$masterport" if (defined($masterport));
$minfo = "$minfo user=$masteruser" if (defined($masteruser));
$minfo = "$minfo password=$masterpassword" if (defined($masterpassword));

if (!defined($lib) || !-e $lib) {
        # find my compiled rserv.so module
        $lib = $0; $lib =~ s#/[^/]+$#/../lib#;
        if ($lib =~ m#^\.#) {
                my $pwd = `pwd`;
                chomp($pwd);
                $lib = "$pwd/$lib/";
        }
        while ($lib =~ s#/[^/]+/\.\./#/#g) {};
        while ($lib =~ s#/\./#/#g) {};
        $lib =~ s#//#/#g;

        if (-e "$lib/rserv.so") {
                $lib .= "rserv.so";
        } else {
                print STDERR "Can't find compiled rserv.so in $lib. Go there and type make.\n";
                exit 1;
        }
}
print "Using lib '$lib'\n" if ($verbose);

sub RollbackAndQuit {
    my $conn = shift @_;

    print STDERR "Error in query: ", $conn->errorMessage;
    $conn->exec("ROLLBACK");
    exit (-1);
}

my $conn = Pg::connectdb($minfo);
if ($conn->status != PGRES_CONNECTION_OK) {
    print STDERR "Failed opening $minfo\n";
    exit 1;
}

$result = $conn->exec("BEGIN");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

$result = $conn->exec("set transaction isolation level serializable");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# List of slave servers
$result = $conn->exec("create table _RSERV_SERVERS_" .
                      " (server serial primary key, host text not ".
                      "null, port int4 default 5432, dbase text not ".
                      "null, unique(host,port,dbase))");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# List of replicated tables
$result = $conn->exec("create table _RSERV_TABLES_" .
                      " (tname name not null, cname name not null, ".
                      "reloid oid primary key, key int4 not null)");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

## should always call MasterDelTable
#$result = $conn->exec("CREATE RULE _rserv_deltrig_ AS ON delete to _RSERV_TABLES_ DO (DELETE FROM pg_trigger WHERE tgname='_rserv_trigger_t_' AND tgrelid=(SELECT oid FROM pg_class WHERE relname=old.tname);UPDATE pg_class SET reltriggers=reltriggers-1 WHERE relname=old.tname)");
#RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# Bookkeeping log for row replication   
$result = $conn->exec("create table _RSERV_LOG_" .
                      " (reloid oid REFERENCES _RSERV_TABLES_(reloid) ON ".
                      "DELETE CASCADE ON UPDATE CASCADE, logid int4 not ".
                      "null, logtime timestamp not null, insert smallint, ".
                      "update smallint, delete smallint, key text, ".
                      "server int4, ".
                      "CONSTRAINT only_one CHECK (insert+update+delete=1))");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# This is to speedup lookup of deleted tuples
$result = $conn->exec("create index _RSERV_LOG_INDX_DLT_ID_ on _RSERV_LOG_ ".
                      "(delete, logid) WHERE delete = 1");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# This is to speedup lookup of updated tuples
$result = $conn->exec("create index _RSERV_LOG_INDX_UPD_ID_ on _RSERV_LOG_ ".
                      "(update, logid) WHERE update = 1");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# This is to speedup lookup of insert tuples
$result = $conn->exec("create index _RSERV_LOG_INDX_INS_ID_ on _RSERV_LOG_ ".
                      "(insert, logid) WHERE insert = 1");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# This is to speedup cleanup
$result = $conn->exec("create index _RSERV_LOG_INDX_TM_ID_ on _RSERV_LOG_ ".
                      "(logtime, logid)");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# This is to speedup trigger 
$result = $conn->exec("create index _RSERV_LOG_INDX_REL_KEY_ on _RSERV_LOG_ ".
                      "(reloid, key)");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# View to help managing _rserv_log_ table
$result = $conn->exec("CREATE VIEW _RSERV_HUMAN_LOG_ AS SELECT log.logid, ".
                      "tab.tname AS table_name, tab.cname AS column_name, ".
                      "log.key AS column_value, log.insert, log.update, ".
                      "log.delete, log.logtime FROM _RSERV_LOG_ log, ".
                      "_RSERV_TABLES_ tab WHERE tab.reloid ".
                      "= log.reloid ORDER BY log.logtime");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# View to help logging daily transactions
$result = $conn->exec("CREATE VIEW _RSERV_DAILY_LOG_ AS ".
                      "SELECT count(*) AS \"# records\", ".
                      "to_char(_rserv_log_.logtime, 'YYYY-MM-DD') ".
                      "AS day FROM _rserv_log_ GROUP BY day");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# Sync point for each slave server
$result = $conn->exec("create table _RSERV_SYNC_ " .
                      "(server int REFERENCES _RSERV_SERVERS_(server) ".
                      "ON DELETE CASCADE ON UPDATE CASCADE, syncid int4 ".
                      "not null, synctime timestamp, status int4 not null,".
                      " minid int4 not null, maxid int4 not null, active text)");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

$result = $conn->exec("create index _RSERV_SYNC_INDX_SRV_ID_ on _RSERV_SYNC_ ".
                      "(server, syncid)");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

# Sync point reference numbers
$result = $conn->exec("create sequence _rserv_sync_seq_");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

$result = $conn->exec("CREATE FUNCTION _rserv_log_() RETURNS opaque" .
                      " AS '$lib' LANGUAGE 'c'");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

$result = $conn->exec("CREATE FUNCTION _rserv_sync_(int4) RETURNS int4" .
                      " AS '$lib' LANGUAGE 'c'");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

$result = $conn->exec("CREATE FUNCTION _rserv_debug_(int4) RETURNS int4" .
                      " AS '$lib' LANGUAGE 'c'");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

$result = $conn->exec("COMMIT");
RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);

exit (0);
1	#!/usr/bin/perl -w
2	# MasterInit
3	# Vadim Mikheev, (c) 2000, PostgreSQL Inc.
4
5	eval '(exit $?0)' && eval 'exec perl -S $0 ${1+"$@"}'
6	& eval 'exec perl -S $0 $argv:q'
7	if 0;
8
9	use strict;
10	use Pg;
11	use Getopt::Long;
12
13	$\| = 1;
14
15	my ($debug,$verbose) = (0,0);
16	my ($help,$masterhost,$masterport,$masteruser,$masterpassword);
17	my $lib;
18
19	my $result = GetOptions(
20	"debug!" => \$debug, "verbose!" => \$verbose, "help" => \$help,
21	"masterhost=s" => \$masterhost, "masterport=i" => \$masterport,
22	"masteruser=s" => \$masteruser, "masterpassword=s" => \$masterpassword,
23	"lib=s" => \$lib);
24
25	if (defined($help) \|\| (scalar(@ARGV) < 1)) {
26	print "Usage: $0 [options] masterdb
27	Options:
28	--masterhost=hostname --masterport=port
29	--masteruser=username --masterpassword=string
30	--lib=libpath
31	";
32	exit ((scalar(@ARGV) < 1)? 1:0);
33	}
34
35	my $master = $ARGV[0] \|\| "master";
36
37	my $minfo = "dbname=$master";
38	$minfo = "$minfo host=$masterhost" if (defined($masterhost));
39	$minfo = "$minfo port=$masterport" if (defined($masterport));
40	$minfo = "$minfo user=$masteruser" if (defined($masteruser));
41	$minfo = "$minfo password=$masterpassword" if (defined($masterpassword));
42
43	if (!defined($lib) \|\| !-e $lib) {
44	# find my compiled rserv.so module
45	$lib = $0; $lib =~ s#/[^/]+$#/../lib#;
46	if ($lib =~ m#^\.#) {
47	my $pwd = `pwd`;
48	chomp($pwd);
49	$lib = "$pwd/$lib/";
50	}
51	while ($lib =~ s#/[^/]+/\.\./#/#g) {};
52	while ($lib =~ s#/\./#/#g) {};
53	$lib =~ s#//#/#g;
54
55	if (-e "$lib/rserv.so") {
56	$lib .= "rserv.so";
57	} else {
58	print STDERR "Can't find compiled rserv.so in $lib. Go there and type make.\n";
59	exit 1;
60	}
61	}
62	print "Using lib '$lib'\n" if ($verbose);
63
64	sub RollbackAndQuit {
65	my $conn = shift @_;
66
67	print STDERR "Error in query: ", $conn->errorMessage;
68	$conn->exec("ROLLBACK");
69	exit (-1);
70	}
71
72	my $conn = Pg::connectdb($minfo);
73	if ($conn->status != PGRES_CONNECTION_OK) {
74	print STDERR "Failed opening $minfo\n";
75	exit 1;
76	}
77
78	$result = $conn->exec("BEGIN");
79	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
80
81	$result = $conn->exec("set transaction isolation level serializable");
82	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
83
84	# List of slave servers
85	$result = $conn->exec("create table _RSERV_SERVERS_" .
86	" (server serial primary key, host text not ".
87	"null, port int4 default 5432, dbase text not ".
88	"null, unique(host,port,dbase))");
89	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
90
91	# List of replicated tables
92	$result = $conn->exec("create table _RSERV_TABLES_" .
93	" (tname name not null, cname name not null, ".
94	"reloid oid primary key, key int4 not null)");
95	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
96
97	## should always call MasterDelTable
98	#$result = $conn->exec("CREATE RULE _rserv_deltrig_ AS ON delete to _RSERV_TABLES_ DO (DELETE FROM pg_trigger WHERE tgname='_rserv_trigger_t_' AND tgrelid=(SELECT oid FROM pg_class WHERE relname=old.tname);UPDATE pg_class SET reltriggers=reltriggers-1 WHERE relname=old.tname)");
99	#RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
100
101	# Bookkeeping log for row replication
102	$result = $conn->exec("create table _RSERV_LOG_" .
103	" (reloid oid REFERENCES _RSERV_TABLES_(reloid) ON ".
104	"DELETE CASCADE ON UPDATE CASCADE, logid int4 not ".
105	"null, logtime timestamp not null, insert smallint, ".
106	"update smallint, delete smallint, key text, ".
107	"server int4, ".
108	"CONSTRAINT only_one CHECK (insert+update+delete=1))");
109	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
110
111	# This is to speedup lookup of deleted tuples
112	$result = $conn->exec("create index _RSERV_LOG_INDX_DLT_ID_ on _RSERV_LOG_ ".
113	"(delete, logid) WHERE delete = 1");
114	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
115
116	# This is to speedup lookup of updated tuples
117	$result = $conn->exec("create index _RSERV_LOG_INDX_UPD_ID_ on _RSERV_LOG_ ".
118	"(update, logid) WHERE update = 1");
119	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
120
121	# This is to speedup lookup of insert tuples
122	$result = $conn->exec("create index _RSERV_LOG_INDX_INS_ID_ on _RSERV_LOG_ ".
123	"(insert, logid) WHERE insert = 1");
124	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
125
126	# This is to speedup cleanup
127	$result = $conn->exec("create index _RSERV_LOG_INDX_TM_ID_ on _RSERV_LOG_ ".
128	"(logtime, logid)");
129	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
130
131	# This is to speedup trigger
132	$result = $conn->exec("create index _RSERV_LOG_INDX_REL_KEY_ on _RSERV_LOG_ ".
133	"(reloid, key)");
134	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
135
136	# View to help managing _rserv_log_ table
137	$result = $conn->exec("CREATE VIEW _RSERV_HUMAN_LOG_ AS SELECT log.logid, ".
138	"tab.tname AS table_name, tab.cname AS column_name, ".
139	"log.key AS column_value, log.insert, log.update, ".
140	"log.delete, log.logtime FROM _RSERV_LOG_ log, ".
141	"_RSERV_TABLES_ tab WHERE tab.reloid ".
142	"= log.reloid ORDER BY log.logtime");
143	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
144
145	# View to help logging daily transactions
146	$result = $conn->exec("CREATE VIEW _RSERV_DAILY_LOG_ AS ".
147	"SELECT count(*) AS \"# records\", ".
148	"to_char(_rserv_log_.logtime, 'YYYY-MM-DD') ".
149	"AS day FROM _rserv_log_ GROUP BY day");
150	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
151
152	# Sync point for each slave server
153	$result = $conn->exec("create table _RSERV_SYNC_ " .
154	"(server int REFERENCES _RSERV_SERVERS_(server) ".
155	"ON DELETE CASCADE ON UPDATE CASCADE, syncid int4 ".
156	"not null, synctime timestamp, status int4 not null,".
157	" minid int4 not null, maxid int4 not null, active text)");
158	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
159
160	$result = $conn->exec("create index _RSERV_SYNC_INDX_SRV_ID_ on _RSERV_SYNC_ ".
161	"(server, syncid)");
162	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
163
164	# Sync point reference numbers
165	$result = $conn->exec("create sequence _rserv_sync_seq_");
166	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
167
168	$result = $conn->exec("CREATE FUNCTION _rserv_log_() RETURNS opaque" .
169	" AS '$lib' LANGUAGE 'c'");
170	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
171
172	$result = $conn->exec("CREATE FUNCTION _rserv_sync_(int4) RETURNS int4" .
173	" AS '$lib' LANGUAGE 'c'");
174	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
175
176	$result = $conn->exec("CREATE FUNCTION _rserv_debug_(int4) RETURNS int4" .
177	" AS '$lib' LANGUAGE 'c'");
178	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
179
180	$result = $conn->exec("COMMIT");
181	RollbackAndQuit($conn) if ($result->resultStatus ne PGRES_COMMAND_OK);
182
183	exit (0);