FixMountStorages.php 2.3 KB

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