BigIntMigration.php 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. <?php
  2. /**
  3. * @copyright Copyright (c) 2017 Joas Schilling <coding@schilljs.com>
  4. *
  5. * @author Daniel Kesselberg <mail@danielkesselberg.de>
  6. * @author Joas Schilling <coding@schilljs.com>
  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 OCP\Migration;
  25. use Doctrine\DBAL\Types\Type;
  26. use Doctrine\DBAL\Types\Types;
  27. use OCP\DB\ISchemaWrapper;
  28. /**
  29. * @since 13.0.0
  30. */
  31. abstract class BigIntMigration extends SimpleMigrationStep {
  32. /**
  33. * @return array Returns an array with the following structure
  34. * ['table1' => ['column1', 'column2'], ...]
  35. * @since 13.0.0
  36. */
  37. abstract protected function getColumnsByTable();
  38. /**
  39. * @param IOutput $output
  40. * @param \Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper`
  41. * @param array $options
  42. * @return null|ISchemaWrapper
  43. * @since 13.0.0
  44. */
  45. public function changeSchema(IOutput $output, \Closure $schemaClosure, array $options) {
  46. /** @var ISchemaWrapper $schema */
  47. $schema = $schemaClosure();
  48. $tables = $this->getColumnsByTable();
  49. foreach ($tables as $tableName => $columns) {
  50. $table = $schema->getTable($tableName);
  51. foreach ($columns as $columnName) {
  52. $column = $table->getColumn($columnName);
  53. if ($column->getType()->getName() !== Types::BIGINT) {
  54. $column->setType(Type::getType(Types::BIGINT));
  55. $column->setOptions(['length' => 20]);
  56. }
  57. }
  58. }
  59. return $schema;
  60. }
  61. }