Merge pull request #6910 from MrPetovan/task/6778-storage-move-loop

6778 Alternative 2: Wrap storage move in a loop
This commit is contained in:
Michael Vogel 2019-03-20 15:31:38 +01:00 committed by GitHub
commit 591d7fcbff
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 108 additions and 98 deletions

View file

@ -1200,7 +1200,9 @@ function admin_page_site_post(App $a)
* @var $storagebackend \Friendica\Model\Storage\IStorage
*/
$storagebackend = Strings::escapeTags(trim(defaults($_POST, 'storagebackend', '')));
StorageManager::setBackend($storagebackend);
if (!StorageManager::setBackend($storagebackend)) {
info(L10n::t('Invalid storage backend setting value.'));
}
// save storage backend form
if (!is_null($storagebackend) && $storagebackend != "") {

View file

@ -28,9 +28,10 @@ Synopsis
Set current storage backend
name storage backend to use. see "list".
bin/console storage move [table]
bin/console storage move [table] [-n 5000]
Move stored data to current storage backend.
table one of "photo" or "attach". default to both
-n limit of processed entry batch size
HELP;
return $help;
}
@ -49,36 +50,36 @@ HELP;
return -1;
}
switch($this->args[0]) {
case 'list':
return $this->do_list();
break;
case 'set':
return $this->do_set();
break;
case 'move':
return $this->do_move();
break;
switch ($this->args[0]) {
case 'list':
return $this->doList();
break;
case 'set':
return $this->doSet();
break;
case 'move':
return $this->doMove();
break;
}
$this->out(sprintf('Invalid action "%s"', $this->args[0]));
return -1;
}
protected function do_list()
protected function doList()
{
$rowfmt = ' %-3s | %-20s';
$current = StorageManager::getBackend();
$this->out(sprintf($rowfmt, 'Sel', 'Name'));
$this->out('-----------------------');
$isregisterd = false;
foreach(StorageManager::listBackends() as $name => $class) {
foreach (StorageManager::listBackends() as $name => $class) {
$issel = ' ';
if ($current === $class) {
$issel = '*';
$isregisterd = true;
};
$this->out(sprintf($rowfmt, $issel , $name ));
$this->out(sprintf($rowfmt, $issel, $name));
}
if ($current === '') {
@ -92,7 +93,7 @@ HELP;
return 0;
}
protected function do_set()
protected function doSet()
{
if (count($this->args) !== 2) {
throw new CommandArgsException('Invalid arguments');
@ -106,11 +107,15 @@ HELP;
return -1;
}
StorageManager::setBackend($class);
if (!StorageManager::setBackend($class)) {
$this->out($class . ' is not a valid backend storage class.');
return -1;
}
return 0;
}
protected function do_move()
protected function doMove()
{
$tables = null;
if (count($this->args) < 1 || count($this->args) > 2) {
@ -126,7 +131,17 @@ HELP;
}
$current = StorageManager::getBackend();
$r = StorageManager::move($current, $tables);
$this->out(sprintf('Moved %d files', $r));
$total = 0;
do {
$moved = StorageManager::move($current, $tables, $this->getOption('n', 5000));
if ($moved) {
$this->out(date('[Y-m-d H:i:s] ') . sprintf('Moved %d files', $moved));
}
$total += $moved;
} while ($moved);
$this->out(sprintf(date('[Y-m-d H:i:s] ') . 'Moved %d files total', $total));
}
}

View file

@ -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);
}
}
@ -54,12 +54,18 @@ class StorageManager
* @brief Set current storage backend class
*
* @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;
}
/**
@ -105,20 +111,20 @@ class StorageManager
/**
* @brief Move resources to storage $dest
* @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 $dest Destination storage class name
* @param array|null $tables Tables to look in for resources. Optional, defaults to ['photo', 'attach']
*
* @throws \Exception
* @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($dest, $tables = null)
public static function move($destination, $tables = null, $limit = 5000)
{
if (is_null($dest) || empty($dest)) {
if (empty($destination)) {
throw new \Exception('Can\'t move to NULL storage backend');
}
@ -129,43 +135,42 @@ class StorageManager
$moved = 0;
foreach ($tables as $table) {
// Get the rows where backend class is not the destination backend class
$rr = DBA::select(
$resources = DBA::select(
$table,
['id', 'data', 'backend-class', 'backend-ref'],
['`backend-class` IS NULL or `backend-class` != ?' , $dest ]
['`backend-class` IS NULL or `backend-class` != ?', $destination],
['limit' => $limit]
);
if (DBA::isResult($rr)) {
while($r = DBA::fetch($rr)) {
$id = $r['id'];
$data = $r['data'];
/** @var IStorage $backendClass */
$backendClass = $r['backend-class'];
$backendRef = $r['backend-ref'];
if (!is_null($backendClass) && $backendClass !== '') {
Logger::log("get data from old backend " . $backendClass . " : " . $backendRef);
$data = $backendClass::get($backendRef);
}
Logger::log("save data to new backend " . $dest);
/** @var IStorage $dest */
$ref = $dest::put($data);
Logger::log("saved data as " . $ref);
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);
}
if ($ref !== '') {
Logger::log("update row");
$ru = DBA::update($table, ['backend-class' => $dest, 'backend-ref' => $ref, 'data' => ''], ['id' => $id]);
if ($ru) {
if (!is_null($backendClass) && $backendClass !== '') {
Logger::log("delete data from old backend " . $backendClass . " : " . $backendRef);
$backendClass::delete($backendRef);
}
$moved++;
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;

View file

@ -1484,7 +1484,7 @@ class DBA
}
$limit_string = '';
if (isset($params['limit']) && is_int($params['limit'])) {
if (isset($params['limit']) && is_numeric($params['limit'])) {
$limit_string = " LIMIT " . intval($params['limit']);
}

View file

@ -35,54 +35,42 @@ class CronJobs
Logger::log("Starting cronjob " . $command, Logger::DEBUG);
// Call possible post update functions
// see src/Database/PostUpdate.php for more details
if ($command == 'post_update') {
PostUpdate::update();
return;
}
switch($command) {
case 'post_update':
PostUpdate::update();
break;
// update nodeinfo data
if ($command == 'nodeinfo') {
nodeinfo_cron();
return;
}
case 'nodeinfo':
nodeinfo_cron();
break;
// Expire and remove user entries
if ($command == 'expire_and_remove_users') {
self::expireAndRemoveUsers();
return;
}
case 'expire_and_remove_users':
self::expireAndRemoveUsers();
break;
if ($command == 'update_contact_birthdays') {
Contact::updateBirthdays();
return;
}
case 'update_contact_birthdays':
Contact::updateBirthdays();
break;
if ($command == 'update_photo_albums') {
self::updatePhotoAlbums();
return;
}
case 'update_photo_albums':
self::updatePhotoAlbums();
break;
// Clear cache entries
if ($command == 'clear_cache') {
self::clearCache($a);
return;
}
case 'clear_cache':
self::clearCache($a);
break;
// Repair missing Diaspora values in contacts
if ($command == 'repair_diaspora') {
self::repairDiaspora($a);
return;
}
case 'repair_diaspora':
self::repairDiaspora($a);
break;
// Repair entries in the database
if ($command == 'repair_database') {
self::repairDatabase();
return;
}
case 'repair_database':
self::repairDatabase();
break;
Logger::log("Xronjob " . $command . " is unknown.", Logger::DEBUG);
default:
Logger::log("Xronjob " . $command . " is unknown.", Logger::DEBUG);
}
return;
}