126 lines
3.0 KiB
Go
126 lines
3.0 KiB
Go
package tickets
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"reanimator/internal/domain"
|
|
)
|
|
|
|
type TicketRepository struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewTicketRepository(db *sql.DB) *TicketRepository {
|
|
return &TicketRepository{db: db}
|
|
}
|
|
|
|
func (r *TicketRepository) BeginTx(ctx context.Context) (*sql.Tx, error) {
|
|
return r.db.BeginTx(ctx, nil)
|
|
}
|
|
|
|
type execer interface {
|
|
ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error)
|
|
QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row
|
|
QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error)
|
|
}
|
|
|
|
func execerFor(db *sql.DB, tx *sql.Tx) execer {
|
|
if tx != nil {
|
|
return tx
|
|
}
|
|
return db
|
|
}
|
|
|
|
func (r *TicketRepository) Upsert(ctx context.Context, tx *sql.Tx, ticket domain.Ticket) (int64, error) {
|
|
execer := execerFor(r.db, tx)
|
|
_, err := execer.ExecContext(ctx,
|
|
`INSERT INTO tickets (source, external_id, title, status, opened_at, closed_at, url)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
ON DUPLICATE KEY UPDATE
|
|
title = VALUES(title),
|
|
status = VALUES(status),
|
|
opened_at = VALUES(opened_at),
|
|
closed_at = VALUES(closed_at),
|
|
url = VALUES(url)`,
|
|
ticket.Source,
|
|
ticket.ExternalID,
|
|
ticket.Title,
|
|
ticket.Status,
|
|
ticket.OpenedAt,
|
|
ticket.ClosedAt,
|
|
ticket.URL,
|
|
)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
var id int64
|
|
row := execer.QueryRowContext(ctx,
|
|
`SELECT id FROM tickets WHERE source = ? AND external_id = ?`,
|
|
ticket.Source,
|
|
ticket.ExternalID,
|
|
)
|
|
if err := row.Scan(&id); err != nil {
|
|
return 0, err
|
|
}
|
|
return id, nil
|
|
}
|
|
|
|
func (r *TicketRepository) LinkToAsset(ctx context.Context, tx *sql.Tx, ticketID, assetID int64) error {
|
|
execer := execerFor(r.db, tx)
|
|
_, err := execer.ExecContext(ctx,
|
|
`INSERT INTO ticket_links (ticket_id, asset_id)
|
|
VALUES (?, ?)
|
|
ON DUPLICATE KEY UPDATE ticket_id = VALUES(ticket_id)`,
|
|
ticketID,
|
|
assetID,
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (r *TicketRepository) ListByAsset(ctx context.Context, assetID int64) ([]domain.Ticket, error) {
|
|
rows, err := r.db.QueryContext(ctx,
|
|
`SELECT t.id, t.source, t.external_id, t.title, t.status, t.opened_at, t.closed_at, t.url, t.created_at, t.updated_at
|
|
FROM ticket_links tl
|
|
JOIN tickets t ON t.id = tl.ticket_id
|
|
WHERE tl.asset_id = ?
|
|
ORDER BY t.updated_at DESC, t.id DESC`,
|
|
assetID,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
items := make([]domain.Ticket, 0)
|
|
for rows.Next() {
|
|
var ticket domain.Ticket
|
|
var openedAt sql.NullTime
|
|
var closedAt sql.NullTime
|
|
var url sql.NullString
|
|
|
|
if err := rows.Scan(&ticket.ID, &ticket.Source, &ticket.ExternalID, &ticket.Title, &ticket.Status, &openedAt, &closedAt, &url, &ticket.CreatedAt, &ticket.UpdatedAt); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if openedAt.Valid {
|
|
value := openedAt.Time
|
|
ticket.OpenedAt = &value
|
|
}
|
|
if closedAt.Valid {
|
|
value := closedAt.Time
|
|
ticket.ClosedAt = &value
|
|
}
|
|
if url.Valid {
|
|
value := url.String
|
|
ticket.URL = &value
|
|
}
|
|
items = append(items, ticket)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
return items, nil
|
|
}
|