From 0234500a89eb2b2af226a2fcdf5fcc4fc491ec76 Mon Sep 17 00:00:00 2001
From: NiseVoid <nisevoid@gmail.com>
Date: Wed, 19 Apr 2017 14:45:11 +0200
Subject: [PATCH] Enforce schema if one is set

---
 migrate.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/migrate.go b/migrate.go
index fe7bf93..dea2aca 100644
--- a/migrate.go
+++ b/migrate.go
@@ -39,6 +39,17 @@ func Migrate(db *sql.DB, version int, o Options, asset AssetFunc) error {
 	table := o.TableName
 	if o.Schema != `` {
 		table = o.Schema + `.` + table
+
+		fmt.Println(`Switching to schema:`, o.Schema)
+		_, err := db.Exec(`CREATE SCHEMA IF NOT EXISTS ` + o.Schema)
+		if err != nil {
+			return err
+		}
+
+		_, err = db.Exec(`SET search_path TO ` + o.Schema + `,public`)
+		if err != nil {
+			return err
+		}
 	}
 
 	_, err := db.Exec(`CREATE TABLE IF NOT EXISTS ` + table + ` (Version integer NOT NULL PRIMARY KEY)`)