Skip to content

Commit b9f909d

Browse files
committed
Schema fixes
1 parent c4a3524 commit b9f909d

File tree

2 files changed

+26
-12
lines changed

2 files changed

+26
-12
lines changed

go/libraries/doltcore/sqle/database.go

Lines changed: 21 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1741,7 +1741,8 @@ func (db Database) GetViewDefinition(ctx *sql.Context, viewName string) (sql.Vie
17411741
return sql.ViewDefinition{Name: viewName, TextDefinition: blameViewTextDef, CreateViewStatement: fmt.Sprintf("CREATE VIEW `%s` AS %s", viewName, blameViewTextDef)}, true, nil
17421742
}
17431743

1744-
schTblHash, ok, err := root.GetTableHash(ctx, doltdb.TableName{Name: doltdb.GetSchemasTableName(), Schema: db.schemaName})
1744+
schemasTableName := getDoltSchemasTableName()
1745+
schTblHash, ok, err := root.GetTableHash(ctx, schemasTableName)
17451746
if err != nil {
17461747
return sql.ViewDefinition{}, false, err
17471748
}
@@ -1764,6 +1765,10 @@ func (db Database) GetViewDefinition(ctx *sql.Context, viewName string) (sql.Vie
17641765
}
17651766
}
17661767

1768+
if resolve.UseSearchPath {
1769+
db.schemaName = "dolt"
1770+
}
1771+
17671772
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
17681773
if err != nil {
17691774
return sql.ViewDefinition{}, false, err
@@ -1827,6 +1832,10 @@ func getViewDefinitionFromSchemaFragmentsOfView(ctx *sql.Context, tbl *WritableD
18271832

18281833
// AllViews implements sql.ViewDatabase
18291834
func (db Database) AllViews(ctx *sql.Context) ([]sql.ViewDefinition, error) {
1835+
if resolve.UseSearchPath {
1836+
db.schemaName = "dolt"
1837+
}
1838+
18301839
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
18311840
if err != nil {
18321841
return nil, err
@@ -1892,6 +1901,9 @@ func (db Database) GetTriggers(ctx *sql.Context) ([]sql.TriggerDefinition, error
18921901
dbState.SessionCache().CacheTriggers(key, triggers, db.schemaName)
18931902
}()
18941903

1904+
if resolve.UseSearchPath {
1905+
db.schemaName = "dolt"
1906+
}
18951907
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
18961908
if err != nil {
18971909
return nil, err
@@ -1945,6 +1957,9 @@ func (db Database) DropTrigger(ctx *sql.Context, name string) error {
19451957

19461958
// GetEvent implements sql.EventDatabase.
19471959
func (db Database) GetEvent(ctx *sql.Context, name string) (sql.EventDefinition, bool, error) {
1960+
if resolve.UseSearchPath {
1961+
db.schemaName = "dolt"
1962+
}
19481963
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
19491964
if err != nil {
19501965
return sql.EventDefinition{}, false, err
@@ -1977,6 +1992,9 @@ func (db Database) GetEvent(ctx *sql.Context, name string) (sql.EventDefinition,
19771992

19781993
// GetEvents implements sql.EventDatabase.
19791994
func (db Database) GetEvents(ctx *sql.Context) (events []sql.EventDefinition, token interface{}, err error) {
1995+
if resolve.UseSearchPath {
1996+
db.schemaName = "dolt"
1997+
}
19801998
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
19811999
if err != nil {
19822000
return nil, nil, err
@@ -2224,16 +2242,8 @@ func (db Database) dropFragFromSchemasTable(ctx *sql.Context, fragType, name str
22242242
return err
22252243
}
22262244

2227-
root, err := db.GetRoot(ctx)
2228-
if err != nil {
2229-
return err
2230-
}
2231-
if resolve.UseSearchPath && db.schemaName == "" {
2232-
schemaName, err := resolve.FirstExistingSchemaOnSearchPath(ctx, root)
2233-
if err != nil {
2234-
return err
2235-
}
2236-
db.schemaName = schemaName
2245+
if resolve.UseSearchPath {
2246+
db.schemaName = "dolt"
22372247
}
22382248

22392249
stbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())

go/libraries/doltcore/sqle/schema_table.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,7 @@ func getOrCreateDoltSchemasTable(ctx *sql.Context, db Database) (retTbl *Writabl
224224
}
225225

226226
tname := getDoltSchemasTableName()
227-
if resolve.UseSearchPath && db.schemaName == "" {
227+
if resolve.UseSearchPath {
228228
db.schemaName = "dolt"
229229
}
230230

@@ -332,6 +332,10 @@ func migrateOldSchemasTableToNew(ctx *sql.Context, db Database, schemasTable *Wr
332332
return nil, err
333333
}
334334

335+
if resolve.UseSearchPath {
336+
db.schemaName = "dolt"
337+
}
338+
335339
tbl, _, err := db.GetTableInsensitive(ctx, tname.Name)
336340
if err != nil {
337341
return nil, err

0 commit comments

Comments
 (0)