Initial commit

driver/shared/shared.go

@@ -0,0 +1,21 @@
+package shared
+
+import (
+	"database/sql"
+	"net/url"
+)
+
+// DatabaseName returns the database name from a URL
+func DatabaseName(u *url.URL) string {
+	name := u.Path
+	if len(name) > 0 && name[:1] == "/" {
+		name = name[1:len(name)]
+	}
+
+	return name
+}
+
+// Transaction can represent a database or open transaction
+type Transaction interface {
+	Exec(query string, args ...interface{}) (sql.Result, error)
+}

driver/shared/shared_test.go

@@ -0,0 +1,24 @@
+package shared_test
+
+import (
+	"github.com/adrianmacneil/dbmate/driver/shared"
+	"github.com/stretchr/testify/require"
+	"net/url"
+	"testing"
+)
+
+func TestDatabaseName(t *testing.T) {
+	u, err := url.Parse("ignore://localhost/foo?query")
+	require.Nil(t, err)
+
+	name := shared.DatabaseName(u)
+	require.Equal(t, "foo", name)
+}
+
+func TestDatabaseName_Empty(t *testing.T) {
+	u, err := url.Parse("ignore://localhost")
+	require.Nil(t, err)
+
+	name := shared.DatabaseName(u)
+	require.Equal(t, "", name)
+}