0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
143 lines
2.8 KiB
Go
143 lines
2.8 KiB
Go
package catalog
|
|
|
|
import (
|
|
"github.com/sqlc-dev/sqlc/internal/sql/ast"
|
|
)
|
|
|
|
// Catalog describes a database instance consisting of metadata in which database objects are defined
|
|
type Catalog struct {
|
|
Comment string
|
|
DefaultSchema string
|
|
Name string
|
|
Schemas []*Schema
|
|
SearchPath []string
|
|
LoadExtension func(string) *Schema
|
|
|
|
// TODO: un-export
|
|
Extensions map[string]struct{}
|
|
}
|
|
|
|
// New creates a new catalog
|
|
func New(defaultSchema string) *Catalog {
|
|
|
|
newCatalog := &Catalog{
|
|
DefaultSchema: defaultSchema,
|
|
Schemas: make([]*Schema, 0),
|
|
Extensions: make(map[string]struct{}),
|
|
}
|
|
|
|
if newCatalog.DefaultSchema != "" {
|
|
newCatalog.Schemas = append(newCatalog.Schemas, &Schema{Name: defaultSchema})
|
|
}
|
|
|
|
return newCatalog
|
|
}
|
|
|
|
func (c *Catalog) Build(stmts []ast.Statement) error {
|
|
for i := range stmts {
|
|
if err := c.Update(stmts[i], nil); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Catalog) Update(stmt ast.Statement, colGen columnGenerator) error {
|
|
if stmt.Raw == nil {
|
|
return nil
|
|
}
|
|
var err error
|
|
switch n := stmt.Raw.Stmt.(type) {
|
|
|
|
case *ast.AlterTableStmt:
|
|
err = c.alterTable(n)
|
|
|
|
case *ast.AlterTableSetSchemaStmt:
|
|
err = c.alterTableSetSchema(n)
|
|
|
|
case *ast.AlterTypeAddValueStmt:
|
|
err = c.alterTypeAddValue(n)
|
|
|
|
case *ast.AlterTypeRenameValueStmt:
|
|
err = c.alterTypeRenameValue(n)
|
|
|
|
case *ast.AlterTypeSetSchemaStmt:
|
|
err = c.alterTypeSetSchema(n)
|
|
|
|
case *ast.CommentOnColumnStmt:
|
|
err = c.commentOnColumn(n)
|
|
|
|
case *ast.CommentOnSchemaStmt:
|
|
err = c.commentOnSchema(n)
|
|
|
|
case *ast.CommentOnTableStmt:
|
|
err = c.commentOnTable(n)
|
|
|
|
case *ast.CommentOnTypeStmt:
|
|
err = c.commentOnType(n)
|
|
|
|
case *ast.CommentOnViewStmt:
|
|
err = c.commentOnView(n)
|
|
|
|
case *ast.CompositeTypeStmt:
|
|
err = c.createCompositeType(n)
|
|
|
|
case *ast.CreateEnumStmt:
|
|
err = c.createEnum(n)
|
|
|
|
case *ast.CreateExtensionStmt:
|
|
err = c.createExtension(n)
|
|
|
|
case *ast.CreateFunctionStmt:
|
|
err = c.createFunction(n)
|
|
|
|
case *ast.CreateSchemaStmt:
|
|
err = c.createSchema(n)
|
|
|
|
case *ast.CreateTableStmt:
|
|
err = c.createTable(n)
|
|
|
|
case *ast.CreateTableAsStmt:
|
|
err = c.createTableAs(n, colGen)
|
|
|
|
case *ast.ViewStmt:
|
|
err = c.createView(n, colGen)
|
|
|
|
case *ast.DropFunctionStmt:
|
|
err = c.dropFunction(n)
|
|
|
|
case *ast.DropSchemaStmt:
|
|
err = c.dropSchema(n)
|
|
|
|
case *ast.DropTableStmt:
|
|
err = c.dropTable(n)
|
|
|
|
case *ast.DropTypeStmt:
|
|
err = c.dropType(n)
|
|
|
|
case *ast.RenameColumnStmt:
|
|
err = c.renameColumn(n)
|
|
|
|
case *ast.RenameTableStmt:
|
|
err = c.renameTable(n)
|
|
|
|
case *ast.RenameTypeStmt:
|
|
err = c.renameType(n)
|
|
|
|
case *ast.List:
|
|
for _, nn := range n.Items {
|
|
if err = c.Update(ast.Statement{
|
|
Raw: &ast.RawStmt{
|
|
Stmt: nn,
|
|
StmtLocation: stmt.Raw.StmtLocation,
|
|
StmtLen: stmt.Raw.StmtLen,
|
|
},
|
|
}, colGen); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
}
|
|
return err
|
|
}
|