refactor: merge some .go files (#131)

dynmgrm.go

@@ -5,7 +5,9 @@ package dynmgrm
 
 import (
 	"database/sql"
+	"errors"
 	"fmt"
+	"github.com/btnguyen2k/godynamo"
 	"gorm.io/gorm/migrator"
 	"strconv"
 	"strings"
@@ -278,3 +280,39 @@ func (dialector Dialector) Migrator(db *gorm.DB) gorm.Migrator {
 		},
 	}
 }
+
+// Translate it will translate the error to native gorm errors.
+func (dialector Dialector) Translate(err error) error {
+	switch {
+	case errors.Is(err, godynamo.ErrTxCommitting),
+		errors.Is(err, godynamo.ErrTxRollingBack),
+		errors.Is(err, godynamo.ErrInTx),
+		errors.Is(err, godynamo.ErrInvalidTxStage),
+		errors.Is(err, godynamo.ErrNoTx):
+		return gorm.ErrInvalidTransaction
+	}
+	return err
+}
+
+type dbOpener struct {
+	dsn        string
+	driverName string
+}
+
+func (o dbOpener) DSN() string {
+	return o.dsn
+}
+
+func (o dbOpener) DriverName() string {
+	return o.driverName
+}
+
+func (o dbOpener) Apply() (*sql.DB, error) {
+	return sql.Open(o.DriverName(), o.DSN())
+}
+
+type callbacksRegisterer struct{}
+
+func (c *callbacksRegisterer) Register(db *gorm.DB, config *callbacks.Config) {
+	callbacks.RegisterDefaultCallbacks(db, config)
+}

dynmgrm_test.go

@@ -3,6 +3,7 @@ package dynmgrm
 import (
 	"database/sql"
 	"errors"
+	"github.com/btnguyen2k/godynamo"
 	"github.com/miyamo2/dynmgrm/internal/mocks"
 	"go.uber.org/mock/gomock"
 	"reflect"
@@ -484,3 +485,50 @@ func TestDialector_Initialize(t *testing.T) {
 		})
 	}
 }
+
+func TestDialector_Translate(t *testing.T) {
+	type test struct {
+		args error
+		want error
+	}
+	errOther := errors.New("other")
+	tests := map[string]test{
+		"happy_path/ErrTxCommitting": {
+			args: godynamo.ErrInTx,
+			want: gorm.ErrInvalidTransaction,
+		},
+		"happy_path/ErrTxRollingBack": {
+			args: godynamo.ErrTxRollingBack,
+			want: gorm.ErrInvalidTransaction,
+		},
+		"happy_path/ErrInTx": {
+			args: godynamo.ErrInTx,
+			want: gorm.ErrInvalidTransaction,
+		},
+		"happy_path/ErrInvalidTxStage": {
+			args: godynamo.ErrInvalidTxStage,
+			want: gorm.ErrInvalidTransaction,
+		},
+		"happy_path/ErrNoTx": {
+			args: godynamo.ErrNoTx,
+			want: gorm.ErrInvalidTransaction,
+		},
+		"happy_path/other": {
+			args: errOther,
+			want: errOther,
+		},
+		"happy_path/nil": {
+			args: nil,
+			want: nil,
+		},
+	}
+	for name, tt := range tests {
+		t.Run(name, func(t *testing.T) {
+			dialector := Dialector{}
+			err := dialector.Translate(tt.args)
+			if !errors.Is(err, tt.want) {
+				t.Errorf("Translate() error = %v, want %v", err, tt.want)
+			}
+		})
+	}
+}