feat: add support for postgres

postgres/migration.go

@@ -0,0 +1,51 @@
+package postgres
+
+import (
+	"database/sql"
+	"strings"
+
+	"github.com/golang-migrate/migrate"
+	_postgres "github.com/golang-migrate/migrate/database/postgres"
+)
+
+type migration struct {
+	Migrate *migrate.Migrate
+}
+
+func (this *migration) Up() (error, bool) {
+	err := this.Migrate.Up()
+	if err != nil {
+		if err == migrate.ErrNoChange {
+			return nil, true
+		}
+		return err, false
+	}
+	return nil, true
+}
+
+func (this *migration) Down() (error, bool) {
+	err := this.Migrate.Down()
+	if err != nil {
+		return err, false
+	}
+	return nil, true
+}
+
+func runMigration(dbConn *sql.DB, migrationsFolderLocation string) (*migration, error) {
+	dataPath := []string{}
+	dataPath = append(dataPath, "file://")
+	dataPath = append(dataPath, migrationsFolderLocation)
+
+	pathToMigrate := strings.Join(dataPath, "")
+
+	driver, err := _postgres.WithInstance(dbConn, &_postgres.Config{})
+	if err != nil {
+		return nil, err
+	}
+
+	m, err := migrate.NewWithDatabaseInstance(pathToMigrate, postgres, driver)
+	if err != nil {
+		return nil, err
+	}
+	return &migration{Migrate: m}, nil
+}

postgres/suite.go

@@ -0,0 +1,55 @@
+package postgres
+
+import (
+	"database/sql"
+
+	driverSql "github.com/go-sql-driver/mysql"
+	_ "github.com/golang-migrate/migrate/source/file"
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/suite"
+)
+
+const postgres = "postgres"
+
+// PostgresSuite struct for MySQL Suite
+type PostgresSuite struct {
+	suite.Suite
+	DSN                     string
+	DBConn                  *sql.DB
+	Migration               *migration
+	MigrationLocationFolder string
+	DBName                  string
+}
+
+// SetupSuite setup at the beginning of test
+func (s *PostgresSuite) SetupSuite() {
+	DisableLogging()
+
+	var err error
+
+	s.DBConn, err = sql.Open(postgres, s.DSN)
+	err = s.DBConn.Ping()
+	require.NoError(s.T(), err)
+
+	s.Migration, err = runMigration(s.DBConn, s.MigrationLocationFolder)
+	require.NoError(s.T(), err)
+}
+
+// TearDownSuite teardown at the end of test
+func (s *PostgresSuite) TearDownSuite() {
+	err := s.DBConn.Close()
+	require.NoError(s.T(), err)
+}
+
+func DisableLogging() {
+	nopLogger := NopLogger{}
+	if err := driverSql.SetLogger(nopLogger); err != nil {
+		panic(err)
+	}
+
+}
+
+type NopLogger struct {
+}
+
+func (l NopLogger) Print(v ...interface{}) {}