mirror of
https://github.com/BaaRai/ARTEX.git
synced 2026-11-13 19:50:54 +08:00
BaaRai
353a947285
deadline_at is a wall-clock timestamp, so pause time counted against the task timeout: pausing overnight (or shutting the machine down while paused) made the deadline ring while parked, and resuming landed straight into the settle sequence and a timeout terminal state. ApplyTaskPause now stamps paused_at, and ApplyTaskAdmission credits the paused interval back onto deadline_at in the same atomic UPDATE that ends the pause (paused_at is cleared there too). The shifted deadline flows to the live handle via the existing RETURNING mirror, so the engine clock context picks it up. Matches the re-basing the archive-restore path already did for paused tasks. Migration is an idempotent ADD COLUMN; tasks paused before the upgrade have no paused_at and keep the old behavior. Fixes #1
79 lines
2.6 KiB
Go
79 lines
2.6 KiB
Go
package server
|
|
|
|
import (
|
|
"database/sql"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// Pausing a task must freeze its wall-clock timeout budget: the admission that
|
|
// ends the pause credits the paused interval back onto deadline_at, so a task
|
|
// paused overnight (or while the machine is off) resumes with its remaining
|
|
// time instead of immediately settling into the timeout terminal state (#1).
|
|
func TestPauseShiftsTaskDeadlineOnResume(t *testing.T) {
|
|
s, _, _ := trafficEvidenceServer(t)
|
|
m := s.m
|
|
pg := s.m.pg
|
|
|
|
task, err := m.CreateTask("pause deadline test", "verify the deadline is frozen while paused", nil, 3600, 0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
id, err := strconv.ParseInt(task.ID, 10, 64)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { pg.Exec(`DELETE FROM tasks WHERE id=$1`, id) })
|
|
|
|
// A running task with a stamped deadline (first_run_at + 1h budget).
|
|
if _, err := m.StampTaskFirstRun(task.ID); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := pg.Exec(`UPDATE tasks SET status='running' WHERE id=$1`, id); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var before time.Time
|
|
if err := pg.QueryRow(`SELECT deadline_at FROM tasks WHERE id=$1`, id).Scan(&before); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Pause stamps paused_at alongside the user pause.
|
|
if err := m.ApplyTaskPause(task.ID); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var pausedAt sql.NullTime
|
|
if err := pg.QueryRow(`SELECT paused_at FROM tasks WHERE id=$1`, id).Scan(&pausedAt); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !pausedAt.Valid || pausedAt.Time.After(time.Now()) {
|
|
t.Fatalf("pause should stamp paused_at at the current time, got %v", pausedAt.Time)
|
|
}
|
|
|
|
// Keep the task parked long enough for the credited interval to be measurable.
|
|
const paused = 1100 * time.Millisecond
|
|
time.Sleep(paused)
|
|
|
|
// Admission out of the paused state (a paused task keeps its lifecycle status).
|
|
if err := m.ApplyTaskAdmission(task.ID, "running", "running", false, "resume", false); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var after time.Time
|
|
var cleared sql.NullTime
|
|
if err := pg.QueryRow(`SELECT deadline_at, paused_at FROM tasks WHERE id=$1`, id).Scan(&after, &cleared); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if shifted := after.Sub(before); shifted < paused-100*time.Millisecond {
|
|
t.Fatalf("resume should credit the paused interval onto the deadline, got shift=%v (paused=%v)", shifted, paused)
|
|
}
|
|
if cleared.Valid {
|
|
t.Fatalf("admission should clear paused_at, got %v", cleared.Time)
|
|
}
|
|
// The live handle must mirror the shifted deadline (the engine clock context
|
|
// is built from it).
|
|
if got := task.lifecycleSnapshot().DeadlineAt; got != after.Unix() {
|
|
t.Fatalf("live handle deadline = %d, want the shifted %d", got, after.Unix())
|
|
}
|
|
}
|