generated from VLADIMIR/template
104 lines
1.6 KiB
Go
104 lines
1.6 KiB
Go
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
|
|
}
|