Changeset 28534 for branches/pap/ippToPsps/scripts/createDb.pl

Timestamp:

Jun 28, 2010, 8:11:04 PM (16 years ago)

Author:

Paul Price

Message:

Merging trunk in preparation for merging branch to trunk.

Location:

branches/pap

Files:

• . (modified) (1 prop)
• ippToPsps/scripts/createDb.pl (modified) (5 diffs)

branches/pap/ippToPsps/scripts/createDb.pl

@@ -25 +25 @@
 print "* Connected to '$dbname' on 'dbserver'\n";
 
-if (!doesTableExist("revision")) {
-    createRevision_1();
+my $currentRevision = -1;
+
+my $latestRevision = 3;
+    print "* Latest revision = $latestRevision\n";
+
+while ($currentRevision != $latestRevision) {
+
+    $currentRevision = getRevision();
+    print "* Current revision = $currentRevision\n";
+
+    if ($currentRevision == 0) {createRevision_1();}
+    elsif ($currentRevision == 1) {createRevision_2();}
+    elsif ($currentRevision == 2) {createRevision_3();}
+
 }
-
-
 $db->disconnect();
 print "* Disconnected from '$dbname' on 'dbserver'\n";
 print "*\n*******************************************************************************\n\n";
+
 
 #######################################################################################
@@ -50 +61 @@
             primary key (revision)
             );
-SQL
-        $query->execute;
-
-    $query = $db->prepare(<<SQL);
-    INSERT INTO revision (revision) VALUES (1);
 SQL
         $query->execute;
@@ -77 +83 @@
     $query->execute;
 
+setRevision(1);
+
+}
+
+#######################################################################################
+# 
+# Create revision 2 of the database 
+#
+#######################################################################################
+sub createRevision_2 {
+
+    print "* Creating revision 2 of '$dbname'\n";
+
+        my $query = $db->prepare(<<SQL);
+
+        ALTER TABLE batches 
+        ADD COLUMN merged TINYINT DEFAULT 0
+SQL
+    $query->execute;
+
+setRevision(2);
+}
+
+#######################################################################################
+# 
+# Create revision 3 of the database 
+#
+#######################################################################################
+sub createRevision_3 {
+
+    print "* Creating revision 3 of '$dbname'\n";
+
+        my $query = $db->prepare(<<SQL);
+
+        ALTER TABLE batches 
+        ADD COLUMN total_detections BIGINT DEFAULT 0
+SQL
+    $query->execute;
+
+setRevision(3);
+}
+
+
+#######################################################################################
+# 
+# Sets current revision of ippToPsps database
+#
+#######################################################################################
+sub setRevision {
+    my ($revision) = @_;
+
+    my $query = $db->prepare(<<SQL);
+    INSERT INTO revision (revision) VALUES ($revision);
+SQL
+        $query->execute;
 }
 
@@ -82 +143 @@
 # 
 # Gets current revision of ippToPsps database
+#
+#######################################################################################
+sub getRevision {
+
+    if (!doesTableExist("revision")) {return 0;}
+
+    my $query = $db->prepare(<<SQL);
+
+    SELECT revision
+        FROM revision
+        ORDER BY revision DESC LIMIT 1;
+SQL
+   $query->execute;
+   my @row = $query->fetchrow_array();
+
+   return $row[0];
+}
+
+#######################################################################################
+# 
+# Checks whether a certain table exists 
 #
 #######################################################################################
@@ -99 +181 @@
     my $count = $query->fetchrow_array();
 
-    printf( "* Table '$table' %s\n", $count ? "exists" : "does not exist");
-
     return $count;
 }