blob: 8ff952b8710e8d5175a606c74143482e060d3ccb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
<?php
/**
* SPDX-FileCopyrightText: 2017 Nextcloud GmbH and Nextcloud contributors
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
namespace OC\DB;
use Doctrine\DBAL\Exception;
use Doctrine\DBAL\Platforms\AbstractPlatform;
use Doctrine\DBAL\Schema\Schema;
use OCP\DB\ISchemaWrapper;
class SchemaWrapper implements ISchemaWrapper {
/** @var Connection */
protected $connection;
/** @var Schema */
protected $schema;
/** @var array */
protected $tablesToDelete = [];
public function __construct(Connection $connection) {
$this->connection = $connection;
$this->schema = $this->connection->createSchema();
}
public function getWrappedSchema() {
return $this->schema;
}
public function performDropTableCalls() {
foreach ($this->tablesToDelete as $tableName => $true) {
$this->connection->dropTable($tableName);
unset($this->tablesToDelete[$tableName]);
}
}
/**
* Gets all table names
*
* @return array
*/
public function getTableNamesWithoutPrefix() {
$tableNames = $this->schema->getTableNames();
return array_map(function ($tableName) {
if (str_starts_with($tableName, $this->connection->getPrefix())) {
return substr($tableName, strlen($this->connection->getPrefix()));
}
return $tableName;
}, $tableNames);
}
// Overwritten methods
/**
* @return array
*/
public function getTableNames() {
return $this->schema->getTableNames();
}
/**
* @param string $tableName
*
* @return \Doctrine\DBAL\Schema\Table
* @throws \Doctrine\DBAL\Schema\SchemaException
*/
public function getTable($tableName) {
return $this->schema->getTable($this->connection->getPrefix() . $tableName);
}
/**
* Does this schema have a table with the given name?
*
* @param string $tableName
*
* @return boolean
*/
public function hasTable($tableName) {
return $this->schema->hasTable($this->connection->getPrefix() . $tableName);
}
/**
* Creates a new table.
*
* @param string $tableName
* @return \Doctrine\DBAL\Schema\Table
*/
public function createTable($tableName) {
unset($this->tablesToDelete[$tableName]);
return $this->schema->createTable($this->connection->getPrefix() . $tableName);
}
/**
* Drops a table from the schema.
*
* @param string $tableName
* @return \Doctrine\DBAL\Schema\Schema
*/
public function dropTable($tableName) {
$this->tablesToDelete[$tableName] = true;
return $this->schema->dropTable($this->connection->getPrefix() . $tableName);
}
/**
* Gets all tables of this schema.
*
* @return \Doctrine\DBAL\Schema\Table[]
*/
public function getTables() {
return $this->schema->getTables();
}
/**
* Gets the DatabasePlatform for the database.
*
* @return AbstractPlatform
*
* @throws Exception
*/
public function getDatabasePlatform() {
return $this->connection->getDatabasePlatform();
}
}
|
