[go] refactoring

1 files changed, 19 insertions, 0 deletions

diff --git a/golang/pkg/database/db.go b/golang/pkg/database/db.go
new file mode 100644
index 0000000..3e07db3
--- /dev/null
+++ b/golang/pkg/database/db.go
@@ -0,0 +1,19 @@
+package database
+
+import (
+	"context"
+	"database/sql"
+)
+
+type DB struct {
+	ctx context.Context
+	db  *sql.DB
+}
+
+func (db *DB) Exec(query string, args ...any) (sql.Result, error) {
+	return db.db.ExecContext(db.ctx, query, args...)
+}
+
+func (db *DB) QueryRow(query string, args ...any) *sql.Row {
+	return db.db.QueryRowContext(db.ctx, query, args...)
+}