[hw-4] add postgres db

loms/internal/domain/repository/orders/sqlc/db.go

@@ -0,0 +1,32 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.29.0
+
+package sqlc
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgconn"
+)
+
+type DBTX interface {
+	Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error)
+	Query(context.Context, string, ...interface{}) (pgx.Rows, error)
+	QueryRow(context.Context, string, ...interface{}) pgx.Row
+}
+
+func New(db DBTX) *Queries {
+	return &Queries{db: db}
+}
+
+type Queries struct {
+	db DBTX
+}
+
+func (q *Queries) WithTx(tx pgx.Tx) *Queries {
+	return &Queries{
+		db: tx,
+	}
+}

loms/internal/domain/repository/orders/sqlc/models.go

@@ -0,0 +1,5 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.29.0
+
+package sqlc

loms/internal/domain/repository/orders/sqlc/querier.go

@@ -0,0 +1,18 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.29.0
+
+package sqlc
+
+import (
+	"context"
+)
+
+type Querier interface {
+	OrderAddItem(ctx context.Context, arg *OrderAddItemParams) error
+	OrderCreate(ctx context.Context, arg *OrderCreateParams) (int64, error)
+	OrderGetByID(ctx context.Context, id int64) ([]*OrderGetByIDRow, error)
+	OrderSetStatus(ctx context.Context, arg *OrderSetStatusParams) (int64, error)
+}
+
+var _ Querier = (*Queries)(nil)

loms/internal/domain/repository/orders/sqlc/query.sql

@@ -0,0 +1,25 @@
+-- name: OrderCreate :one
+insert into orders (status_name, user_id)
+values ($1, $2)
+returning id;
+
+-- name: OrderAddItem :exec
+insert into order_items (order_id, sku, count)
+select $1, $2, $3;
+
+-- name: OrderSetStatus :execrows
+update orders
+set status_name = $2
+where id = $1;
+
+-- name: OrderGetByID :many
+select
+  o.id as order_id,
+  o.status_name as status,
+  o.user_id,
+  oi.sku,
+  oi.count
+from orders o
+left join order_items oi on oi.order_id = o.id
+where o.id = $1
+order by oi.id;

loms/internal/domain/repository/orders/sqlc/query.sql.go

