init

internal/database/sql/database.go

@@ -0,0 +1,55 @@
+// Package sql implements the SQL storage layer.
+package sql
+
+import (
+	"context"
+
+	"emperror.dev/errors"
+	"github.com/jackc/pgx/v5/pgxpool"
+	"github.com/oklog/ulid/v2"
+)
+
+// Base is the base database pool used by storage layers.
+type Base struct {
+	pool *pgxpool.Pool
+}
+
+// NewBase creates a new instance of Base with the specified connection string.
+func NewBase(ctx context.Context, connString string) (*Base, error) {
+	pool, err := pgxpool.New(ctx, connString)
+	if err != nil {
+		return nil, errors.Wrap(err, "creating pool")
+	}
+
+	base := &Base{
+		pool: pool,
+	}
+	return base, nil
+}
+
+// Acquire acquires a connection from the database pool.
+// It is the caller's responsibility to call the Release method.
+func (base *Base) Acquire(ctx context.Context) (ReleaseableQuerier, error) {
+	conn, err := base.pool.Acquire(ctx)
+	if err != nil {
+		return nil, errors.Wrap(err, "acquiring connection")
+	}
+	return conn, nil
+}
+
+func (base *Base) BeginTx(ctx context.Context) (Tx, error) {
+	tx, err := base.pool.Begin(ctx)
+	if err != nil {
+		return nil, errors.Wrap(err, "beginning transaction")
+	}
+
+	return tx, nil
+}
+
+func (base *Base) PoolQuerier() Querier {
+	return base.pool
+}
+
+func makeULID() ulid.ULID {
+	return ulid.Make()
+}