| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294 |
- <?php
- namespace Illuminate\Database\Schema;
- use Illuminate\Database\Concerns\ParsesSearchPath;
- use InvalidArgumentException;
- class PostgresBuilder extends Builder
- {
- use ParsesSearchPath {
- parseSearchPath as baseParseSearchPath;
- }
- /**
- * Create a database in the schema.
- *
- * @param string $name
- * @return bool
- */
- public function createDatabase($name)
- {
- return $this->connection->statement(
- $this->grammar->compileCreateDatabase($name, $this->connection)
- );
- }
- /**
- * Drop a database from the schema if the database exists.
- *
- * @param string $name
- * @return bool
- */
- public function dropDatabaseIfExists($name)
- {
- return $this->connection->statement(
- $this->grammar->compileDropDatabaseIfExists($name)
- );
- }
- /**
- * Determine if the given table exists.
- *
- * @param string $table
- * @return bool
- */
- public function hasTable($table)
- {
- [$schema, $table] = $this->parseSchemaAndTable($table);
- $table = $this->connection->getTablePrefix().$table;
- foreach ($this->getTables() as $value) {
- if (strtolower($table) === strtolower($value['name'])
- && strtolower($schema) === strtolower($value['schema'])) {
- return true;
- }
- }
- return false;
- }
- /**
- * Determine if the given view exists.
- *
- * @param string $view
- * @return bool
- */
- public function hasView($view)
- {
- [$schema, $view] = $this->parseSchemaAndTable($view);
- $view = $this->connection->getTablePrefix().$view;
- foreach ($this->getViews() as $value) {
- if (strtolower($view) === strtolower($value['name'])
- && strtolower($schema) === strtolower($value['schema'])) {
- return true;
- }
- }
- return false;
- }
- /**
- * Get the user-defined types that belong to the database.
- *
- * @return array
- */
- public function getTypes()
- {
- return $this->connection->getPostProcessor()->processTypes(
- $this->connection->selectFromWriteConnection($this->grammar->compileTypes())
- );
- }
- /**
- * Drop all tables from the database.
- *
- * @return void
- */
- public function dropAllTables()
- {
- $tables = [];
- $excludedTables = $this->grammar->escapeNames(
- $this->connection->getConfig('dont_drop') ?? ['spatial_ref_sys']
- );
- $schemas = $this->grammar->escapeNames($this->getSchemas());
- foreach ($this->getTables() as $table) {
- $qualifiedName = $table['schema'].'.'.$table['name'];
- if (empty(array_intersect($this->grammar->escapeNames([$table['name'], $qualifiedName]), $excludedTables))
- && in_array($this->grammar->escapeNames([$table['schema']])[0], $schemas)) {
- $tables[] = $qualifiedName;
- }
- }
- if (empty($tables)) {
- return;
- }
- $this->connection->statement(
- $this->grammar->compileDropAllTables($tables)
- );
- }
- /**
- * Drop all views from the database.
- *
- * @return void
- */
- public function dropAllViews()
- {
- $views = [];
- $schemas = $this->grammar->escapeNames($this->getSchemas());
- foreach ($this->getViews() as $view) {
- if (in_array($this->grammar->escapeNames([$view['schema']])[0], $schemas)) {
- $views[] = $view['schema'].'.'.$view['name'];
- }
- }
- if (empty($views)) {
- return;
- }
- $this->connection->statement(
- $this->grammar->compileDropAllViews($views)
- );
- }
- /**
- * Drop all types from the database.
- *
- * @return void
- */
- public function dropAllTypes()
- {
- $types = [];
- $domains = [];
- $schemas = $this->grammar->escapeNames($this->getSchemas());
- foreach ($this->getTypes() as $type) {
- if (! $type['implicit'] && in_array($this->grammar->escapeNames([$type['schema']])[0], $schemas)) {
- if ($type['type'] === 'domain') {
- $domains[] = $type['schema'].'.'.$type['name'];
- } else {
- $types[] = $type['schema'].'.'.$type['name'];
- }
- }
- }
- if (! empty($types)) {
- $this->connection->statement($this->grammar->compileDropAllTypes($types));
- }
- if (! empty($domains)) {
- $this->connection->statement($this->grammar->compileDropAllDomains($domains));
- }
- }
- /**
- * Get the columns for a given table.
- *
- * @param string $table
- * @return array
- */
- public function getColumns($table)
- {
- [$schema, $table] = $this->parseSchemaAndTable($table);
- $table = $this->connection->getTablePrefix().$table;
- $results = $this->connection->selectFromWriteConnection(
- $this->grammar->compileColumns($schema, $table)
- );
- return $this->connection->getPostProcessor()->processColumns($results);
- }
- /**
- * Get the indexes for a given table.
- *
- * @param string $table
- * @return array
- */
- public function getIndexes($table)
- {
- [$schema, $table] = $this->parseSchemaAndTable($table);
- $table = $this->connection->getTablePrefix().$table;
- return $this->connection->getPostProcessor()->processIndexes(
- $this->connection->selectFromWriteConnection($this->grammar->compileIndexes($schema, $table))
- );
- }
- /**
- * Get the foreign keys for a given table.
- *
- * @param string $table
- * @return array
- */
- public function getForeignKeys($table)
- {
- [$schema, $table] = $this->parseSchemaAndTable($table);
- $table = $this->connection->getTablePrefix().$table;
- return $this->connection->getPostProcessor()->processForeignKeys(
- $this->connection->selectFromWriteConnection($this->grammar->compileForeignKeys($schema, $table))
- );
- }
- /**
- * Get the schemas for the connection.
- *
- * @return array
- */
- protected function getSchemas()
- {
- return $this->parseSearchPath(
- $this->connection->getConfig('search_path') ?: $this->connection->getConfig('schema') ?: 'public'
- );
- }
- /**
- * Parse the database object reference and extract the schema and table.
- *
- * @param string $reference
- * @return array
- */
- protected function parseSchemaAndTable($reference)
- {
- $parts = explode('.', $reference);
- if (count($parts) > 2) {
- $database = $parts[0];
- throw new InvalidArgumentException("Using three-part reference is not supported, you may use `Schema::connection('$database')` instead.");
- }
- // We will use the default schema unless the schema has been specified in the
- // query. If the schema has been specified in the query then we can use it
- // instead of a default schema configured in the connection search path.
- $schema = $this->getSchemas()[0];
- if (count($parts) === 2) {
- $schema = $parts[0];
- array_shift($parts);
- }
- return [$schema, $parts[0]];
- }
- /**
- * Parse the "search_path" configuration value into an array.
- *
- * @param string|array|null $searchPath
- * @return array
- */
- protected function parseSearchPath($searchPath)
- {
- return array_map(function ($schema) {
- return $schema === '$user'
- ? $this->connection->getConfig('username')
- : $schema;
- }, $this->baseParseSearchPath($searchPath));
- }
- }
|