Skip to content

Commit ec0bc7c

Browse files
committed
Schema fixes
1 parent 7a8d094 commit ec0bc7c

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
@@ -1770,7 +1770,8 @@ func (db Database) GetViewDefinition(ctx *sql.Context, viewName string) (sql.Vie
17701770
return sql.ViewDefinition{Name: viewName, TextDefinition: blameViewTextDef, CreateViewStatement: fmt.Sprintf("CREATE VIEW `%s` AS %s", viewName, blameViewTextDef)}, true, nil
17711771
}
17721772

1773-
schTblHash, ok, err := root.GetTableHash(ctx, doltdb.TableName{Name: doltdb.GetSchemasTableName(), Schema: db.schemaName})
1773+
schemasTableName := getDoltSchemasTableName()
1774+
schTblHash, ok, err := root.GetTableHash(ctx, schemasTableName)
17741775
if err != nil {
17751776
return sql.ViewDefinition{}, false, err
17761777
}
@@ -1793,6 +1794,10 @@ func (db Database) GetViewDefinition(ctx *sql.Context, viewName string) (sql.Vie
17931794
}
17941795
}
17951796

1797+
if resolve.UseSearchPath {
1798+
db.schemaName = "dolt"
1799+
}
1800+
17961801
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
17971802
if err != nil {
17981803
return sql.ViewDefinition{}, false, err
@@ -1856,6 +1861,10 @@ func getViewDefinitionFromSchemaFragmentsOfView(ctx *sql.Context, tbl *WritableD
18561861

18571862
// AllViews implements sql.ViewDatabase
18581863
func (db Database) AllViews(ctx *sql.Context) ([]sql.ViewDefinition, error) {
1864+
if resolve.UseSearchPath {
1865+
db.schemaName = "dolt"
1866+
}
1867+
18591868
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
18601869
if err != nil {
18611870
return nil, err
@@ -1921,6 +1930,9 @@ func (db Database) GetTriggers(ctx *sql.Context) ([]sql.TriggerDefinition, error
19211930
dbState.SessionCache().CacheTriggers(key, triggers, db.schemaName)
19221931
}()
19231932

1933+
if resolve.UseSearchPath {
1934+
db.schemaName = "dolt"
1935+
}
19241936
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
19251937
if err != nil {
19261938
return nil, err
@@ -1974,6 +1986,9 @@ func (db Database) DropTrigger(ctx *sql.Context, name string) error {
19741986

19751987
// GetEvent implements sql.EventDatabase.
19761988
func (db Database) GetEvent(ctx *sql.Context, name string) (sql.EventDefinition, bool, error) {
1989+
if resolve.UseSearchPath {
1990+
db.schemaName = "dolt"
1991+
}
19771992
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
19781993
if err != nil {
19791994
return sql.EventDefinition{}, false, err
@@ -2006,6 +2021,9 @@ func (db Database) GetEvent(ctx *sql.Context, name string) (sql.EventDefinition,
20062021

20072022
// GetEvents implements sql.EventDatabase.
20082023
func (db Database) GetEvents(ctx *sql.Context) (events []sql.EventDefinition, token interface{}, err error) {
2024+
if resolve.UseSearchPath {
2025+
db.schemaName = "dolt"
2026+
}
20092027
tbl, _, err := db.GetTableInsensitive(ctx, doltdb.GetSchemasTableName())
20102028
if err != nil {
20112029
return nil, nil, err
@@ -2253,16 +2271,8 @@ func (db Database) dropFragFromSchemasTable(ctx *sql.Context, fragType, name str
22532271
return err
22542272
}
22552273

2256-
root, err := db.GetRoot(ctx)
2257-
if err != nil {
2258-
return err
2259-
}
2260-
if resolve.UseSearchPath && db.schemaName == "" {
2261-
schemaName, err := resolve.FirstExistingSchemaOnSearchPath(ctx, root)
2262-
if err != nil {
2263-
return err
2264-
}
2265-
db.schemaName = schemaName
2274+
if resolve.UseSearchPath {
2275+
db.schemaName = "dolt"
22662276
}
22672277

22682278
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)