Postgres - Add x-schema-name URL Query support (golang-migrate#95)

spacefill · Nov 28, 2020 · f8f6366 · f8f6366
1 parent a53e6fc
commit f8f6366
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/database/postgres/README.md b/database/postgres/README.md
@@ -5,6 +5,7 @@
 | URL Query  | WithInstance Config | Description |
 |------------|---------------------|-------------|
 | `x-migrations-table` | `MigrationsTable` | Name of the migrations table |
+| `x-schema-name` | `SchemaName` | Schema name of the migratations table |
 | `x-statement-timeout` | `StatementTimeout` | Abort any statement that takes more than the specified number of milliseconds |
 | `dbname` | `DatabaseName` | The name of the database to connect to |
 | `search_path` | | This variable specifies the order in which schemas are searched when an object is referenced by a simple name with no schema specified. |

diff --git a/database/postgres/postgres.go b/database/postgres/postgres.go
@@ -123,6 +123,7 @@ func (p *Postgres) Open(url string) (database.Driver, error) {
 	}
 
 	migrationsTable := purl.Query().Get("x-migrations-table")
+	schemaName := purl.Query().Get("x-schema-name")
 	statementTimeoutString := purl.Query().Get("x-statement-timeout")
 	statementTimeout := 0
 	if statementTimeoutString != "" {
@@ -135,6 +136,7 @@ func (p *Postgres) Open(url string) (database.Driver, error) {
 	px, err := WithInstance(db, &Config{
 		DatabaseName:     purl.Path,
 		MigrationsTable:  migrationsTable,
+		SchemaName:       schemaName,
 		StatementTimeout: time.Duration(statementTimeout) * time.Millisecond,
 	})