Add Debug flag

gosql/common/dbmethods.go

@@ -12,6 +12,7 @@ import (
 type DBMethods struct {
 	DB *sql.DB
 
+	Debug  bool
 	Driver string
 }
 

gosql/engine/engine.go

@@ -12,7 +12,7 @@ type mysql struct {
 	*common.DBMethods
 }
 
-func NewMySQL(dbURL *url.URL, migrationsDir string) (common.Engine, error) {
+func NewMySQL(dbURL *url.URL, migrationsDir string, debug bool) (common.Engine, error) {
 	db, err := common.OpenDB(dbURL, migrationsDir)
 	if err != nil {
 		return nil, err
@@ -20,7 +20,9 @@ func NewMySQL(dbURL *url.URL, migrationsDir string) (common.Engine, error) {
 
 	return &mysql{
 		DBMethods: &common.DBMethods{
-			DB: db, Driver: dbURL.Scheme,
+			DB:     db,
+			Debug:  debug,
+			Driver: dbURL.Scheme,
 		},
 	}, nil
 }
@@ -31,7 +33,7 @@ type postgresql struct {
 	*common.DBMethods
 }
 
-func NewPostgreSQL(dbURL *url.URL, migrationsDir string) (common.Engine, error) {
+func NewPostgreSQL(dbURL *url.URL, migrationsDir string, debug bool) (common.Engine, error) {
 	db, err := common.OpenDB(dbURL, migrationsDir)
 	if err != nil {
 		return nil, err
@@ -39,7 +41,9 @@ func NewPostgreSQL(dbURL *url.URL, migrationsDir string) (common.Engine, error)
 
 	return &postgresql{
 		DBMethods: &common.DBMethods{
-			DB: db, Driver: dbURL.Scheme,
+			DB:     db,
+			Debug:  debug,
+			Driver: dbURL.Scheme,
 		},
 	}, nil
 }
@@ -50,7 +54,7 @@ type sqlite struct {
 	*common.DBMethods
 }
 
-func NewSQLite(dbURL *url.URL, migrationsDir string) (common.Engine, error) {
+func NewSQLite(dbURL *url.URL, migrationsDir string, debug bool) (common.Engine, error) {
 	db, err := common.OpenDB(dbURL, migrationsDir)
 	if err != nil {
 		return nil, err
@@ -58,7 +62,9 @@ func NewSQLite(dbURL *url.URL, migrationsDir string) (common.Engine, error) {
 
 	return &sqlite{
 		DBMethods: &common.DBMethods{
-			DB: db, Driver: dbURL.Scheme,
+			DB:     db,
+			Debug:  debug,
+			Driver: dbURL.Scheme,
 		},
 	}, nil
 }

gosql/gosql.go

@@ -7,7 +7,7 @@ import (
 	"github.com/vladimirok5959/golang-sql/gosql/engine"
 )
 
-func Open(dbURL, migrationsDir string) (common.Engine, error) {
+func Open(dbURL, migrationsDir string, debug bool) (common.Engine, error) {
 	databaseURL, err := common.ParseUrl(dbURL)
 	if err != nil {
 		return nil, err
@@ -15,11 +15,11 @@ func Open(dbURL, migrationsDir string) (common.Engine, error) {
 
 	switch databaseURL.Scheme {
 	case "mysql":
-		return engine.NewMySQL(databaseURL, migrationsDir)
+		return engine.NewMySQL(databaseURL, migrationsDir, debug)
 	case "postgres", "postgresql":
-		return engine.NewPostgreSQL(databaseURL, migrationsDir)
+		return engine.NewPostgreSQL(databaseURL, migrationsDir, debug)
 	case "sqlite", "sqlite3":
-		return engine.NewSQLite(databaseURL, migrationsDir)
+		return engine.NewSQLite(databaseURL, migrationsDir, debug)
 	default:
 		return nil, fmt.Errorf("DB open error")
 	}

gosql/gosql_test.go

@@ -31,7 +31,7 @@ var _ = Describe("gosql", func() {
 		// // Note: you need to up MySQL server for this test case
 		// Context("for MySQL", func() {
 		// 	It("open connection, migrate and select data", func() {
-		// 		db, err := gosql.Open("mysql://root:root@127.0.0.1:3306/gosql", migrationsDir)
+		// 		db, err := gosql.Open("mysql://root:root@127.0.0.1:3306/gosql", migrationsDir, false)
 		// 		Expect(err).To(Succeed())
 
 		// 		err = db.QueryRow(ctx, sql, 1).Scan(&id, &name)
@@ -51,7 +51,7 @@ var _ = Describe("gosql", func() {
 		// // Note: you need to up PostgreSQL server for this test case
 		// Context("for PostgreSQL", func() {
 		// 	It("open connection, migrate and select data", func() {
-		// 		db, err := gosql.Open("postgres://root:root@127.0.0.1:5432/gosql?sslmode=disable", migrationsDir)
+		// 		db, err := gosql.Open("postgres://root:root@127.0.0.1:5432/gosql?sslmode=disable", migrationsDir, false)
 		// 		Expect(err).To(Succeed())
 
 		// 		err = db.QueryRow(ctx, sql, 1).Scan(&id, &name)
@@ -74,7 +74,7 @@ var _ = Describe("gosql", func() {
 				Expect(err).To(Succeed())
 				f.Close()
 
-				db, err := gosql.Open("sqlite://"+f.Name(), migrationsDir)
+				db, err := gosql.Open("sqlite://"+f.Name(), migrationsDir, false)
 				Expect(err).To(Succeed())
 
 				err = db.QueryRow(ctx, sql, 1).Scan(&id, &name)