added generic repository

pkg/storage.go

@@ -0,0 +1,158 @@
+package models
+
+import (
+	"context"
+	"errors"
+	"log/slog"
+
+	"github.com/huandu/go-sqlbuilder"
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgconn"
+)
+
+type Querier interface {
+	Query(context.Context, string, ...any) (pgx.Rows, error)
+	QueryRow(context.Context, string, ...any) pgx.Row
+	Exec(context.Context, string, ...any) (pgconn.CommandTag, error)
+}
+
+var ErrNotFound = errors.New("not found")
+
+// GetAll функция для получения всех записей из базы данных
+func GetAll[T any](ctx context.Context, table string, db Querier, opts ...func(*sqlbuilder.SelectBuilder)) ([]T, error) {
+	sb := sqlbuilder.NewStruct(new(T)).SelectFrom(table)
+
+	sb.From(table)
+
+	sb.OrderByDesc("created_at")
+
+	for _, opt := range opts {
+		opt(sb)
+	}
+
+	query, args := sb.Build()
+
+	rows, err := db.Query(ctx, query, args...)
+	if err != nil {
+		slog.ErrorContext(ctx, "cannot execute get all query",
+			slog.String("query", query),
+			slog.Any("args", args),
+			slog.String("error", err.Error()),
+		)
+		return nil, err
+	}
+	defer rows.Close()
+
+	lists, err := pgx.CollectRows(rows, pgx.RowToStructByNameLax[T])
+	if err != nil {
+		slog.ErrorContext(ctx, "cannot scan courses",
+			slog.String("query", query),
+			slog.String("error", err.Error()),
+		)
+		return nil, err
+	}
+
+	return lists, nil
+}
+
+// GetOne функция для получения одной записи из базы данных
+func GetOne[T any](ctx context.Context, db Querier, table string, opts ...func(*sqlbuilder.SelectBuilder)) (*T, error) {
+	itemsStruct := sqlbuilder.NewStruct(new(T)).For(sqlbuilder.PostgreSQL)
+
+	sb := itemsStruct.SelectFrom(table)
+
+	for _, opt := range opts {
+		opt(sb)
+	}
+
+	query, args := sb.Build()
+
+	rows, err := db.Query(ctx, query, args...)
+	if err != nil {
+		slog.ErrorContext(ctx, "query failed", "query", query, "args", args, "err", err)
+		return nil, err
+	}
+	defer rows.Close()
+
+	item, err := pgx.CollectOneRow(rows, pgx.RowToStructByNameLax[T])
+	if err != nil {
+		if errors.Is(err, pgx.ErrNoRows) {
+			return nil, ErrNotFound
+		}
+		slog.ErrorContext(ctx, "cannot collect row", "query", query, "args", args, "err", err)
+		return nil, err
+	}
+
+	return &item, nil
+}
+
+// Create функция для создания записи в базе данных
+func Create[T any](ctx context.Context, table string, item T, db Querier, opts ...func(*sqlbuilder.SelectBuilder)) error {
+	structs := sqlbuilder.NewStruct(new(T))
+
+	slog.Info("Items", slog.Any("Item", item))
+
+	sb := structs.WithoutTag("db", "-").InsertInto(table, item)
+	sb.SetFlavor(sqlbuilder.PostgreSQL)
+
+	query, args := sb.Build()
+
+	slog.Info("Query", slog.String("query", query), slog.Any("args", args))
+
+	if _, err := db.Exec(ctx, query, args...); err != nil {
+		slog.ErrorContext(ctx, "cannot create item",
+			slog.String("query", query),
+			slog.Any("args", args),
+			slog.String("error", err.Error()),
+		)
+		return err
+	}
+
+	return nil
+}
+
+// Update функция для обновления записи в базе данных
+func Update[T any](ctx context.Context, table string, item T, db Querier, opts ...func(*sqlbuilder.UpdateBuilder)) error {
+	structs := sqlbuilder.NewStruct(new(T))
+
+	sb := structs.WithoutTag("db", "-").WithoutTag("immutable").Update(table, item)
+	sb.SetFlavor(sqlbuilder.PostgreSQL)
+
+	for _, opt := range opts {
+		opt(sb)
+	}
+
+	query, args := sb.Build()
+
+	if _, err := db.Exec(ctx, query, args...); err != nil {
+		slog.ErrorContext(ctx, "cannot update item",
+			slog.String("query", query),
+			slog.Any("args", args),
+			slog.String("error", err.Error()),
+		)
+		return err
+	}
+
+	return nil
+}
+
+// Delete функция для удаления записи из базы данных
+func Delete[T any](ctx context.Context, table string, db Querier, opts ...func(*sqlbuilder.SelectBuilder)) error {
+	structs := sqlbuilder.NewStruct(new(T))
+
+	sb := structs.WithoutTag("db", "-").DeleteFrom(table)
+	sb.SetFlavor(sqlbuilder.PostgreSQL)
+
+	query, args := sb.Build()
+
+	if _, err := db.Exec(ctx, query, args...); err != nil {
+		slog.ErrorContext(ctx, "cannot delete item",
+			slog.String("query", query),
+			slog.Any("args", args),
+			slog.String("error", err.Error()),
+		)
+		return err
+	}
+
+	return nil
+}