From b772658c0b0aca330713c68f15c832d956e84017 Mon Sep 17 00:00:00 2001 From: Andrew Thornton Date: Tue, 4 Jan 2022 18:34:57 +0000 Subject: [PATCH] On postgres put the schema in if it is set Signed-off-by: Andrew Thornton --- engine.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/engine.go b/engine.go index 51a11b35..77688a5c 100644 --- a/engine.go +++ b/engine.go @@ -489,6 +489,10 @@ func (engine *Engine) dumpTables(ctx context.Context, tables []*schemas.Table, w if _, err := io.WriteString(w, "SET sql_mode='NO_BACKSLASH_ESCAPES';\n"); err != nil { return err } + } else if dstDialect.URI().DBType == schemas.POSTGRES && engine.dialect.URI().Schema != "" { + if _, err := fmt.Fprintf(w, `SELECT set_config('search_path', '%s,' || current_setting('search_path'), false)`, strings.ReplaceAll(engine.dialect.URI().Schema, "'", "''")); err != nil { + return err + } } for i, table := range tables {