mirror/dolt
1
0
Fork 0
mirror of https://github.com/dolthub/dolt.git synced 2025-12-30 16:12:39 -06:00
Code Issues Packages Projects Releases 114 Wiki Activity

Move information_schema DatabaseSchema implementation to own file

Browse Source
This commit is contained in:
Taylor Bantle
2024-09-26 10:23:24 -07:00
parent f2a652a350
commit 6825812fc4
2 changed files with 71 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
go/libraries/doltcore/sqle/database.go
View File

@@ -28,7 +28,6 @@ import (
"github.com/dolthub/go-mysql-server/sql/analyzer/analyzererrors"
"github.com/dolthub/go-mysql-server/sql/expression"
"github.com/dolthub/go-mysql-server/sql/fulltext"
"github.com/dolthub/go-mysql-server/sql/information_schema"
"github.com/dolthub/go-mysql-server/sql/plan"
"github.com/dolthub/go-mysql-server/sql/planbuilder"
"github.com/dolthub/go-mysql-server/sql/rowexec"
@@ -1453,54 +1452,6 @@ func (db Database) AllSchemas(ctx *sql.Context) ([]sql.DatabaseSchema, error) {
return dbSchemas, nil
}
type informationSchemaDatabaseSchema struct {
name string
schemaName string
tables map[string]sql.Table
}
var _ sql.DatabaseSchema = (*informationSchemaDatabaseSchema)(nil)
// newInformationSchemaDatabase creates a new INFORMATION_SCHEMA DatabaseSchema.
func newInformationSchemaDatabase(dbName string) sql.DatabaseSchema {
isDb := &informationSchemaDatabaseSchema{
name: dbName,
schemaName: sql.InformationSchemaDatabaseName,
tables: information_schema.GetInformationSchemaTables(),
}
isDb.tables[information_schema.StatisticsTableName] = information_schema.NewDefaultStats()
return isDb
}
// Name implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) Name() string { return db.name }
// SchemaName implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) SchemaName() string { return db.schemaName }
// GetTableInsensitive implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) GetTableInsensitive(ctx *sql.Context, tblName string) (sql.Table, bool, error) {
// The columns table has dynamic information that can't be cached across queries
if strings.ToLower(tblName) == information_schema.ColumnsTableName {
return information_schema.NewColumnsTable(), true, nil
}
tbl, ok := sql.GetTableInsensitive(tblName, db.tables)
return tbl, ok, nil
}
// GetTableNames implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) GetTableNames(ctx *sql.Context) ([]string, error) {
tblNames := make([]string, 0, len(db.tables))
for k := range db.tables {
tblNames = append(tblNames, k)
}
return tblNames, nil
}
// RenameTable implements sql.TableRenamer
func (db Database) RenameTable(ctx *sql.Context, oldName, newName string) error {
if err := dsess.CheckAccessForDb(ctx, db, branch_control.Permissions_Write); err != nil {

71
go/libraries/doltcore/sqle/information_schema_database_schema.go Normal file
View File

@@ -0,0 +1,71 @@
// Copyright 2019-2020 Dolthub, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package sqle
import (
"strings"
"github.com/dolthub/go-mysql-server/sql"
"github.com/dolthub/go-mysql-server/sql/information_schema"
)
// informationSchemaDatabaseSchema is a DatabaseSchema implementation that provides access to the INFORMATION_SCHEMA tables. This is relevant only for Doltgres.
type informationSchemaDatabaseSchema struct {
name string
schemaName string
tables map[string]sql.Table
}
var _ sql.DatabaseSchema = (*informationSchemaDatabaseSchema)(nil)
// newInformationSchemaDatabase creates a new INFORMATION_SCHEMA DatabaseSchema for doltgres databases.
func newInformationSchemaDatabase(dbName string) sql.DatabaseSchema {
isDb := &informationSchemaDatabaseSchema{
name: dbName,
schemaName: sql.InformationSchemaDatabaseName,
tables: information_schema.GetInformationSchemaTables(),
}
isDb.tables[information_schema.StatisticsTableName] = information_schema.NewDefaultStats()
return isDb
}
// Name implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) Name() string { return db.name }
// SchemaName implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) SchemaName() string { return db.schemaName }
// GetTableInsensitive implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) GetTableInsensitive(ctx *sql.Context, tblName string) (sql.Table, bool, error) {
// The columns table has dynamic information that can't be cached across queries
if strings.ToLower(tblName) == information_schema.ColumnsTableName {
return information_schema.NewColumnsTable(), true, nil
}
tbl, ok := sql.GetTableInsensitive(tblName, db.tables)
return tbl, ok, nil
}
// GetTableNames implements the sql.DatabaseSchema interface.
func (db *informationSchemaDatabaseSchema) GetTableNames(ctx *sql.Context) ([]string, error) {
tblNames := make([]string, 0, len(db.tables))
for k := range db.tables {
tblNames = append(tblNames, k)
}
return tblNames, nil
}