add game

internal/repos/actions_repo/repo.go

@@ -0,0 +1,103 @@
+package actions_repo
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+)
+
+type ActionsRepo struct {
+	pool *pgxpool.Pool
+}
+
+func NewActionsRepo(pool *pgxpool.Pool) *ActionsRepo {
+	return &ActionsRepo{
+		pool: pool,
+	}
+}
+
+func (r *ActionsRepo) AddAction(ctx context.Context, teamId int, code string) (int, error) {
+	id := 0
+	err := r.pool.QueryRow(
+		ctx,
+		`INSERT INTO actions (code, team_id)
+		VALUES ($1, $2)
+		RETURNING id`,
+		code,
+		teamId,
+	).Scan(&id)
+
+	if err != nil {
+		return 0, err
+	}
+	return id, nil
+}
+
+func (r *ActionsRepo) GetActionsByTeamID(ctx context.Context, teamId int) ([]string, error) {
+	rows, err := r.pool.Query(
+		ctx,
+		`SELECT 
+			code
+		FROM actions
+		WHERE team_id = $1
+		ORDER BY created_at ASC`,
+		teamId,
+	)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var codes []string
+	for rows.Next() {
+		code := ""
+		err := rows.Scan(
+			&code,
+		)
+		if err != nil {
+			return nil, err
+		}
+		codes = append(codes, code)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+
+	return codes, nil
+}
+
+func (r *ActionsRepo) GetActionsCountByTeamIDs(ctx context.Context, teamIds []int) (map[int]int, error) {
+	rows, err := r.pool.Query(
+		ctx,
+		`SELECT 
+			team_id, 
+			count(*)
+		FROM actions
+		WHERE team_id = ANY($1)
+		GROUP BY team_id`,
+		teamIds,
+	)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	res := make(map[int]int, len(teamIds))
+	for rows.Next() {
+		id := 0
+		count := 0
+		err := rows.Scan(
+			&id,
+			&count,
+		)
+		if err != nil {
+			return nil, err
+		}
+		res[id] = count
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+
+	return res, nil
+}