X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=lib%2Fschemaupdater.php;h=38b5b93865255cc719875385f9e927c999da4abe;hb=15ab9ff9e3303255ff14166ee86ffdf3bc4f52ce;hp=b872b0d5785443317cac587461ca145d451c4d82;hpb=b483a0549f115b2d5f2cfb3d447e6cc5c4f469f3;p=quix0rs-gnu-social.git

diff --git a/lib/schemaupdater.php b/lib/schemaupdater.php
index b872b0d578..38b5b93865 100644
--- a/lib/schemaupdater.php
+++ b/lib/schemaupdater.php
@@ -28,28 +28,40 @@
  * @link      http://status.net/
  */
 
-if (!defined('STATUSNET')) {
-    exit(1);
-}
+if (!defined('GNUSOCIAL')) { exit(1); }
 
 class SchemaUpdater
 {
     public function __construct($schema)
     {
         $this->schema = $schema;
-        $this->conn = $conn;
         $this->checksums = $this->getChecksums();
     }
 
     /**
-     * @param array $tableDefs
+     * @param string $tableName
+     * @param array $tableDef
+     */
+    public function register($tableName, array $tableDef)
+    {
+        // Check if the table we're registering is related to a Managed_DataObject
+        if (is_a(ucfirst($tableName), 'Managed_DataObject', true)) {
+            call_user_func("{$tableName}::beforeSchemaUpdate");
+        }
+
+        $this->tables[$tableName] = $tableDef;
+    }
+
+    /**
+     * Go ping em!
+     *
      * @fixme handle tables that belong on different database servers...?
      */
-    public function checkTables(array $tableDefs)
+    public function checkSchema()
     {
         $checksums = $this->checksums;
-        foreach ($tableDefs as $table => $def) {
-            $checksum = $this->tableChecksum($def);
+        foreach ($this->tables as $table => $def) {
+            $checksum = $this->checksum($def);
             if (empty($checksums[$table])) {
                 common_log(LOG_DEBUG, "No previous schema_version for $table: updating to $checksum");
             } else if ($checksums[$table] == $checksum) {
@@ -58,10 +70,10 @@ class SchemaUpdater
             } else {
                 common_log(LOG_DEBUG, "Last schema_version for $table is {$checksums[$table]}: updating to $checksum");
             }
-            $this->conn->query('BEGIN');
+            //$this->conn->query('BEGIN');
             $this->schema->ensureTable($table, $def);
             $this->saveChecksum($table, $checksum);
-            $this->conn->commit();
+            //$this->conn->commit();
         }
     }
 
@@ -86,10 +98,17 @@ class SchemaUpdater
     {
         $checksums = array();
 
-        $sv = new Schema_version();
-        $sv->find();
-        while ($sv->fetch()) {
-            $checksums[$sv->table_name] = $sv->checksum;
+        try {
+            $sv = new Schema_version();
+            $sv->find();
+            while ($sv->fetch()) {
+                $checksums[$sv->table_name] = $sv->checksum;
+            }
+
+            return $checksums;
+        } catch (Exception $e) {
+            // no dice!
+            common_log(LOG_DEBUG, "Possibly schema_version table doesn't exist yet.");
         }
 
         return $checksums;
@@ -103,14 +122,19 @@ class SchemaUpdater
      */
     protected function saveChecksum($table, $checksum)
     {
-        $sv = new Schema_version();
-        $sv->table_name = $table;
-        $sv->checksum = $checksum;
-        $sv->modified = common_sql_now();
-        if (isset($this->checksums[$table])) {
-            $sv->update();
-        } else {
-            $sv->insert();
+        try {
+            $sv = new Schema_version();
+            $sv->table_name = $table;
+            $sv->checksum = $checksum;
+            $sv->modified = common_sql_now();
+            if (isset($this->checksums[$table])) {
+                $sv->update();
+            } else {
+                $sv->insert();
+            }
+        } catch (Exception $e) {
+            // no dice!
+            common_log(LOG_DEBUG, "Possibly schema_version table doesn't exist yet.");
         }
         $this->checksums[$table] = $checksum;
     }