FixMountStorages.php 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. <?php
  2. /**
  3. * @copyright 2016 Joas Schilling <coding@schilljs.com>
  4. *
  5. * @author Joas Schilling <coding@schilljs.com>
  6. *
  7. * @license GNU AGPL version 3 or any later version
  8. *
  9. * This program is free software: you can redistribute it and/or modify
  10. * it under the terms of the GNU Affero General Public License as
  11. * published by the Free Software Foundation, either version 3 of the
  12. * License, or (at your option) any later version.
  13. *
  14. * This program is distributed in the hope that it will be useful,
  15. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17. * GNU Affero General Public License for more details.
  18. *
  19. * You should have received a copy of the GNU Affero General Public License
  20. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  21. *
  22. */
  23. namespace OC\Repair\NC11;
  24. use OCP\DB\QueryBuilder\IQueryBuilder;
  25. use OCP\IDBConnection;
  26. use OCP\Migration\IOutput;
  27. use OCP\Migration\IRepairStep;
  28. class FixMountStorages implements IRepairStep {
  29. /** @var IDBConnection */
  30. private $db;
  31. /**
  32. * @param IDBConnection $db
  33. */
  34. public function __construct(IDBConnection $db) {
  35. $this->db = $db;
  36. }
  37. /**
  38. * @return string
  39. */
  40. public function getName() {
  41. return 'Fix potential broken mount points';
  42. }
  43. public function run(IOutput $output) {
  44. $query = $this->db->getQueryBuilder();
  45. $query->select('m.id', 'f.storage')
  46. ->from('mounts', 'm')
  47. ->leftJoin('m', 'filecache', 'f', $query->expr()->eq('m.root_id', 'f.fileid'))
  48. ->where($query->expr()->neq('m.storage_id', 'f.storage'));
  49. $update = $this->db->getQueryBuilder();
  50. $update->update('mounts')
  51. ->set('storage_id', $update->createParameter('storage'))
  52. ->where($query->expr()->eq('id', $update->createParameter('mount')));
  53. $result = $query->execute();
  54. $entriesUpdated = 0;
  55. while ($row = $result->fetch()) {
  56. $update->setParameter('storage', $row['storage'], IQueryBuilder::PARAM_INT)
  57. ->setParameter('mount', $row['id'], IQueryBuilder::PARAM_INT);
  58. $update->execute();
  59. $entriesUpdated++;
  60. }
  61. $result->closeCursor();
  62. if ($entriesUpdated > 0) {
  63. $output->info($entriesUpdated . ' mounts updated');
  64. return;
  65. }
  66. $output->info('No mounts updated');
  67. }
  68. }