* @author Alexander Pankratov * @copyright 2016-2024 Daniil Gentili * @copyright 2016-2024 Alexander Pankratov * @license https://opensource.org/license/apache-2-0 Apache 2.0 * @link https://github.com/danog/AsyncOrm AsyncOrm documentation */ namespace danog\AsyncOrm\Driver; use danog\AsyncOrm\DbArray; use danog\AsyncOrm\DbArrayBuilder; use danog\AsyncOrm\Serializer; use danog\AsyncOrm\Settings\DriverSettings; use danog\AsyncOrm\Settings\SqlSettings; use function Amp\async; use function Amp\Future\await; /** * Base class for driver-based arrays. * * @template TKey as array-key * @template TValue * * @psalm-consistent-constructor * * @extends DbArray * * @api */ abstract class DriverArray extends DbArray { private bool $inited = false; /** * @param Serializer $serializer */ protected function __construct( protected readonly DbArrayBuilder $config, protected readonly Serializer $serializer ) { $this->inited = true; } /** * @template TTKey as array-key * @template TTValue * @param DbArray $previous * @return DbArray */ #[\Override] public static function getInstance(DbArrayBuilder $config, DbArray|null $previous): DbArray { $migrate = true; /** @psalm-suppress DocblockTypeContradiction TODO fix in psalm */ if ($previous !== null && $previous::class === static::class && $previous->config == $config ) { if ($previous->inited) { return $previous; } $migrate = false; } \assert($config->settings instanceof DriverSettings); /** @var DbArray */ $instance = new static($config, $config->settings->serializer); if ($previous === null || !$migrate) { return $instance; } /** @psalm-suppress DocblockTypeContradiction TODO fix in psalm */ if ($previous instanceof SqlArray && $instance instanceof SqlArray && $previous::class === $instance::class ) { \assert($config->settings instanceof SqlSettings); \assert($previous->config->settings instanceof SqlSettings); $c = $config->settings->config; $prevC = $config->settings->config; if ($c->getHost() === $prevC->getHost() && $c->getPort() === $prevC->getPort() && $c->getDatabase() === $prevC->getDatabase() ) { if ($config->table !== $previous->config->table) { $instance->importFromTable($previous->config->table); } // Only keyType/valueType changed, and we already migrated those on construction. return $instance; } } $promises = []; foreach ($previous->getIterator() as $key => $value) { $promises []= async($previous->unset(...), $key) ->map(static fn () => $instance->set($key, $value)); if (\count($promises) % 500 === 0) { // @codeCoverageIgnoreStart await($promises); $promises = []; // @codeCoverageIgnoreEnd } } if ($promises) { await($promises); } return $instance; } /** * Sleep function. */ public function __sleep(): array { return ['config']; } }