From 83f0ce5369be5dc5d382ab8adc9c642b8cb60afc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Vin=C3=ADcius=20Garcia?= Date: Sun, 14 Nov 2021 18:57:57 -0300 Subject: [PATCH] Add constructor to SQLAdapter struct --- adapters/kmysql/kmysql.go | 2 +- adapters/ksqlite3/ksqlite3.go | 2 +- adapters/ksqlserver/ksqlserver.go | 2 +- sql_adapter.go | 8 ++++++++ 4 files changed, 11 insertions(+), 3 deletions(-) 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...)