diff --git a/adapters/kmysql/kmysql.go b/adapters/kmysql/kmysql.go index c712701..1794b36 100644 --- a/adapters/kmysql/kmysql.go +++ b/adapters/kmysql/kmysql.go @@ -29,5 +29,5 @@ func New( db.SetMaxOpenConns(config.MaxOpenConns) - return ksql.NewWithAdapter(ksql.SQLAdapter{DB: db}, "mysql") + return ksql.NewWithAdapter(ksql.NewSQLAdapter(db), "mysql") } diff --git a/adapters/ksqlite3/ksqlite3.go b/adapters/ksqlite3/ksqlite3.go index 3d66558..b52dc89 100644 --- a/adapters/ksqlite3/ksqlite3.go +++ b/adapters/ksqlite3/ksqlite3.go @@ -29,5 +29,5 @@ func New( db.SetMaxOpenConns(config.MaxOpenConns) - return ksql.NewWithAdapter(ksql.SQLAdapter{DB: db}, "sqlite3") + return ksql.NewWithAdapter(ksql.NewSQLAdapter(db), "sqlite3") } diff --git a/adapters/ksqlserver/ksqlserver.go b/adapters/ksqlserver/ksqlserver.go index cb7ce66..56f52e3 100644 --- a/adapters/ksqlserver/ksqlserver.go +++ b/adapters/ksqlserver/ksqlserver.go @@ -29,5 +29,5 @@ func New( db.SetMaxOpenConns(config.MaxOpenConns) - return ksql.NewWithAdapter(ksql.SQLAdapter{DB: db}, "sqlserver") + return ksql.NewWithAdapter(ksql.NewSQLAdapter(db), "sqlserver") } diff --git a/sql_adapter.go b/sql_adapter.go index b3fa8a1..7573a8b 100644 --- a/sql_adapter.go +++ b/sql_adapter.go @@ -12,6 +12,14 @@ type SQLAdapter struct { var _ DBAdapter = SQLAdapter{} +// NewSQLAdapter returns a new instance of SQLAdapter with +// the provided database instance. +func NewSQLAdapter(db *sql.DB) SQLAdapter { + return SQLAdapter{ + DB: db, + } +} + // ExecContext implements the DBAdapter interface func (s SQLAdapter) ExecContext(ctx context.Context, query string, args ...interface{}) (Result, error) { return s.DB.ExecContext(ctx, query, args...)