123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138 |
- <?php
- defined('GNUSOCIAL') || die();
- class SchemaUpdater
- {
- public function __construct($schema)
- {
- $this->schema = $schema;
- $this->checksums = $this->getChecksums();
- }
-
- public function register($tableName, array $tableDef)
- {
-
- if (is_a(ucfirst($tableName), 'Managed_DataObject', true)) {
- call_user_func("{$tableName}::beforeSchemaUpdate");
- }
- $this->tables[$tableName] = $tableDef;
- }
-
- public function checkSchema()
- {
- $checksums = $this->checksums;
- 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");
- } elseif ($checksums[$table] === $checksum) {
- common_log(LOG_DEBUG, "Last schema_version for $table up to date: $checksum");
- continue;
- } else {
- common_log(LOG_DEBUG, "Last schema_version for $table is {$checksums[$table]}: updating to $checksum");
- }
-
- $this->schema->ensureTable($table, $def);
- $this->saveChecksum($table, $checksum);
-
- }
- }
-
- public function checksum(array $def)
- {
- $flat = serialize($def);
- return hash('sha3-256', $flat);
- }
-
- protected function getChecksums()
- {
- $checksums = array();
- try {
- $sv = new Schema_version();
- $sv->find();
- while ($sv->fetch()) {
- $checksums[$sv->table_name] = $sv->checksum;
- }
- return $checksums;
- } catch (Exception $e) {
-
- common_log(LOG_DEBUG, "Possibly schema_version table doesn't exist yet.");
- }
- return $checksums;
- }
-
- protected function saveChecksum($table, $checksum)
- {
- 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) {
-
- common_log(LOG_DEBUG, "Possibly schema_version table doesn't exist yet.");
- }
- $this->checksums[$table] = $checksum;
- }
- }
|