X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=src%2FCore%2FStorageManager.php;h=0a8b35ce24a844a62251db06735363fa5e1b7513;hb=c713c2bf622257dbecf223b5f58bf8a98dde9d65;hp=5c14fdbf3388646c954441b30d8aadce96ebe753;hpb=4d9d62bcdbcbccb72562f58332f230df70ead796;p=friendica.git diff --git a/src/Core/StorageManager.php b/src/Core/StorageManager.php index 5c14fdbf33..0a8b35ce24 100644 --- a/src/Core/StorageManager.php +++ b/src/Core/StorageManager.php @@ -2,8 +2,8 @@ namespace Friendica\Core; -use Friendica\Core\Config; - +use Friendica\Database\DBA; +use Friendica\Model\Storage\IStorage; /** @@ -23,7 +23,7 @@ class StorageManager private static function setup() { - if (count(self::$backends)==0) { + if (count(self::$backends) == 0) { self::$backends = Config::get('storage', 'backends', self::$default_backends); } } @@ -31,6 +31,7 @@ class StorageManager /** * @brief Return current storage backend class * @return string + * @throws \Friendica\Network\HTTPException\InternalServerErrorException */ public static function getBackend() { @@ -52,12 +53,19 @@ class StorageManager /** * @brief Set current storage backend class * - * @param string $class Backend class name + * @param string $class Backend class name + * @return bool + * @throws \Friendica\Network\HTTPException\InternalServerErrorException */ public static function setBackend($class) { - /// @todo Check that $class implements IStorage + if (!in_array('Friendica\Model\Storage\IStorage', class_implements($class))) { + return false; + } + Config::set('storage', 'class', $class); + + return true; } /** @@ -72,12 +80,12 @@ class StorageManager } - /** * @brief Register a storage backend class * - * @param string $name User readable backend name - * @param string $class Backend class name + * @param string $name User readable backend name + * @param string $class Backend class name + * @throws \Friendica\Network\HTTPException\InternalServerErrorException */ public static function register($name, $class) { @@ -91,12 +99,81 @@ class StorageManager /** * @brief Unregister a storage backend class * - * @param string $name User readable backend name + * @param string $name User readable backend name + * @throws \Friendica\Network\HTTPException\InternalServerErrorException */ - public static function unregister($class) + public static function unregister($name) { self::setup(); unset(self::$backends[$name]); Config::set('storage', 'backends', self::$backends); } -} \ No newline at end of file + + + /** + * @brief Move up to 5000 resources to storage $dest + * + * Copy existing data to destination storage and delete from source. + * This method cannot move to legacy in-table `data` field. + * + * @param string $destination Storage class name + * @param array|null $tables Tables to look in for resources. Optional, defaults to ['photo', 'attach'] + * @param int $limit Limit of the process batch size, defaults to 5000 + * @return int Number of moved resources + * @throws \Exception + */ + public static function move($destination, $tables = null, $limit = 5000) + { + if (empty($destination)) { + throw new \Exception('Can\'t move to NULL storage backend'); + } + + if (is_null($tables)) { + $tables = ['photo', 'attach']; + } + + $moved = 0; + foreach ($tables as $table) { + // Get the rows where backend class is not the destination backend class + $resources = DBA::select( + $table, + ['id', 'data', 'backend-class', 'backend-ref'], + ['`backend-class` IS NULL or `backend-class` != ?', $destination], + ['limit' => $limit] + ); + + while ($resource = DBA::fetch($resources)) { + $id = $resource['id']; + $data = $resource['data']; + /** @var IStorage $backendClass */ + $backendClass = $resource['backend-class']; + $backendRef = $resource['backend-ref']; + if (!empty($backendClass)) { + Logger::log("get data from old backend " . $backendClass . " : " . $backendRef); + $data = $backendClass::get($backendRef); + } + + Logger::log("save data to new backend " . $destination); + /** @var IStorage $destination */ + $ref = $destination::put($data); + Logger::log("saved data as " . $ref); + + if ($ref !== '') { + Logger::log("update row"); + if (DBA::update($table, ['backend-class' => $destination, 'backend-ref' => $ref, 'data' => ''], ['id' => $id])) { + if (!empty($backendClass)) { + Logger::log("delete data from old backend " . $backendClass . " : " . $backendRef); + $backendClass::delete($backendRef); + } + $moved++; + } + } + } + + DBA::close($resources); + } + + return $moved; + } +} +