Extend search path

xataio · Jul 31, 2024 · d1010d8 · d1010d8
1 parent 8ff9841
commit d1010d8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/roll/roll.go b/pkg/roll/roll.go
@@ -6,6 +6,7 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"strings"
 
 	"github.com/lib/pq"
 
@@ -78,7 +79,8 @@ func setupConn(ctx context.Context, pgURL, schema string, options options) (*sql
 		dsn = pgURL
 	}
 
-	dsn += " search_path=" + schema
+	searchPath := append([]string{schema}, options.searchPath...)
+	dsn += " search_path=" + strings.Join(searchPath, ",")
 
 	conn, err := sql.Open("postgres", dsn)
 	if err != nil {