58 lines
2.1 KiB
Go
58 lines
2.1 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
)
|
||
|
|
|
||
|
|
// UpsertCheck records (or, on retry, overwrites) a single check result. The
|
||
|
|
// UNIQUE(ip_id, attempt_number, source, check_type, target) constraint plus
|
||
|
|
// this upsert is what makes agent/prober result submission safely
|
||
|
|
// retryable without producing duplicate rows.
|
||
|
|
func (d *DB) UpsertCheck(ctx context.Context, c Check) error {
|
||
|
|
_, err := d.ExecContext(ctx, `
|
||
|
|
INSERT INTO checks (ip_id, ip_address, attempt_number, validator_id, source, check_type, target,
|
||
|
|
success, latency_ms, detail, checked_at, created_at)
|
||
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||
|
|
ON CONFLICT(ip_id, attempt_number, source, check_type, target) DO UPDATE SET
|
||
|
|
validator_id=excluded.validator_id,
|
||
|
|
success=excluded.success,
|
||
|
|
latency_ms=excluded.latency_ms,
|
||
|
|
detail=excluded.detail,
|
||
|
|
checked_at=excluded.checked_at
|
||
|
|
`, c.IPID, c.IPAddress, c.AttemptNumber, c.ValidatorID, c.Source, c.CheckType, c.Target,
|
||
|
|
c.Success, c.LatencyMS, c.Detail, timeToDB(c.CheckedAt), timeToDB(Now()))
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// ListChecksForAttempt returns every check recorded for an IP's current
|
||
|
|
// attempt — the input to overall-result aggregation.
|
||
|
|
func (d *DB) ListChecksForAttempt(ctx context.Context, ipID int64, attemptNumber int) ([]Check, error) {
|
||
|
|
rows, err := d.QueryContext(ctx, `
|
||
|
|
SELECT id, ip_id, ip_address, attempt_number, validator_id, source, check_type, target,
|
||
|
|
success, latency_ms, detail, checked_at, created_at
|
||
|
|
FROM checks WHERE ip_id=? AND attempt_number=?
|
||
|
|
ORDER BY source, check_type, target
|
||
|
|
`, ipID, attemptNumber)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer rows.Close()
|
||
|
|
|
||
|
|
var out []Check
|
||
|
|
for rows.Next() {
|
||
|
|
var c Check
|
||
|
|
var checkedAt, createdAt string
|
||
|
|
if err := rows.Scan(&c.ID, &c.IPID, &c.IPAddress, &c.AttemptNumber, &c.ValidatorID, &c.Source,
|
||
|
|
&c.CheckType, &c.Target, &c.Success, &c.LatencyMS, &c.Detail, &checkedAt, &createdAt); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if c.CheckedAt, err = dbToTime(checkedAt); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if c.CreatedAt, err = dbToTime(createdAt); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
out = append(out, c)
|
||
|
|
}
|
||
|
|
return out, rows.Err()
|
||
|
|
}
|