diff --git a/Makefile b/Makefile index 82842d2..3bc0de9 100644 --- a/Makefile +++ b/Makefile @@ -1,5 +1,5 @@ SOURCE ?= file go_bindata github aws_s3 google_cloud_storage godoc_vfs gitlab -DATABASE ?= postgres mysql redshift cassandra spanner cockroachdb clickhouse mongodb mssql +DATABASE ?= postgres mysql redshift cassandra spanner cockroachdb clickhouse mongodb sqlserver VERSION ?= $(shell git describe --tags 2>/dev/null | cut -c 2-) TEST_FLAGS ?= REPO_OWNER ?= $(shell cd .. && basename "$$(pwd)") diff --git a/README.md b/README.md index 763e585..0b38a5e 100644 --- a/README.md +++ b/README.md @@ -37,7 +37,7 @@ Database drivers run migrations. [Add a new database?](database/driver.go) * [CockroachDB](database/cockroachdb) * [ClickHouse](database/clickhouse) * [Firebird](database/firebird) ([todo #49](https://github.com/golang-migrate/migrate/issues/49)) - * [MS SQL Server](database/mssql) + * [MS SQL Server](database/sqlserver) ### Database URLs diff --git a/database/mssql/README.md b/database/sqlserver/README.md similarity index 100% rename from database/mssql/README.md rename to database/sqlserver/README.md diff --git a/database/mssql/examples/migrations/1085649617_create_users_table.down.sql b/database/sqlserver/examples/migrations/1085649617_create_users_table.down.sql similarity index 100% rename from database/mssql/examples/migrations/1085649617_create_users_table.down.sql rename to database/sqlserver/examples/migrations/1085649617_create_users_table.down.sql diff --git a/database/mssql/examples/migrations/1085649617_create_users_table.up.sql b/database/sqlserver/examples/migrations/1085649617_create_users_table.up.sql similarity index 100% rename from database/mssql/examples/migrations/1085649617_create_users_table.up.sql rename to database/sqlserver/examples/migrations/1085649617_create_users_table.up.sql diff --git a/database/mssql/examples/migrations/1185749658_add_city_to_users.down.sql b/database/sqlserver/examples/migrations/1185749658_add_city_to_users.down.sql similarity index 100% rename from database/mssql/examples/migrations/1185749658_add_city_to_users.down.sql rename to database/sqlserver/examples/migrations/1185749658_add_city_to_users.down.sql diff --git a/database/mssql/examples/migrations/1185749658_add_city_to_users.up.sql b/database/sqlserver/examples/migrations/1185749658_add_city_to_users.up.sql similarity index 100% rename from database/mssql/examples/migrations/1185749658_add_city_to_users.up.sql rename to database/sqlserver/examples/migrations/1185749658_add_city_to_users.up.sql diff --git a/database/mssql/examples/migrations/1285849751_add_index_on_user_emails.down.sql b/database/sqlserver/examples/migrations/1285849751_add_index_on_user_emails.down.sql similarity index 100% rename from database/mssql/examples/migrations/1285849751_add_index_on_user_emails.down.sql rename to database/sqlserver/examples/migrations/1285849751_add_index_on_user_emails.down.sql diff --git a/database/mssql/examples/migrations/1285849751_add_index_on_user_emails.up.sql b/database/sqlserver/examples/migrations/1285849751_add_index_on_user_emails.up.sql similarity index 100% rename from database/mssql/examples/migrations/1285849751_add_index_on_user_emails.up.sql rename to database/sqlserver/examples/migrations/1285849751_add_index_on_user_emails.up.sql diff --git a/database/mssql/examples/migrations/1385949617_create_books_table.down.sql b/database/sqlserver/examples/migrations/1385949617_create_books_table.down.sql similarity index 100% rename from database/mssql/examples/migrations/1385949617_create_books_table.down.sql rename to database/sqlserver/examples/migrations/1385949617_create_books_table.down.sql diff --git a/database/mssql/examples/migrations/1385949617_create_books_table.up.sql b/database/sqlserver/examples/migrations/1385949617_create_books_table.up.sql similarity index 100% rename from database/mssql/examples/migrations/1385949617_create_books_table.up.sql rename to database/sqlserver/examples/migrations/1385949617_create_books_table.up.sql diff --git a/database/mssql/examples/migrations/1485949617_create_movies_table.down.sql b/database/sqlserver/examples/migrations/1485949617_create_movies_table.down.sql similarity index 100% rename from database/mssql/examples/migrations/1485949617_create_movies_table.down.sql rename to database/sqlserver/examples/migrations/1485949617_create_movies_table.down.sql diff --git a/database/mssql/examples/migrations/1485949617_create_movies_table.up.sql b/database/sqlserver/examples/migrations/1485949617_create_movies_table.up.sql similarity index 100% rename from database/mssql/examples/migrations/1485949617_create_movies_table.up.sql rename to database/sqlserver/examples/migrations/1485949617_create_movies_table.up.sql diff --git a/database/mssql/examples/migrations/1585849751_just_a_comment.up.sql b/database/sqlserver/examples/migrations/1585849751_just_a_comment.up.sql similarity index 100% rename from database/mssql/examples/migrations/1585849751_just_a_comment.up.sql rename to database/sqlserver/examples/migrations/1585849751_just_a_comment.up.sql diff --git a/database/mssql/examples/migrations/1685849751_another_comment.up.sql b/database/sqlserver/examples/migrations/1685849751_another_comment.up.sql similarity index 100% rename from database/mssql/examples/migrations/1685849751_another_comment.up.sql rename to database/sqlserver/examples/migrations/1685849751_another_comment.up.sql diff --git a/database/mssql/examples/migrations/1785849751_another_comment.up.sql b/database/sqlserver/examples/migrations/1785849751_another_comment.up.sql similarity index 100% rename from database/mssql/examples/migrations/1785849751_another_comment.up.sql rename to database/sqlserver/examples/migrations/1785849751_another_comment.up.sql diff --git a/database/mssql/examples/migrations/1885849751_another_comment.up.sql b/database/sqlserver/examples/migrations/1885849751_another_comment.up.sql similarity index 100% rename from database/mssql/examples/migrations/1885849751_another_comment.up.sql rename to database/sqlserver/examples/migrations/1885849751_another_comment.up.sql diff --git a/database/mssql/mssql.go b/database/sqlserver/sqlserver.go similarity index 93% rename from database/mssql/mssql.go rename to database/sqlserver/sqlserver.go index d66adab..0294814 100644 --- a/database/mssql/mssql.go +++ b/database/sqlserver/sqlserver.go @@ -1,4 +1,4 @@ -package mssql +package sqlserver import ( "context" @@ -15,8 +15,7 @@ import ( ) func init() { - db := MSSQL{} - database.Register("sqlserver", &db) + database.Register("sqlserver", &SQLServer{}) } // DefaultMigrationsTable is the name of the migrations table in the database @@ -44,7 +43,7 @@ type Config struct { } // MSSQL connection -type MSSQL struct { +type SQLServer struct { // Locking and unlocking need to use the same connection conn *sql.Conn db *sql.DB @@ -100,7 +99,7 @@ func WithInstance(instance *sql.DB, config *Config) (database.Driver, error) { return nil, err } - ss := &MSSQL{ + ss := &SQLServer{ conn: conn, db: instance, config: config, @@ -114,7 +113,7 @@ func WithInstance(instance *sql.DB, config *Config) (database.Driver, error) { } // Open a connection to the database -func (ss *MSSQL) Open(url string) (database.Driver, error) { +func (ss *SQLServer) Open(url string) (database.Driver, error) { purl, err := nurl.Parse(url) if err != nil { return nil, err @@ -140,7 +139,7 @@ func (ss *MSSQL) Open(url string) (database.Driver, error) { } // Close the database connection -func (ss *MSSQL) Close() error { +func (ss *SQLServer) Close() error { connErr := ss.conn.Close() dbErr := ss.db.Close() if connErr != nil || dbErr != nil { @@ -150,7 +149,7 @@ func (ss *MSSQL) Close() error { } // Lock creates an advisory local on the database to prevent multiple migrations from running at the same time. -func (ss *MSSQL) Lock() error { +func (ss *SQLServer) Lock() error { if ss.isLocked { return database.ErrLocked } @@ -177,7 +176,7 @@ func (ss *MSSQL) Lock() error { } // Unlock froms the migration lock from the database -func (ss *MSSQL) Unlock() error { +func (ss *SQLServer) Unlock() error { if !ss.isLocked { return nil } @@ -198,7 +197,7 @@ func (ss *MSSQL) Unlock() error { } // Run the migrations for the database -func (ss *MSSQL) Run(migration io.Reader) error { +func (ss *SQLServer) Run(migration io.Reader) error { migr, err := ioutil.ReadAll(migration) if err != nil { return err @@ -221,7 +220,7 @@ func (ss *MSSQL) Run(migration io.Reader) error { } // SetVersion for the current database -func (ss *MSSQL) SetVersion(version int, dirty bool) error { +func (ss *SQLServer) SetVersion(version int, dirty bool) error { tx, err := ss.conn.BeginTx(context.Background(), &sql.TxOptions{}) if err != nil { @@ -258,7 +257,7 @@ func (ss *MSSQL) SetVersion(version int, dirty bool) error { } // Version of the current database state -func (ss *MSSQL) Version() (version int, dirty bool, err error) { +func (ss *SQLServer) Version() (version int, dirty bool, err error) { query := `SELECT TOP 1 version, dirty FROM "` + ss.config.MigrationsTable + `"` err = ss.conn.QueryRowContext(context.Background(), query).Scan(&version, &dirty) switch { @@ -275,7 +274,7 @@ func (ss *MSSQL) Version() (version int, dirty bool, err error) { } // Drop all tables from the database. -func (ss *MSSQL) Drop() error { +func (ss *SQLServer) Drop() error { // drop all referential integrity constraints query := ` @@ -309,7 +308,7 @@ func (ss *MSSQL) Drop() error { return nil } -func (ss *MSSQL) ensureVersionTable() (err error) { +func (ss *SQLServer) ensureVersionTable() (err error) { if err = ss.Lock(); err != nil { return err } diff --git a/database/mssql/mssql_test.go b/database/sqlserver/sqlserver_test.go old mode 100755 new mode 100644 similarity index 92% rename from database/mssql/mssql_test.go rename to database/sqlserver/sqlserver_test.go index d26341d..a4844ad --- a/database/mssql/mssql_test.go +++ b/database/sqlserver/sqlserver_test.go @@ -1,4 +1,4 @@ -package mssql +package sqlserver import ( "context" @@ -74,7 +74,7 @@ func Test(t *testing.T) { } addr := msConnectionString(ip, port) - p := &MSSQL{} + p := &SQLServer{} d, err := p.Open(addr) if err != nil { t.Fatalf("%v", err) @@ -98,7 +98,7 @@ func TestMigrate(t *testing.T) { } addr := msConnectionString(ip, port) - p := &MSSQL{} + p := &SQLServer{} d, err := p.Open(addr) if err != nil { t.Fatalf("%v", err) @@ -126,7 +126,7 @@ func TestMultiStatement(t *testing.T) { } addr := msConnectionString(ip, port) - ms := &MSSQL{} + ms := &SQLServer{} d, err := ms.Open(addr) if err != nil { t.Fatal(err) @@ -142,7 +142,7 @@ func TestMultiStatement(t *testing.T) { // make sure second table exists var exists int - if err := d.(*MSSQL).conn.QueryRowContext(context.Background(), "SELECT COUNT(1) FROM information_schema.tables WHERE table_name = 'bar' AND table_schema = (SELECT schema_name()) AND table_catalog = (SELECT db_name())").Scan(&exists); err != nil { + if err := d.(*SQLServer).conn.QueryRowContext(context.Background(), "SELECT COUNT(1) FROM information_schema.tables WHERE table_name = 'bar' AND table_schema = (SELECT schema_name()) AND table_catalog = (SELECT db_name())").Scan(&exists); err != nil { t.Fatal(err) } if exists != 1 { @@ -159,7 +159,7 @@ func TestErrorParsing(t *testing.T) { } addr := msConnectionString(ip, port) - p := &MSSQL{} + p := &SQLServer{} d, err := p.Open(addr) if err != nil { t.Fatal(err) @@ -189,14 +189,14 @@ func TestLockWorks(t *testing.T) { } addr := fmt.Sprintf("sqlserver://sa:%v@%v:%v?master", saPassword, ip, port) - p := &MSSQL{} + p := &SQLServer{} d, err := p.Open(addr) if err != nil { t.Fatalf("%v", err) } dt.Test(t, d, []byte("SELECT 1")) - ms := d.(*MSSQL) + ms := d.(*SQLServer) err = ms.Lock() if err != nil { diff --git a/internal/cli/build_mssql.go b/internal/cli/build_mssql.go deleted file mode 100644 index 7d8f83e..0000000 --- a/internal/cli/build_mssql.go +++ /dev/null @@ -1,7 +0,0 @@ -// +build mssql - -package cli - -import ( - _ "github.com/golang-migrate/migrate/v4/database/mssql" -) diff --git a/internal/cli/build_sqlserver.go b/internal/cli/build_sqlserver.go new file mode 100644 index 0000000..c852f17 --- /dev/null +++ b/internal/cli/build_sqlserver.go @@ -0,0 +1,7 @@ +// +build sqlserver + +package cli + +import ( + _ "github.com/golang-migrate/migrate/v4/database/sqlserver" +)