Extend adapter interface with DBType function

go-rel · Oct 3, 2023 · 123b7ba · 123b7ba
1 parent 0384795
commit 123b7ba
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/adapter.go b/adapter.go
@@ -6,6 +6,7 @@ import (
 
 // Adapter interface
 type Adapter interface {
+	DBType() string
 	Close() error
 
 	Instrumentation(instrumenter Instrumenter)

diff --git a/adapter_test.go b/adapter_test.go
@@ -90,3 +90,8 @@ func (ta *testAdapter) Exec(ctx context.Context, stmt string, args []any) (int64
 	mockArgs := ta.Called(ctx, stmt, args)
 	return int64(mockArgs.Int(0)), int64(mockArgs.Int(1)), mockArgs.Error(2)
 }
+
+func (ta *testAdapter) DBType() string {
+	args := ta.Called()
+	return args.String(0)
+}