65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"github.com/robindittmar/dttmr-api/internal/domain"
|
|
)
|
|
|
|
type ListRepo struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewListRepo(db *sql.DB) *ListRepo {
|
|
return &ListRepo{db: db}
|
|
}
|
|
|
|
func (r *ListRepo) CreateList(ctx context.Context, name string, userIDs []string) (*domain.List, error) {
|
|
tx, err := r.db.BeginTx(ctx, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("begin transaction: %w", err)
|
|
}
|
|
defer func() {
|
|
err := tx.Rollback()
|
|
if err != nil {
|
|
slog.Error("Failed to rollback transaction", slog.Any("error", err))
|
|
}
|
|
}()
|
|
|
|
list := &domain.List{Name: name}
|
|
|
|
err = tx.QueryRowContext(ctx,
|
|
"INSERT INTO lists (name) VALUES ($1) RETURNING id, created_at",
|
|
name,
|
|
).Scan(&list.ID, &list.CreatedAt)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to insert list: %w", err)
|
|
}
|
|
|
|
stmt, err := tx.PrepareContext(ctx, "INSERT INTO list_users (list_id, user_id) VALUES ($1, $2)")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to prepare user/list association statement: %w", err)
|
|
}
|
|
defer func() {
|
|
err := stmt.Close()
|
|
if err != nil {
|
|
|
|
}
|
|
}()
|
|
|
|
for _, userID := range userIDs {
|
|
if _, err = stmt.ExecContext(ctx, list.ID, userID); err != nil {
|
|
return nil, fmt.Errorf("failed to insert user/list association: %w", err)
|
|
}
|
|
}
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
return nil, fmt.Errorf("commit transaction: %w", err)
|
|
}
|
|
|
|
return list, err
|
|
}
|