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
79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package core
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/core/catalogdb"
|
|
)
|
|
|
|
func (c *Catalog) CreateClass(namespaceOID int64, name string, kind string) (int64, error) {
|
|
oid, err := c.q.CreateClass(context.Background(), catalogdb.CreateClassParams{
|
|
NamespaceOid: namespaceOID,
|
|
Name: name,
|
|
Kind: kind,
|
|
})
|
|
if err != nil {
|
|
return 0, fmt.Errorf("create class %q: %w", name, err)
|
|
}
|
|
return oid, nil
|
|
}
|
|
|
|
func (c *Catalog) ClassOID(namespaceOID int64, name string) (int64, error) {
|
|
oid, err := c.q.ClassOID(context.Background(), catalogdb.ClassOIDParams{
|
|
NamespaceOid: namespaceOID,
|
|
Name: name,
|
|
})
|
|
if err != nil {
|
|
return 0, fmt.Errorf("class %q: %w", name, err)
|
|
}
|
|
return oid, nil
|
|
}
|
|
|
|
func (c *Catalog) ClassOIDByName(name string) (int64, error) {
|
|
oid, err := c.q.ClassOIDByName(context.Background(), name)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("class %q: %w", name, err)
|
|
}
|
|
return oid, nil
|
|
}
|
|
|
|
func (c *Catalog) DropClass(classOID int64) error {
|
|
ctx := context.Background()
|
|
if err := c.q.DeleteAttributesByClass(ctx, classOID); err != nil {
|
|
return fmt.Errorf("drop class %d attributes: %w", classOID, err)
|
|
}
|
|
if err := c.q.DeleteClass(ctx, classOID); err != nil {
|
|
return fmt.Errorf("drop class %d: %w", classOID, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Catalog) RenameClass(classOID int64, name string) error {
|
|
err := c.q.RenameClass(context.Background(), catalogdb.RenameClassParams{
|
|
Oid: classOID,
|
|
NewName: name,
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("rename class %d: %w", classOID, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type ClassInfo struct {
|
|
OID int64
|
|
Name string
|
|
}
|
|
|
|
func (c *Catalog) TablesInNamespace(namespaceOID int64) ([]ClassInfo, error) {
|
|
rows, err := c.q.ListTablesInNamespace(context.Background(), namespaceOID)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("list tables in namespace %d: %w", namespaceOID, err)
|
|
}
|
|
out := make([]ClassInfo, 0, len(rows))
|
|
for _, r := range rows {
|
|
out = append(out, ClassInfo{OID: r.Oid, Name: r.Name})
|
|
}
|
|
return out, nil
|
|
}
|