83 lines
2.4 KiB
PHP
83 lines
2.4 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace danog\MadelineProto\Db;
|
||
|
|
|
||
|
|
use function Amp\call;
|
||
|
|
|
||
|
|
abstract class SqlArray extends DriverArray
|
||
|
|
{
|
||
|
|
/**
|
||
|
|
* Create table for property.
|
||
|
|
*
|
||
|
|
* @return array|null
|
||
|
|
* @throws \Throwable
|
||
|
|
*/
|
||
|
|
abstract protected function prepareTable();
|
||
|
|
|
||
|
|
abstract protected function renameTable(string $from, string $to);
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @param string $name
|
||
|
|
* @param DbArray|array|null $value
|
||
|
|
* @param string $tablePrefix
|
||
|
|
* @param array $settings
|
||
|
|
*
|
||
|
|
* @return Promise
|
||
|
|
*/
|
||
|
|
public static function getInstance(string $name, $value = null, string $tablePrefix = '', array $settings = []): Promise
|
||
|
|
{
|
||
|
|
$tableName = "{$tablePrefix}_{$name}";
|
||
|
|
if ($value instanceof static && $value->table === $tableName) {
|
||
|
|
$instance = &$value;
|
||
|
|
} else {
|
||
|
|
$instance = new static();
|
||
|
|
$instance->table = $tableName;
|
||
|
|
}
|
||
|
|
|
||
|
|
$instance->settings = $settings;
|
||
|
|
$instance->ttl = $settings['cache_ttl'] ?? $instance->ttl;
|
||
|
|
|
||
|
|
$instance->startCacheCleanupLoop();
|
||
|
|
|
||
|
|
return call(static function () use ($instance, $value, $settings) {
|
||
|
|
yield from $instance->initConnection($settings);
|
||
|
|
yield from $instance->prepareTable();
|
||
|
|
|
||
|
|
// Skip migrations if its same object
|
||
|
|
if ($instance !== $value) {
|
||
|
|
if ($value instanceof DriverArray) {
|
||
|
|
yield from $value->initConnection($value->settings);
|
||
|
|
}
|
||
|
|
yield from static::renameTmpTable($instance, $value);
|
||
|
|
yield from static::migrateDataToDb($instance, $value);
|
||
|
|
}
|
||
|
|
|
||
|
|
return $instance;
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Rename table of old database, if is a temporary table name.
|
||
|
|
*
|
||
|
|
* Otherwise, change name of table in new database to match old table name.
|
||
|
|
*
|
||
|
|
* @param self $new New db
|
||
|
|
* @param DbArray|array|null $old Old db
|
||
|
|
*
|
||
|
|
* @return \Generator
|
||
|
|
*/
|
||
|
|
protected static function renameTmpTable(self $new, $old): \Generator
|
||
|
|
{
|
||
|
|
if ($old instanceof static && $old->table) {
|
||
|
|
if (
|
||
|
|
$old->table !== $new->table &&
|
||
|
|
\mb_strpos($new->table, 'tmp') !== 0
|
||
|
|
) {
|
||
|
|
yield from $new->renameTable($old->table, $new->table);
|
||
|
|
} else {
|
||
|
|
$new->table = $old->table;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|