[Bio] / FigKernelPackages / DBrtns.pm Repository:
ViewVC logotype

Annotation of /FigKernelPackages/DBrtns.pm

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.19 - (view) (download) (as text)

1 : efrank 1.1 package DBrtns;
2 :    
3 : parrello 1.18 # Inherit the DBKernel methods. We must do this BEFORE the "use strict".
4 :     use DBKernel;
5 :     @ISA = qw(DBKernel);
6 :    
7 : efrank 1.1 use strict;
8 : olson 1.5 use POSIX;
9 : efrank 1.1 use DBI;
10 :     use FIG_Config;
11 :    
12 :     use Data::Dumper;
13 : overbeek 1.2 use Carp;
14 :    
15 :     sub new {
16 : olson 1.19 my($class,$dbms,$dbname,$dbuser,$dbpass,$dbport, $dbhost) = @_;
17 : efrank 1.1
18 :     $dbms = defined($dbms) ? $dbms : $FIG_Config::dbms;
19 :     $dbname = defined($dbname) ? $dbname : $FIG_Config::db;
20 :     $dbuser = defined($dbuser) ? $dbuser : $FIG_Config::dbuser;
21 :     $dbpass = defined($dbpass) ? $dbpass : $FIG_Config::dbpass;
22 :     $dbport = defined($dbport) ? $dbport : $FIG_Config::dbport;
23 : olson 1.19 $dbhost = defined($dbhost) ? $dbhost : $FIG_Config::dbhost;
24 : efrank 1.1
25 : olson 1.19 return DBKernel::new($class, $dbms, $dbname, $dbuser, $dbpass, $dbport, $dbhost);
26 : efrank 1.1 }
27 :    
28 :     sub vacuum_it {
29 :     my($self,@tables) = @_;
30 :     my($table);
31 :    
32 :     my $dbh = $self->{_dbh};
33 :     my $dbms = $self->{_dbms};
34 : parrello 1.18 if ($dbms eq "mysql") {
35 : efrank 1.1 return;
36 :     }
37 :    
38 :     # this chunk is for Pg (Postgres)
39 : parrello 1.18 if (@tables == 0) {
40 :     $self->SQL("VACUUM ANALYZE");
41 :     } else {
42 :     foreach $table (@tables) {
43 :     $self->SQL("VACUUM ANALYZE $table");
44 :     }
45 : efrank 1.1 }
46 :     }
47 :    
48 : olson 1.11 =head1 get_inserted_id
49 :    
50 :     Return the last ID of a row inserted into an autonumber/serial-containing table.
51 :    
52 :     =cut
53 :    
54 : parrello 1.18 sub get_inserted_id {
55 : olson 1.11 my($self, $table, $sth) = @_;
56 : parrello 1.18 if ($self->{_dbms} eq "Pg") {
57 :     my $oid = $sth->{pg_oid_status};
58 :     my $ret = $self->SQL("select id from $table where oid = ?", undef, $oid);
59 :     return $ret->[0]->[0];
60 :     } elsif ($self->{_dbms} eq "mysql") {
61 :     my $id = $self->{_dbh}->{mysql_insertid};
62 :     # print "mysql got $id\n";
63 :     return $id;
64 : olson 1.11 }
65 :     }
66 :    
67 : olson 1.5 #
68 :     # Following are database administration routines. They create an instance of a ServerAdmin class
69 :     # for the appropriate server type (in order to eliminate the if mysql / if pg / etc stuff).
70 :     #
71 :    
72 :     sub get_server_admin
73 :     {
74 :     if ($FIG_Config::dbms eq "mysql")
75 :     {
76 :     return MysqlAdmin->new();
77 :     }
78 :     elsif ($FIG_Config::dbms eq "Pg")
79 :     {
80 :     return new PostgresAdmin();
81 :     }
82 :     else
83 :     {
84 :     warn "Unknown server type $FIG_Config::dbms\n";
85 :     return undef;
86 :     }
87 :     }
88 :     package MysqlAdmin;
89 :    
90 :     use POSIX;
91 :     use DBI;
92 :    
93 :     sub new
94 :     {
95 :     my($class) = @_;
96 :    
97 :     my $self = {};
98 :    
99 :     return bless($self, $class);
100 :     }
101 :    
102 :     sub init_db
103 :     {
104 :     my($self, $db_dir) = @_;
105 :    
106 :     if (!$db_dir)
107 :     {
108 :     warn "init_db failed: db_dir must be provided\n";
109 :     return;
110 :     }
111 :    
112 :     if (-d "$db_dir/mysql")
113 :     {
114 :     warn "init_db: mysql data directory already exists\n";
115 :     return;
116 :     }
117 :    
118 :     my $exe = "$FIG_Config::ext_bin/mysql_install_db";
119 :     if (! -x $exe)
120 :     {
121 :     $exe = "mysql_install_db";
122 : parrello 1.18 }
123 : olson 1.5
124 :     my $rc = system($exe,
125 :     "--datadir=$db_dir",
126 : olson 1.15 "--basedir=$FIG_Config::common_runtime",
127 : olson 1.5 "--user=$FIG_Config::dbuser");
128 :     if ($rc != 0)
129 :     {
130 :     my $err = $?;
131 :     if (WIFEXITED($err))
132 :     {
133 :     my $exitstat = WEXITSTATUS($err);
134 :     warn "init_db failed: $exe returned result code $exitstat\n";
135 :     }
136 :     else
137 :     {
138 :     warn "init_db failed: $exe died with signal ", WTERMSIG($err), "\n";
139 :     }
140 :     return;
141 :     }
142 :    
143 :     return 1;
144 :     }
145 :    
146 :     sub create_database
147 :     {
148 :     my($self, $db_name) = @_;
149 :    
150 :     my $drh = DBI->install_driver("mysql");
151 :    
152 :     my @dbs = DBI->data_sources("mysql", { host => $FIG_Config::dbhost,
153 :     user => $FIG_Config::dbuser,
154 :     password => $FIG_Config::dbpass });
155 :     if (grep { $_ eq $db_name } @dbs)
156 :     {
157 :     warn "Database $db_name already exists\n";
158 :     return;
159 :     }
160 :    
161 :     my $rc = $drh->func('createdb', $db_name, $FIG_Config::dbhost,
162 :     $FIG_Config::dbuser, $FIG_Config::dbpass, 'admin');
163 : parrello 1.18
164 : olson 1.5
165 :     if (!$rc)
166 :     {
167 :     warn "create_database: createdb call failed: $DBI::errstr\n";
168 :     return;
169 :     }
170 : parrello 1.18
171 : olson 1.5 return 1;
172 :     }
173 :    
174 :     sub start_server
175 :     {
176 : olson 1.16 my($self, $dont_fork) = @_;
177 : olson 1.5
178 :     print "Starting mysql server\n";
179 :    
180 :     my(@opts);
181 :    
182 :     push(@opts, "--port=$FIG_Config::dbport");
183 :     #
184 :     # Don't do this; dbuser isn't the unix uid that we are using.
185 :     #
186 :     #push(@opts, "--user=$FIG_Config::dbuser");
187 :     push(@opts, "--basedir=$FIG_Config::common_runtime");
188 :     push(@opts, "--datadir=$FIG_Config::db_datadir");
189 :     push(@opts, "--ledir=$FIG_Config::common_runtime/libexec");
190 : olson 1.6 push(@opts, "--old-password");
191 :     push(@opts, "--max-allowed-packet=128M");
192 : olson 1.5 #
193 :     # Oddly, this doesn't seem to work. need to set the environment variable.
194 :     #
195 :     #push(@opts, "--port=$FIG_Config::dbport");
196 :    
197 : olson 1.15 if (@FIG_Config::db_server_startup_options)
198 :     {
199 :     push(@opts, @FIG_Config::db_server_startup_options)
200 :     }
201 :    
202 : olson 1.5 #
203 :     # We are going to assume that if mysql has shipped with this release, we'll use it. Otherwise
204 :     # try to use a system one.
205 :     #
206 :    
207 :     my $exe = "$FIG_Config::ext_bin/mysqld_safe";
208 :     if (! -x $exe)
209 :     {
210 :     $exe = "mysqld_safe";
211 :     }
212 :    
213 : olson 1.16 if ($dont_fork)
214 : olson 1.5 {
215 :     $ENV{MYSQL_TCP_PORT} = $FIG_Config::dbport;
216 :     exec $exe, @opts;
217 :     }
218 : olson 1.16 else
219 :     {
220 :     my $pid = fork;
221 :    
222 :     if ($pid == 0)
223 :     {
224 :     POSIX::setsid();
225 : parrello 1.18
226 : olson 1.16 $ENV{MYSQL_TCP_PORT} = $FIG_Config::dbport;
227 :     exec $exe, @opts;
228 :     }
229 :     print "Forked db server $pid\n";
230 :     }
231 : olson 1.5
232 :     }
233 :    
234 : parrello 1.18 1;

MCS Webmaster
ViewVC Help
Powered by ViewVC 1.0.3