28 lines
695 B
Go
28 lines
695 B
Go
package store
|
|
|
|
import "context"
|
|
|
|
type Run struct {
|
|
ID int64
|
|
TaskID int64
|
|
Status string
|
|
TotalCopied int64
|
|
TotalSkipped int64
|
|
TotalErrors int64
|
|
}
|
|
|
|
func (s *Store) CreateRun(ctx context.Context, taskID int64) (int64, error) {
|
|
var id int64
|
|
err := s.Pool.QueryRow(ctx,
|
|
`INSERT INTO runs (task_id) VALUES ($1) RETURNING id`, taskID).Scan(&id)
|
|
return id, err
|
|
}
|
|
|
|
func (s *Store) FinishRun(ctx context.Context, id int64, status string, copied, skipped, errs int64) error {
|
|
_, err := s.Pool.Exec(ctx,
|
|
`UPDATE runs SET status=$2, finished_at=now(),
|
|
total_copied=$3, total_skipped=$4, total_errors=$5 WHERE id=$1`,
|
|
id, status, copied, skipped, errs)
|
|
return err
|
|
}
|