diff --git a/database/gdb/gdb_model_hook.go b/database/gdb/gdb_model_hook.go index 7c88428f7dc..cbdd43391cc 100644 --- a/database/gdb/gdb_model_hook.go +++ b/database/gdb/gdb_model_hook.go @@ -159,6 +159,10 @@ func (h *HookSelectInput) Next(ctx context.Context) (result Result, err error) { if err != nil { return } + h.Model.db.GetCore().schema = h.Schema + defer func() { + h.Model.db.GetCore().schema = h.originalSchemaName.String() + }() } return h.Model.db.DoSelect(ctx, h.link, toBeCommittedSql, h.Args...) } @@ -195,6 +199,10 @@ func (h *HookInsertInput) Next(ctx context.Context) (result sql.Result, err erro if err != nil { return } + h.Model.db.GetCore().schema = h.Schema + defer func() { + h.Model.db.GetCore().schema = h.originalSchemaName.String() + }() } return h.Model.db.DoInsert(ctx, h.link, h.Table, h.Data, h.Option) } @@ -238,6 +246,10 @@ func (h *HookUpdateInput) Next(ctx context.Context) (result sql.Result, err erro if err != nil { return } + h.Model.db.GetCore().schema = h.Schema + defer func() { + h.Model.db.GetCore().schema = h.originalSchemaName.String() + }() } return h.Model.db.DoUpdate(ctx, h.link, h.Table, h.Data, h.Condition, h.Args...) } @@ -281,6 +293,10 @@ func (h *HookDeleteInput) Next(ctx context.Context) (result sql.Result, err erro if err != nil { return } + h.Model.db.GetCore().schema = h.Schema + defer func() { + h.Model.db.GetCore().schema = h.originalSchemaName.String() + }() } return h.Model.db.DoDelete(ctx, h.link, h.Table, h.Condition, h.Args...) }