Merge pull request #1 from KSerik/feat/yugabyte-support

Feat/yugabyte support
KSerik · May 23, 2022 · 9d811eb · 9d811eb
2 parents 331a15d + 91d7dff
commit 9d811eb
Show file tree

Hide file tree

Showing 122 changed files with 863 additions and 193 deletions.
diff --git a/Makefile b/Makefile
@@ -1,5 +1,5 @@
 SOURCE ?= file go_bindata github github_ee bitbucket aws_s3 google_cloud_storage godoc_vfs gitlab
-DATABASE ?= postgres mysql redshift cassandra spanner cockroachdb clickhouse mongodb sqlserver firebird neo4j pgx
+DATABASE ?= postgres mysql redshift cassandra spanner cockroachdb yugabytedb clickhouse mongodb sqlserver firebird neo4j pgx
 DATABASE_TEST ?= $(DATABASE) sqlite sqlite3 sqlcipher
 VERSION ?= $(shell git describe --tags 2>/dev/null | cut -c 2-)
 TEST_FLAGS ?=

diff --git a/README.md b/README.md
@@ -38,6 +38,7 @@ Database drivers run migrations. [Add a new database?](database/driver.go)
 * [Shell](database/shell) ([todo #171](https://github.com/mattes/migrate/issues/171))
 * [Google Cloud Spanner](database/spanner)
 * [CockroachDB](database/cockroachdb)
+* [YugabyteDB](database/yugabytedb)
 * [ClickHouse](database/clickhouse)
 * [Firebird](database/firebird)
 * [MS SQL Server](database/sqlserver)

diff --git a/cli/main.go b/cli/main.go
@@ -1,6 +1,6 @@
 package main
 
-import "github.com/golang-migrate/migrate/v4/internal/cli"
+import "github.com/KSerik/migrate/internal/cli"
 
 // Deprecated, please use cmd/migrate
 func main() {

diff --git a/cmd/migrate/main.go b/cmd/migrate/main.go
@@ -1,6 +1,6 @@
 package main
 
-import "github.com/golang-migrate/migrate/v4/internal/cli"
+import "github.com/KSerik/migrate/internal/cli"
 
 func main() {
 	cli.Main(Version)

diff --git a/database/cassandra/cassandra.go b/database/cassandra/cassandra.go
@@ -11,9 +11,9 @@ import (
 	"strings"
 	"time"
 
+	"github.com/KSerik/migrate/database"
+	"github.com/KSerik/migrate/database/multistmt"
 	"github.com/gocql/gocql"
-	"github.com/golang-migrate/migrate/v4/database"
-	"github.com/golang-migrate/migrate/v4/database/multistmt"
 	"github.com/hashicorp/go-multierror"
 )
 

diff --git a/database/cassandra/cassandra_test.go b/database/cassandra/cassandra_test.go
@@ -3,7 +3,7 @@ package cassandra
 import (
 	"context"
 	"fmt"
-	"github.com/golang-migrate/migrate/v4"
+	"github.com/KSerik/migrate"
 	"strconv"
 	"testing"
 )
@@ -14,9 +14,9 @@ import (
 )
 
 import (
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 var (

diff --git a/database/clickhouse/clickhouse.go b/database/clickhouse/clickhouse.go
@@ -12,9 +12,9 @@ import (
 
 	"go.uber.org/atomic"
 
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database"
-	"github.com/golang-migrate/migrate/v4/database/multistmt"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database"
+	"github.com/KSerik/migrate/database/multistmt"
 	"github.com/hashicorp/go-multierror"
 )
 

diff --git a/database/clickhouse/clickhouse_test.go b/database/clickhouse/clickhouse_test.go
@@ -9,12 +9,12 @@ import (
 	"testing"
 
 	_ "github.com/ClickHouse/clickhouse-go"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database/clickhouse"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 	"github.com/dhui/dktest"
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database/clickhouse"
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
 )
 
 const defaultPort = 9000

diff --git a/database/cockroachdb/cockroachdb.go b/database/cockroachdb/cockroachdb.go
@@ -19,8 +19,8 @@ import (
 )
 
 import (
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database"
 )
 
 func init() {

diff --git a/database/cockroachdb/cockroachdb_test.go b/database/cockroachdb/cockroachdb_test.go
@@ -6,7 +6,7 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
-	"github.com/golang-migrate/migrate/v4"
+	"github.com/KSerik/migrate"
 	"log"
 	"strings"
 	"testing"
@@ -18,9 +18,9 @@ import (
 )
 
 import (
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 const defaultPort = 26257

diff --git a/database/driver.go b/database/driver.go
@@ -9,7 +9,7 @@ import (
 	"io"
 	"sync"
 
-	iurl "github.com/golang-migrate/migrate/v4/internal/url"
+	iurl "github.com/KSerik/migrate/internal/url"
 )
 
 var (

diff --git a/database/firebird/firebird.go b/database/firebird/firebird.go
@@ -7,8 +7,8 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database"
 	"github.com/hashicorp/go-multierror"
 	_ "github.com/nakagami/firebirdsql"
 	"go.uber.org/atomic"

diff --git a/database/firebird/firebird_test.go b/database/firebird/firebird_test.go
@@ -7,16 +7,16 @@ import (
 	"fmt"
 	"log"
 
-	"github.com/golang-migrate/migrate/v4"
+	"github.com/KSerik/migrate"
 	"io"
 	"strings"
 	"testing"
 
 	"github.com/dhui/dktest"
 
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 
 	_ "github.com/nakagami/firebirdsql"
 )

diff --git a/database/mongodb/mongodb.go b/database/mongodb/mongodb.go
@@ -3,8 +3,8 @@ package mongodb
 import (
 	"context"
 	"fmt"
+	"github.com/KSerik/migrate/database"
 	"github.com/cenkalti/backoff/v4"
-	"github.com/golang-migrate/migrate/v4/database"
 	"github.com/hashicorp/go-multierror"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"

diff --git a/database/mongodb/mongodb_test.go b/database/mongodb/mongodb_test.go
@@ -7,7 +7,7 @@ import (
 
 	"log"
 
-	"github.com/golang-migrate/migrate/v4"
+	"github.com/KSerik/migrate"
 	"io"
 	"os"
 	"strconv"
@@ -23,9 +23,9 @@ import (
 )
 
 import (
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 var (

diff --git a/database/multistmt/parse_test.go b/database/multistmt/parse_test.go
@@ -6,7 +6,7 @@ import (
 
 	"github.com/stretchr/testify/assert"
 
-	"github.com/golang-migrate/migrate/v4/database/multistmt"
+	"github.com/KSerik/migrate/database/multistmt"
 )
 
 const maxMigrationSize = 1024

diff --git a/database/mysql/mysql.go b/database/mysql/mysql.go
@@ -16,8 +16,8 @@ import (
 	"strconv"
 	"strings"
 
+	"github.com/KSerik/migrate/database"
 	"github.com/go-sql-driver/mysql"
-	"github.com/golang-migrate/migrate/v4/database"
 	"github.com/hashicorp/go-multierror"
 )
 

diff --git a/database/mysql/mysql_test.go b/database/mysql/mysql_test.go
@@ -26,10 +26,10 @@ import (
 )
 
 import (
-	"github.com/golang-migrate/migrate/v4"
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/KSerik/migrate"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 const defaultPort = 3306

diff --git a/database/neo4j/neo4j.go b/database/neo4j/neo4j.go
@@ -9,8 +9,8 @@ import (
 	"strconv"
 	"sync/atomic"
 
-	"github.com/golang-migrate/migrate/v4/database"
-	"github.com/golang-migrate/migrate/v4/database/multistmt"
+	"github.com/KSerik/migrate/database"
+	"github.com/KSerik/migrate/database/multistmt"
 	"github.com/hashicorp/go-multierror"
 	"github.com/neo4j/neo4j-go-driver/neo4j"
 )

diff --git a/database/neo4j/neo4j_test.go b/database/neo4j/neo4j_test.go
@@ -10,10 +10,10 @@ import (
 	"github.com/dhui/dktest"
 	"github.com/neo4j/neo4j-go-driver/neo4j"
 
-	"github.com/golang-migrate/migrate/v4"
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/KSerik/migrate"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 var (

diff --git a/database/pgx/pgx.go b/database/pgx/pgx.go
@@ -16,9 +16,9 @@ import (
 	"strings"
 	"time"
 
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database"
-	"github.com/golang-migrate/migrate/v4/database/multistmt"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database"
+	"github.com/KSerik/migrate/database/multistmt"
 	multierror "github.com/hashicorp/go-multierror"
 	"github.com/jackc/pgconn"
 	"github.com/jackc/pgerrcode"

diff --git a/database/pgx/pgx_test.go b/database/pgx/pgx_test.go
@@ -16,14 +16,14 @@ import (
 	"sync"
 	"testing"
 
-	"github.com/golang-migrate/migrate/v4"
+	"github.com/KSerik/migrate"
 
 	"github.com/dhui/dktest"
 
-	"github.com/golang-migrate/migrate/v4/database"
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/KSerik/migrate/database"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 const (

diff --git a/database/postgres/postgres.go b/database/postgres/postgres.go
@@ -17,9 +17,9 @@ import (
 
 	"go.uber.org/atomic"
 
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database"
-	"github.com/golang-migrate/migrate/v4/database/multistmt"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database"
+	"github.com/KSerik/migrate/database/multistmt"
 	"github.com/hashicorp/go-multierror"
 	"github.com/lib/pq"
 )

diff --git a/database/postgres/postgres_test.go b/database/postgres/postgres_test.go
@@ -15,14 +15,14 @@ import (
 	"sync"
 	"testing"
 
-	"github.com/golang-migrate/migrate/v4"
+	"github.com/KSerik/migrate"
 
 	"github.com/dhui/dktest"
 
-	"github.com/golang-migrate/migrate/v4/database"
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/KSerik/migrate/database"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 const (

diff --git a/database/ql/ql.go b/database/ql/ql.go
@@ -11,8 +11,8 @@ import (
 
 	nurl "net/url"
 
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database"
 	_ "modernc.org/ql/driver"
 )
 

diff --git a/database/ql/ql_test.go b/database/ql/ql_test.go
@@ -6,9 +6,9 @@ import (
 	"path/filepath"
 	"testing"
 
-	"github.com/golang-migrate/migrate/v4"
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/KSerik/migrate"
+	dt "github.com/KSerik/migrate/database/testing"
+	_ "github.com/KSerik/migrate/source/file"
 	_ "modernc.org/ql/driver"
 )
 

diff --git a/database/redshift/redshift.go b/database/redshift/redshift.go
@@ -14,8 +14,8 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/golang-migrate/migrate/v4"
-	"github.com/golang-migrate/migrate/v4/database"
+	"github.com/KSerik/migrate"
+	"github.com/KSerik/migrate/database"
 	"github.com/hashicorp/go-multierror"
 	"github.com/lib/pq"
 )

diff --git a/database/redshift/redshift_test.go b/database/redshift/redshift_test.go
@@ -10,7 +10,7 @@ import (
 	"fmt"
 	"log"
 
-	"github.com/golang-migrate/migrate/v4"
+	"github.com/KSerik/migrate"
 	"io"
 	"strconv"
 	"strings"
@@ -22,10 +22,10 @@ import (
 )
 
 import (
-	"github.com/golang-migrate/migrate/v4/database"
-	dt "github.com/golang-migrate/migrate/v4/database/testing"
-	"github.com/golang-migrate/migrate/v4/dktesting"
-	_ "github.com/golang-migrate/migrate/v4/source/file"
+	"github.com/KSerik/migrate/database"
+	dt "github.com/KSerik/migrate/database/testing"
+	"github.com/KSerik/migrate/dktesting"
+	_ "github.com/KSerik/migrate/source/file"
 )
 
 var (

diff --git a/database/snowflake/snowflake.go b/database/snowflake/snowflake.go
@@ -11,7 +11,7 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/golang-migrate/migrate/v4/database"
+	"github.com/KSerik/migrate/database"
 	"github.com/hashicorp/go-multierror"
 	"github.com/lib/pq"
 	sf "github.com/snowflakedb/gosnowflake"