Merge pull request #914 from soyayaos/master

fix: postgresql search_path can be empty (#696)
golang-migrate · Apr 7, 2023 · 84009cf · 84009cf
2 parents e968557 + ddaf669
commit 84009cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/database/postgres/postgres.go b/database/postgres/postgres.go
@@ -90,16 +90,16 @@ func WithConnection(ctx context.Context, conn *sql.Conn, config *Config) (*Postg
 
 	if config.SchemaName == "" {
 		query := `SELECT CURRENT_SCHEMA()`
-		var schemaName string
+		var schemaName sql.NullString
 		if err := conn.QueryRowContext(ctx, query).Scan(&schemaName); err != nil {
 			return nil, &database.Error{OrigErr: err, Query: []byte(query)}
 		}
 
-		if len(schemaName) == 0 {
+		if !schemaName.Valid {
 			return nil, ErrNoSchema
 		}
 
-		config.SchemaName = schemaName
+		config.SchemaName = schemaName.String
 	}
 
 	if len(config.MigrationsTable) == 0 {