103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package usecase
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"time"
|
||
|
||
"apps/backend/internal/module/radar/domain"
|
||
)
|
||
|
||
// DailySweepHourUTC is the fixed daily schedule (spec §4.2：UTC 22:00 ＝台北 06:00)。
|
||
const DailySweepHourUTC = 22
|
||
|
||
// SweepJobScheduler schedules one radar_sweep job per watch per UTC day.
|
||
// Implemented by a thin adapter over job usecase.Service (see worker / service_context).
|
||
type SweepJobScheduler interface {
|
||
ScheduleRadarSweep(ctx context.Context, ownerUID int64, watchID string, runAt int64) (jobID string, err error)
|
||
}
|
||
|
||
// SweepJobSchedulerFunc adapts a function to SweepJobScheduler.
|
||
type SweepJobSchedulerFunc func(ctx context.Context, ownerUID int64, watchID string, runAt int64) (jobID string, err error)
|
||
|
||
func (f SweepJobSchedulerFunc) ScheduleRadarSweep(ctx context.Context, ownerUID int64, watchID string, runAt int64) (string, error) {
|
||
return f(ctx, ownerUID, watchID, runAt)
|
||
}
|
||
|
||
// ScheduleDailySweeps 在已過當日 UTC 22:00 時，為每個 active watch 建一筆 radar_sweep Job。
|
||
//
|
||
// 呼叫端必須已持有 worker maintenance Redis lock（value = workerID），
|
||
// 本函式本身不做分散式鎖；未過 22:00 時回 0 且不建 Job。
|
||
//
|
||
// 回傳建立（或已存在而回傳）的 job 數。
|
||
func (s *Service) ScheduleDailySweeps(ctx context.Context, now time.Time) (int, error) {
|
||
if s.SweepJobs == nil {
|
||
return 0, fmt.Errorf("%w: sweep job scheduler not configured", domain.ErrNotReady)
|
||
}
|
||
if now.IsZero() {
|
||
now = time.Now().UTC()
|
||
} else {
|
||
now = now.UTC()
|
||
}
|
||
if !PastDailySweepSlot(now) {
|
||
return 0, nil
|
||
}
|
||
runAt := DailySweepRunAt(now)
|
||
|
||
watches, err := s.Repo.ListAllActiveWatches(ctx)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
n := 0
|
||
for _, w := range watches {
|
||
if w == nil || w.Status != domain.WatchActive {
|
||
continue
|
||
}
|
||
if _, err := s.SweepJobs.ScheduleRadarSweep(ctx, w.OwnerUID, w.ID, runAt); err != nil {
|
||
return n, fmt.Errorf("schedule watch %s owner %d: %w", w.ID, w.OwnerUID, err)
|
||
}
|
||
n++
|
||
}
|
||
return n, nil
|
||
}
|
||
|
||
// PastDailySweepSlot reports whether now is at or after today's UTC 22:00.
|
||
func PastDailySweepSlot(now time.Time) bool {
|
||
now = now.UTC()
|
||
slot := time.Date(now.Year(), now.Month(), now.Day(), DailySweepHourUTC, 0, 0, 0, time.UTC)
|
||
return !now.Before(slot)
|
||
}
|
||
|
||
// DailySweepRunAt returns unix-ns for today's UTC 22:00 (the slot being scheduled).
|
||
func DailySweepRunAt(now time.Time) int64 {
|
||
now = now.UTC()
|
||
slot := time.Date(now.Year(), now.Month(), now.Day(), DailySweepHourUTC, 0, 0, 0, time.UTC)
|
||
return slot.UnixNano()
|
||
}
|
||
|
||
// BeginSweepRecord creates the RadarSweep shell for a claimed job (T527).
|
||
// Fetch / judge (T528–T530) attach progress onto the same record via UpdateSweep.
|
||
func (s *Service) BeginSweepRecord(ctx context.Context, ownerUID int64, watchID, jobID, path string) (*domain.RadarSweep, error) {
|
||
if ownerUID <= 0 || watchID == "" {
|
||
return nil, fmt.Errorf("%w: owner_uid and watch_id required", domain.ErrValidation)
|
||
}
|
||
if path == "" {
|
||
path = domain.SweepPathAPI
|
||
}
|
||
sw := &domain.RadarSweep{
|
||
ID: domain.NewID(),
|
||
OwnerUID: ownerUID,
|
||
WatchID: watchID,
|
||
JobID: jobID,
|
||
Path: path,
|
||
StartedAt: domain.NowNano(),
|
||
}
|
||
if err := sw.Normalize(); err != nil {
|
||
return nil, err
|
||
}
|
||
if err := s.Repo.CreateSweep(ctx, sw); err != nil {
|
||
return nil, err
|
||
}
|
||
return sw, nil
|
||
}
|