@@ -0,0 +1,110 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.29.0
+// source: query.sql
+
+package sqlc
+
+import (
+	"context"
+)
+
+const orderAddItem = `-- name: OrderAddItem :exec
+insert into order_items (order_id, sku, count)
+select $1, $2, $3
+`
+
+type OrderAddItemParams struct {
+	OrderID int64
+	Sku     int64
+	Count   int64
+}
+
+func (q *Queries) OrderAddItem(ctx context.Context, arg *OrderAddItemParams) error {
+	_, err := q.db.Exec(ctx, orderAddItem, arg.OrderID, arg.Sku, arg.Count)
+	return err
+}
+
+const orderCreate = `-- name: OrderCreate :one
+insert into orders (status_name, user_id)
+values ($1, $2)
+returning id
+`
+
+type OrderCreateParams struct {
+	StatusName string
+	UserID     int64
+}
+
+func (q *Queries) OrderCreate(ctx context.Context, arg *OrderCreateParams) (int64, error) {
+	row := q.db.QueryRow(ctx, orderCreate, arg.StatusName, arg.UserID)
+	var id int64
+	err := row.Scan(&id)
+	return id, err
+}
+
+const orderGetByID = `-- name: OrderGetByID :many
+select
+  o.id as order_id,
+  o.status_name as status,
+  o.user_id,
+  oi.sku,
+  oi.count
+from orders o
+left join order_items oi on oi.order_id = o.id
+where o.id = $1
+order by oi.id
+`
+
+type OrderGetByIDRow struct {
+	OrderID int64
+	Status  string
+	UserID  int64
+	Sku     *int64
+	Count   *int64
+}
+
+func (q *Queries) OrderGetByID(ctx context.Context, id int64) ([]*OrderGetByIDRow, error) {
+	rows, err := q.db.Query(ctx, orderGetByID, id)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []*OrderGetByIDRow
+	for rows.Next() {
+		var i OrderGetByIDRow
+		if err := rows.Scan(
+			&i.OrderID,
+			&i.Status,
+			&i.UserID,
+			&i.Sku,
+			&i.Count,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, &i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
+const orderSetStatus = `-- name: OrderSetStatus :execrows
+update orders
+set status_name = $2
+where id = $1
+`
+
+type OrderSetStatusParams struct {
+	ID         int64
+	StatusName string
+}
+
+func (q *Queries) OrderSetStatus(ctx context.Context, arg *OrderSetStatusParams) (int64, error) {
+	result, err := q.db.Exec(ctx, orderSetStatus, arg.ID, arg.StatusName)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected(), nil
+}

loms/internal/domain/repository/orders/sqlc/repository.go

@@ -0,0 +1,105 @@
+package sqlc
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+
+	"route256/loms/internal/domain/entity"
+	"route256/loms/internal/domain/model"
+	"route256/loms/internal/domain/service"
+	"route256/loms/internal/infra/postgres"
+)
+
+type orderRepo struct {
+	pool *pgxpool.Pool
+}
+
+func NewOrderRepository(pool *pgxpool.Pool) service.OrderRepository {
+	return &orderRepo{
+		pool: pool,
+	}
+}
+
+func (o *orderRepo) GetQuerier(ctx context.Context) *Queries {
+	tx, ok := postgres.TxFromCtx(ctx)
+	if ok {
+		return New(tx)
+	}
+
+	return New(o.pool)
+}
+
+func (o *orderRepo) OrderCreate(ctx context.Context, order *entity.Order) (entity.ID, error) {
+	querier := o.GetQuerier(ctx)
+
+	id, err := querier.OrderCreate(ctx, &OrderCreateParams{
+		StatusName: order.Status,
+		UserID:     int64(order.UserID),
+	})
+	if err != nil {
+		return 0, fmt.Errorf("querier.OrderCreate: %w", err)
+	}
+
+	for _, item := range order.Items {
+		if err := querier.OrderAddItem(ctx, &OrderAddItemParams{
+			OrderID: id,
+			Sku:     int64(item.ID),
+			Count:   int64(item.Count),
+		}); err != nil {
+			return 0, fmt.Errorf("querier.OrderAddItem: %w", err)
+		}
+	}
+
+	return entity.ID(id), nil
+}
+
+func (o *orderRepo) OrderGetByID(ctx context.Context, orderID entity.ID) (*entity.Order, error) {
+	querier := o.GetQuerier(ctx)
+
+	rows, err := querier.OrderGetByID(ctx, int64(orderID))
+	if err != nil {
+		return nil, fmt.Errorf("querier.OrderGetByID: %w", err)
+	}
+
+	if len(rows) == 0 {
+		return nil, model.ErrOrderNotFound
+	}
+
+	items := make([]entity.OrderItem, len(rows))
+	for i, row := range rows {
+		items[i] = entity.OrderItem{
+			ID: entity.Sku(*row.Sku),
+			//nolint:gosec // will not overflow, uint32 is stored as int64
+			Count: uint32(*row.Count),
+		}
+	}
+
+	order := &entity.Order{
+		OrderID: orderID,
+		Status:  rows[0].Status,
+		UserID:  entity.ID(rows[0].UserID),
+		Items:   items,
+	}
+
+	return order, nil
+}
+
+func (o *orderRepo) OrderSetStatus(ctx context.Context, orderID entity.ID, newStatus string) error {
+	querier := o.GetQuerier(ctx)
+
+	rows, err := querier.OrderSetStatus(ctx, &OrderSetStatusParams{
+		ID:         int64(orderID),
+		StatusName: newStatus,
+	})
+	if err != nil {
+		return fmt.Errorf("querier.OrderSetStatus: %w", err)
+	}
+
+	if rows == 0 {
+		return model.ErrOrderNotFound
+	}
+
+	return nil
+}