fix: integrate recoverable and fair Job notification retries
This commit is contained in:
@@ -20,6 +20,7 @@ pub enum JobNotificationFacts {
|
||||
Result(String),
|
||||
NoFindings,
|
||||
StaleContent,
|
||||
/// SourceLookupFailed leaves routing undecided; other failures are terminal.
|
||||
Failed(NotificationFailureCode),
|
||||
}
|
||||
|
||||
@@ -74,6 +75,23 @@ pub(in crate::store) fn install(conn: &Connection) -> Result<()> {
|
||||
FOREIGN KEY(workspace_id,delivery_id) REFERENCES notification_decisions(workspace_id,delivery_id));
|
||||
CREATE TRIGGER notification_job_legacy_no_update BEFORE UPDATE ON notification_job_legacy_receipts BEGIN SELECT RAISE(ABORT,'Legacy notification evidence is immutable'); END;
|
||||
CREATE TRIGGER notification_job_legacy_no_delete BEFORE DELETE ON notification_job_legacy_receipts BEGIN SELECT RAISE(ABORT,'Legacy notification evidence is immutable'); END;")?;
|
||||
// Preparation may fail before a routing decision exists. Keep observations
|
||||
// against the common source, never invent a terminal decision to hold them.
|
||||
conn.execute_batch("CREATE TABLE notification_source_preparation_failures (
|
||||
sequence INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
source_sequence INTEGER NOT NULL REFERENCES notification_sources(sequence),
|
||||
code TEXT NOT NULL,
|
||||
recorded_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ','now')));
|
||||
CREATE INDEX notification_source_preparation_failures_source ON notification_source_preparation_failures(source_sequence,sequence);
|
||||
CREATE TRIGGER notification_source_preparation_failures_no_update BEFORE UPDATE ON notification_source_preparation_failures BEGIN SELECT RAISE(ABORT,'Source preparation evidence is immutable'); END;
|
||||
CREATE TRIGGER notification_source_preparation_failures_no_delete BEFORE DELETE ON notification_source_preparation_failures BEGIN SELECT RAISE(ABORT,'Source preparation evidence is immutable'); END;")?;
|
||||
// Mutable scheduling only, independent of immutable source/decision evidence.
|
||||
// Both undecided preparation retries and pinned deliveries use this clock.
|
||||
conn.execute_batch(
|
||||
"CREATE TABLE notification_source_offers (
|
||||
source_sequence INTEGER PRIMARY KEY REFERENCES notification_sources(sequence),
|
||||
last_offered INTEGER NOT NULL CHECK(last_offered > 0));",
|
||||
)?;
|
||||
// Includes no-target results and the crash gap before the former reservation.
|
||||
let identities = conn.prepare("SELECT a.workspace_id,a.job_id,a.attempt_id FROM backend_job_attempts a JOIN backend_jobs j USING(workspace_id,job_id) WHERE a.state='completed' AND j.state='completed' AND a.attempt=j.current_attempt")?
|
||||
.query_map([], |r| Ok((r.get::<_,String>(0)?,r.get::<_,String>(1)?,r.get::<_,String>(2)?)))?
|
||||
@@ -112,15 +130,17 @@ pub(in crate::store) fn pending(
|
||||
) -> Result<Vec<JobNotificationSource>> {
|
||||
store.with_conn_mut(|conn| {
|
||||
let tx=conn.transaction_with_behavior(TransactionBehavior::Immediate)?;
|
||||
let rows = tx.prepare("SELECT s.facts_json,s.sequence FROM notification_sources s LEFT JOIN notification_decisions d ON d.source_sequence=s.sequence WHERE s.workspace_id=?1 AND s.role='backend_job' AND (d.source_sequence IS NULL OR d.disposition='deliver') AND NOT EXISTS(SELECT 1 FROM notification_acknowledgements a WHERE a.workspace_id=?1 AND a.delivery_id=s.input_event_id) AND NOT EXISTS(SELECT 1 FROM notification_job_legacy_receipts l WHERE l.workspace_id=?1 AND l.delivery_id=s.input_event_id AND l.runtime_accepted=1) ORDER BY COALESCE(d.last_offered,0),s.sequence LIMIT ?2")?
|
||||
let rows = tx.prepare("SELECT s.facts_json,s.sequence FROM notification_sources s LEFT JOIN notification_decisions d ON d.source_sequence=s.sequence LEFT JOIN notification_source_offers o ON o.source_sequence=s.sequence WHERE s.workspace_id=?1 AND s.role='backend_job' AND (d.source_sequence IS NULL OR d.disposition='deliver') AND NOT EXISTS(SELECT 1 FROM notification_acknowledgements a WHERE a.workspace_id=?1 AND a.delivery_id=s.input_event_id) AND NOT EXISTS(SELECT 1 FROM notification_job_legacy_receipts l WHERE l.workspace_id=?1 AND l.delivery_id=s.input_event_id AND l.runtime_accepted=1) ORDER BY COALESCE(o.last_offered,0),s.sequence LIMIT ?2")?
|
||||
.query_map(params![workspace,limit.clamp(1,100) as i64],|r|Ok((r.get::<_,String>(0)?,r.get::<_,i64>(1)?)))?
|
||||
.collect::<rusqlite::Result<Vec<_>>>()?;
|
||||
let tick:i64=tx.query_row("SELECT COALESCE(MAX(last_offered),0)+1 FROM notification_decisions WHERE workspace_id=?1",[workspace],|r|r.get(0))?;
|
||||
// Advance every offered source under the scan's write lock, even if no
|
||||
// routing decision exists yet. Failure observation IDs are not a clock.
|
||||
let tick:i64=tx.query_row("SELECT COALESCE(MAX(o.last_offered),0)+1 FROM notification_source_offers o JOIN notification_sources s ON s.sequence=o.source_sequence WHERE s.workspace_id=?1 AND s.role='backend_job'",[workspace],|r|r.get(0))?;
|
||||
let mut sources=Vec::new();
|
||||
for (facts,sequence) in rows {
|
||||
let facts:serde_json::Value=decode(&facts)?;
|
||||
sources.push(serde_json::from_value(facts["backend_job"].clone()).map_err(|e|Error::Store(e.to_string()))?);
|
||||
tx.execute("UPDATE notification_decisions SET last_offered=?2 WHERE source_sequence=?1",params![sequence,tick])?;
|
||||
tx.execute("INSERT INTO notification_source_offers(source_sequence,last_offered) VALUES(?1,?2) ON CONFLICT(source_sequence) DO UPDATE SET last_offered=excluded.last_offered",params![sequence,tick])?;
|
||||
}
|
||||
tx.commit()?; Ok(sources)
|
||||
})
|
||||
@@ -172,6 +192,13 @@ pub(in crate::store) fn route(
|
||||
let ack:bool=tx.query_row("SELECT EXISTS(SELECT 1 FROM notification_acknowledgements WHERE workspace_id=?1 AND delivery_id=?2) OR EXISTS(SELECT 1 FROM notification_job_legacy_receipts WHERE workspace_id=?1 AND delivery_id=?2 AND runtime_accepted=1)",params![workspace,source.delivery_id],|r|r.get(0))?;
|
||||
tx.commit()?; return Ok(disposition=="deliver" && !ack);
|
||||
}
|
||||
if let JobNotificationFacts::Failed(NotificationFailureCode::SourceLookupFailed)=facts {
|
||||
// A transient checker read is not suppression or an immutable defer.
|
||||
// With no decision, the existing pending scan retries the same source.
|
||||
// Check existing decisions first: never replace pinned request evidence.
|
||||
tx.execute("INSERT INTO notification_source_preparation_failures(source_sequence,code) SELECT sequence,?3 FROM notification_sources WHERE workspace_id=?1 AND input_event_id=?2 AND role='backend_job'",params![workspace,source.delivery_id,code(NotificationFailureCode::SourceLookupFailed)?])?;
|
||||
tx.commit()?; return Ok(false);
|
||||
}
|
||||
let (disposition,reason)=policy(job.request.notification_target.is_some(),facts);
|
||||
save_decision(&tx, workspace, &source.delivery_id, disposition, reason)?;
|
||||
if let JobNotificationFacts::Failed(failure)=facts { save_failure(&tx,workspace,&source.delivery_id,*failure)?; }
|
||||
@@ -393,6 +420,11 @@ pub(in crate::store) fn authorize(
|
||||
|
||||
pub(in crate::store) fn verify(conn: &Connection) -> Result<()> {
|
||||
for name in [
|
||||
"notification_source_offers",
|
||||
"notification_source_preparation_failures",
|
||||
"notification_source_preparation_failures_source",
|
||||
"notification_source_preparation_failures_no_update",
|
||||
"notification_source_preparation_failures_no_delete",
|
||||
"notification_job_identity",
|
||||
"notification_job_source_guard",
|
||||
"notification_job_targets",
|
||||
|
||||
@@ -102,14 +102,6 @@ async fn job_notification_routing_common_policy_distinguishes_suppression_failur
|
||||
"stale_content",
|
||||
None,
|
||||
),
|
||||
(
|
||||
"missing",
|
||||
true,
|
||||
JobNotificationFacts::Failed(NotificationFailureCode::SourceLookupFailed),
|
||||
"defer",
|
||||
"source_unavailable",
|
||||
Some("source_lookup_failed"),
|
||||
),
|
||||
(
|
||||
"invalid",
|
||||
true,
|
||||
@@ -201,6 +193,153 @@ async fn job_notification_routing_common_policy_distinguishes_suppression_failur
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn job_notification_preparation_failures_remain_pending_and_append_only() {
|
||||
let (_directory, store) = tracking_store_fixture().await;
|
||||
for target in [true, false] {
|
||||
let (_, source) = job_notice_fixture(&store, if target { "retry" } else { "retry-no-target" }, target);
|
||||
for _ in 0..2 {
|
||||
assert!(!store.route_job_notification("w", &source, &JobNotificationFacts::Failed(NotificationFailureCode::SourceLookupFailed)).unwrap());
|
||||
}
|
||||
store.with_conn(|conn| {
|
||||
let sequence: i64 = conn.query_row("SELECT sequence FROM notification_sources WHERE input_event_id=?1", [&source.delivery_id], |r| r.get(0))?;
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM notification_source_preparation_failures WHERE source_sequence=?1 AND code='source_lookup_failed'", [sequence], |r| r.get::<_, i64>(0))?, 2);
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM notification_decisions WHERE source_sequence=?1", [sequence], |r| r.get::<_, i64>(0))?, 0);
|
||||
assert!(conn.execute("UPDATE notification_source_preparation_failures SET code='invalid_result' WHERE source_sequence=?1", [sequence]).is_err());
|
||||
assert!(conn.execute("DELETE FROM notification_source_preparation_failures WHERE source_sequence=?1", [sequence]).is_err());
|
||||
assert!(conn.execute("INSERT INTO notification_source_preparation_failures(source_sequence,code) VALUES(-1,'source_lookup_failed')", []).is_err());
|
||||
Ok(())
|
||||
}).unwrap();
|
||||
assert!(store.list_pending_job_notifications("w", 100).unwrap().iter().any(|s| s.delivery_id == source.delivery_id));
|
||||
assert_eq!(store.route_job_notification("w", &source, &JobNotificationFacts::Result("recovered body".into())).unwrap(), target);
|
||||
let pinned = store.job_notification_request("w", &source.delivery_id).unwrap();
|
||||
// Later failed reads cannot overwrite the pinned decision/body/recipient
|
||||
// or fabricate a new pre-decision observation after routing has finished.
|
||||
assert_eq!(store.route_job_notification("w", &source, &JobNotificationFacts::Failed(NotificationFailureCode::SourceLookupFailed)).unwrap(), target);
|
||||
assert_eq!(store.job_notification_request("w", &source.delivery_id).unwrap(), pinned);
|
||||
store.with_conn(|conn| {
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM notification_source_preparation_failures f JOIN notification_sources s ON s.sequence=f.source_sequence WHERE s.input_event_id=?1", [&source.delivery_id], |r| r.get::<_, i64>(0))?, 2);
|
||||
Ok(())
|
||||
}).unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn job_notification_fair_offers_survive_reopen_and_bound_preparation_and_ack_retries() {
|
||||
let (directory, mut store) = tracking_store_fixture().await;
|
||||
let mut failing = Vec::new();
|
||||
for i in 0..25 {
|
||||
let (_, source) = job_notice_fixture(&store, &format!("failing-{i}"), true);
|
||||
assert!(!store.route_job_notification(
|
||||
"w", &source,
|
||||
&JobNotificationFacts::Failed(NotificationFailureCode::SourceLookupFailed),
|
||||
).unwrap());
|
||||
failing.push(source);
|
||||
}
|
||||
let (_, healthy) = job_notice_fixture(&store, "later-healthy", true);
|
||||
let (_, retry) = job_notice_fixture(&store, "later-pinned-retry", true);
|
||||
let target = TicketWorkerNotificationTarget {
|
||||
worker: RuntimeWorkerRef::new("r", "replacement"),
|
||||
authentication_binding_id: "tracking-binding".into(),
|
||||
};
|
||||
assert!(store.route_job_notification(
|
||||
"w", &retry, &JobNotificationFacts::Result("original pinned bytes".into()),
|
||||
).unwrap());
|
||||
assert_eq!(store.authorize_job_notification("w", &retry.delivery_id, &target).unwrap(), None);
|
||||
let retry_key = NotificationDeliveryKey::WorkerNotification {
|
||||
delivery_id: retry.delivery_id.clone(),
|
||||
};
|
||||
store.record_notification_delivery_failure(
|
||||
"w", &retry_key, NotificationFailureCode::AcceptanceUnknown,
|
||||
).unwrap();
|
||||
let pinned = store.job_notification_request("w", &retry.delivery_id).unwrap();
|
||||
// Start with an already-offered pinned retry, not only a never-offered result.
|
||||
// All sources share the same offer clock; 25 failure observations are not ticks.
|
||||
assert_eq!(store.list_pending_job_notifications("w", 100).unwrap().len(), 27);
|
||||
let snapshot = |store: &SqliteWorkspaceStore, table: &str| {
|
||||
store.with_conn(|conn| Ok(stored_fixture_rows(conn, table))).unwrap()
|
||||
};
|
||||
let original_sources = snapshot(&store, "notification_sources");
|
||||
let original_decisions = snapshot(&store, "notification_decisions");
|
||||
let original_preparation_failures = snapshot(&store, "notification_source_preparation_failures");
|
||||
let original_delivery_failures = snapshot(&store, "notification_failures");
|
||||
let original_jobs = snapshot(&store, "backend_jobs");
|
||||
let mut failure_offers = vec![0; failing.len()];
|
||||
let mut healthy_round = None;
|
||||
let mut retry_rounds = Vec::new();
|
||||
|
||||
for round in 1..=4 {
|
||||
// Reopen before every production-sized round, including the first retry.
|
||||
let saved_offers = snapshot(&store, "notification_source_offers");
|
||||
drop(store);
|
||||
store = SqliteWorkspaceStore::open_for_test(directory.path().join("workspace.db")).unwrap();
|
||||
assert_eq!(snapshot(&store, "notification_source_offers"), saved_offers);
|
||||
let pending = store.list_pending_job_notifications("w", 20).unwrap();
|
||||
assert_eq!(pending.len(), 20);
|
||||
if round == 1 {
|
||||
assert!(pending.iter().all(|s| s.job_id.starts_with("failing-")));
|
||||
}
|
||||
for source in pending {
|
||||
if let Some(index) = failing.iter().position(|s| s.delivery_id == source.delivery_id) {
|
||||
failure_offers[index] += 1;
|
||||
assert!(!store.route_job_notification(
|
||||
"w", &source,
|
||||
&JobNotificationFacts::Failed(NotificationFailureCode::SourceLookupFailed),
|
||||
).unwrap());
|
||||
} else if source.delivery_id == healthy.delivery_id {
|
||||
assert!(healthy_round.replace(round).is_none());
|
||||
assert!(store.route_job_notification(
|
||||
"w", &source, &JobNotificationFacts::Result("healthy body".into()),
|
||||
).unwrap());
|
||||
assert_eq!(store.authorize_job_notification("w", &source.delivery_id, &target).unwrap(), None);
|
||||
store.ack_job_notification("w", &source.delivery_id, &target.worker).unwrap();
|
||||
} else {
|
||||
assert_eq!(source.delivery_id, retry.delivery_id);
|
||||
retry_rounds.push(round);
|
||||
// Neither scheduling nor a new rendering may replace the request.
|
||||
assert!(store.route_job_notification(
|
||||
"w", &source, &JobNotificationFacts::Result("must not replace pinned bytes".into()),
|
||||
).unwrap());
|
||||
assert_eq!(store.job_notification_request("w", &source.delivery_id).unwrap(), pinned);
|
||||
if retry_rounds.len() == 1 {
|
||||
store.record_notification_delivery_failure(
|
||||
"w", &retry_key, NotificationFailureCode::AcceptanceUnknown,
|
||||
).unwrap();
|
||||
} else {
|
||||
store.ack_job_notification("w", &source.delivery_id, &target.worker).unwrap();
|
||||
}
|
||||
}
|
||||
}
|
||||
if round == 2 {
|
||||
assert!(failure_offers.iter().all(|count| *count >= 1));
|
||||
assert_eq!(healthy_round, Some(2));
|
||||
assert_eq!(retry_rounds, vec![2]);
|
||||
}
|
||||
}
|
||||
// With 26 sources still pending after the healthy ACK, even the last source
|
||||
// in an offer-time tie returns within ceil(26 / 20) = 2 further rounds.
|
||||
assert_eq!(retry_rounds, vec![2, 4], "a pinned retry must not starve behind undecided sources");
|
||||
assert!(job_notice_evidence(&store, &healthy).2);
|
||||
assert!(job_notice_evidence(&store, &retry).2);
|
||||
assert_eq!(snapshot(&store, "notification_sources"), original_sources);
|
||||
assert_eq!(snapshot(&store, "backend_jobs"), original_jobs);
|
||||
let decisions = snapshot(&store, "notification_decisions");
|
||||
assert!(original_decisions.iter().all(|row| decisions.contains(row)));
|
||||
let preparation_failures = snapshot(&store, "notification_source_preparation_failures");
|
||||
assert!(preparation_failures.starts_with(&original_preparation_failures));
|
||||
assert_eq!(preparation_failures.len(), 25 + failure_offers.iter().sum::<usize>());
|
||||
assert!(snapshot(&store, "notification_failures").starts_with(&original_delivery_failures));
|
||||
assert_eq!(store.job_notification_request("w", &retry.delivery_id).unwrap(), pinned);
|
||||
assert!(store.list_pending_job_notifications("w", 20).unwrap().iter().all(|s| s.job_id.starts_with("failing-")));
|
||||
store.with_conn(|conn| {
|
||||
assert!(conn.execute("INSERT INTO notification_source_offers VALUES(-1,1)", []).is_err());
|
||||
assert!(conn.execute("UPDATE notification_source_offers SET last_offered=0", []).is_err());
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM notification_sources s JOIN notification_decisions d ON d.source_sequence=s.sequence WHERE s.role='backend_job'", [], |r| r.get::<_, i64>(0))?, 2);
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM pragma_foreign_key_check", [], |r| r.get::<_, i64>(0))?, 0);
|
||||
Ok(())
|
||||
}).unwrap();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn job_notification_result_commit_reopen_parallel_recovery_and_fixed_target_receipt() {
|
||||
let (directory, store) = tracking_store_fixture().await;
|
||||
@@ -432,6 +571,46 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
migrate_assignment_delivery_v92_to_v93(&conn).unwrap();
|
||||
migrate_ticket_posts_v93_to_v94(&conn).unwrap();
|
||||
migrate_notification_routing_v94_to_v95(&conn).unwrap();
|
||||
// Populate the v95 common graph before rebuilding its source parent. Include
|
||||
// both accepted and retryable Ticket requests, with retained failure history.
|
||||
for id in ["ticket-ack", "ticket-pending"] {
|
||||
conn.execute("INSERT INTO notification_sources(workspace_id,ticket_id,input_event_id,role,assignment_id,kind,facts_json) VALUES('w','pending',?1,'worker','ticket-worker-assignment','comment','{\"principal_kind\":\"worker\",\"principal_id\":null,\"runtime_id\":\"r\",\"worker_id\":\"worker\",\"status\":null,\"case_state\":\"open\",\"released\":false}')", [id]).unwrap();
|
||||
conn.execute("INSERT INTO notification_decisions(source_sequence,workspace_id,delivery_id,disposition,reason,purpose,last_offered) SELECT sequence,'w',?1,'deliver','attention','attention',17 FROM notification_sources WHERE input_event_id=?1", [id]).unwrap();
|
||||
conn.execute("INSERT INTO notification_requests VALUES('w',?1,'original Ticket bytes','{\"worker\":{\"runtime_id\":\"r\",\"worker_id\":\"worker\"},\"authentication_binding_id\":\"original-binding\"}')", [id]).unwrap();
|
||||
conn.execute("INSERT INTO notification_failures(workspace_id,delivery_id,code,recorded_at) VALUES('w',?1,'acceptance_unknown','original-failure-time')", [id]).unwrap();
|
||||
}
|
||||
conn.execute("INSERT INTO notification_acknowledgements SELECT workspace_id,delivery_id,target_json FROM notification_requests WHERE delivery_id='ticket-ack'", []).unwrap();
|
||||
let ticket_graph = |conn: &Connection| {
|
||||
[
|
||||
"SELECT * FROM notification_sources WHERE role!='backend_job' ORDER BY sequence".to_string(),
|
||||
"SELECT d.* FROM notification_decisions d JOIN notification_sources s ON s.sequence=d.source_sequence WHERE s.role!='backend_job' ORDER BY d.rowid".to_string(),
|
||||
].into_iter().chain(["notification_requests", "notification_acknowledgements", "notification_failures"].map(|table| format!("SELECT f.* FROM {table} f JOIN notification_decisions d USING(workspace_id,delivery_id) JOIN notification_sources s ON s.sequence=d.source_sequence WHERE s.role!='backend_job' ORDER BY f.rowid")))
|
||||
.map(|sql| fixture_query_rows(conn, &sql)).collect::<Vec<_>>()
|
||||
};
|
||||
let ticket_before = ticket_graph(&conn);
|
||||
assert!(ticket_before.iter().all(|rows| !rows.is_empty()));
|
||||
let schema_before = fixture_query_rows(&conn, "SELECT type,name,sql FROM sqlite_schema WHERE type IN ('trigger','index') AND sql IS NOT NULL ORDER BY name");
|
||||
let check_guards = |conn: &Connection| {
|
||||
for sql in [
|
||||
"UPDATE notification_sources SET kind='changed' WHERE input_event_id='ticket-ack'",
|
||||
"DELETE FROM notification_sources WHERE input_event_id='ticket-ack'",
|
||||
"UPDATE notification_decisions SET reason='changed' WHERE delivery_id='ticket-ack'",
|
||||
"DELETE FROM notification_decisions WHERE delivery_id='ticket-ack'",
|
||||
"UPDATE notification_requests SET content='changed' WHERE delivery_id='ticket-ack'",
|
||||
"DELETE FROM notification_acknowledgements WHERE delivery_id='ticket-ack'",
|
||||
"UPDATE notification_failures SET code='changed' WHERE delivery_id='ticket-ack'",
|
||||
"INSERT INTO notification_decisions(source_sequence,workspace_id,delivery_id,disposition,reason,purpose) VALUES(-1,'w','orphan','deliver','result','result')",
|
||||
"INSERT INTO notification_requests VALUES('w','orphan','content','{}')",
|
||||
"INSERT INTO notification_acknowledgements VALUES('w','orphan','{}')",
|
||||
"INSERT INTO notification_failures(workspace_id,delivery_id,code) VALUES('w','orphan','acceptance_unknown')",
|
||||
] {
|
||||
assert!(conn.execute_batch(sql).is_err(), "guard lost: {sql}");
|
||||
}
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM pragma_foreign_key_check", [], |r| r.get::<_, i64>(0)).unwrap(), 0);
|
||||
assert_eq!(conn.query_row("PRAGMA legacy_alter_table", [], |r| r.get::<_, i64>(0)).unwrap(), 0);
|
||||
assert_eq!(conn.query_row("PRAGMA foreign_keys", [], |r| r.get::<_, i64>(0)).unwrap(), 1);
|
||||
};
|
||||
check_guards(&conn);
|
||||
let accepted = job_notice_legacy_seed(&conn, "ack", Some("completed"), None, true);
|
||||
let clean = job_notice_legacy_seed(
|
||||
&conn,
|
||||
@@ -492,6 +671,15 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
[checker_result.to_string()],
|
||||
)
|
||||
.unwrap();
|
||||
// Multiple retained identities must never create a fresh send: an ACK wins
|
||||
// over unknown history; without an ACK the ambiguity stays terminal.
|
||||
let multi_ack = job_notice_legacy_seed(&conn, "multi-ack", Some("completed"), None, true);
|
||||
let multi_unknown = job_notice_legacy_seed(&conn, "multi-unknown", Some("unknown"), Some("notification_outcome_unknown"), true);
|
||||
for id in [&multi_ack, &multi_unknown] {
|
||||
conn.execute("INSERT INTO backend_job_deliveries(workspace_id,delivery_id,job_id,attempt_id,target_runtime_id,target_worker_id,state,failure_category,created_at,updated_at) SELECT workspace_id,'another-' || delivery_id,job_id,attempt_id,target_runtime_id,target_worker_id,'unknown','notification_outcome_unknown','0','0' FROM backend_job_deliveries WHERE delivery_id=?1", [id]).unwrap();
|
||||
}
|
||||
// Earlier unknown beats the derived identity only when there is no real ACK.
|
||||
let multi_unknown_retained = format!("another-{multi_unknown}");
|
||||
let before = stored_fixture_rows(&conn, "backend_job_deliveries");
|
||||
// Force migration failure after the source parent rebuild and verify rollback
|
||||
// restores the old parent, rows, constraints, version and connection pragmas.
|
||||
@@ -499,6 +687,9 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
.unwrap();
|
||||
assert!(migrate_job_notifications_v95_to_v96(&conn).is_err());
|
||||
assert_eq!(current_schema_version(&conn).unwrap(), 95);
|
||||
assert_eq!(ticket_graph(&conn), ticket_before);
|
||||
assert_eq!(fixture_query_rows(&conn, "SELECT type,name,sql FROM sqlite_schema WHERE type IN ('trigger','index') AND sql IS NOT NULL ORDER BY name"), schema_before);
|
||||
check_guards(&conn);
|
||||
assert_eq!(stored_fixture_rows(&conn, "backend_job_deliveries"), before);
|
||||
assert_eq!(
|
||||
conn.query_row("PRAGMA foreign_keys", [], |r| r.get::<_, i64>(0))
|
||||
@@ -508,6 +699,11 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
conn.execute_batch("DROP TABLE notification_job_targets")
|
||||
.unwrap();
|
||||
migrate_job_notifications_v95_to_v96(&conn).unwrap();
|
||||
assert_eq!(current_schema_version(&conn).unwrap(), 96);
|
||||
assert_eq!(ticket_graph(&conn), ticket_before);
|
||||
check_guards(&conn);
|
||||
let schema_after = fixture_query_rows(&conn, "SELECT type,name,sql FROM sqlite_schema WHERE type IN ('trigger','index') AND sql IS NOT NULL ORDER BY name");
|
||||
assert!(schema_before.iter().all(|row| schema_after.contains(row)));
|
||||
assert_eq!(stored_fixture_rows(&conn, "backend_job_deliveries"), before);
|
||||
for (id, disposition, reason, runtime_accepted) in [
|
||||
(&accepted, "deliver", "result", true),
|
||||
@@ -515,6 +711,8 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
(&stale, "suppress", "stale_content", false),
|
||||
(&unknown, "deliver", "result", false),
|
||||
(&ambiguous, "defer", "source_unavailable", false),
|
||||
(&multi_ack, "deliver", "result", true),
|
||||
(&multi_unknown_retained, "defer", "source_unavailable", false),
|
||||
] {
|
||||
let got=conn.query_row("SELECT d.disposition,d.reason,l.runtime_accepted FROM notification_decisions d JOIN notification_job_legacy_receipts l USING(workspace_id,delivery_id) WHERE d.workspace_id='w' AND d.delivery_id=?1",[id],|r|Ok((r.get::<_,String>(0)?,r.get::<_,String>(1)?,r.get::<_,bool>(2)?))).unwrap();
|
||||
assert_eq!(got, (disposition.into(), reason.into(), runtime_accepted));
|
||||
@@ -526,9 +724,13 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
|r| r.get::<_, i64>(0)
|
||||
)
|
||||
.unwrap(),
|
||||
0,
|
||||
"never fabricate modern ACKs or request bodies for old receipts"
|
||||
1,
|
||||
"retain the Ticket ACK, never fabricate modern ACKs for legacy Job receipts"
|
||||
);
|
||||
for id in [&multi_ack, &multi_unknown_retained] {
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM notification_requests WHERE delivery_id=?1", [id], |r| r.get::<_, i64>(0)).unwrap(), 0);
|
||||
}
|
||||
assert_eq!(conn.query_row("SELECT code FROM notification_failures WHERE delivery_id=?1", [&multi_unknown_retained], |r| r.get::<_, String>(0)).unwrap(), "preparation_failed");
|
||||
assert_eq!(
|
||||
conn.query_row("SELECT COUNT(*) FROM pragma_foreign_key_check", [], |r| r
|
||||
.get::<_, i64>(
|
||||
@@ -547,6 +749,12 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
assert!(pinned.contains("Durable result digest: digest."));
|
||||
drop(conn);
|
||||
let store = SqliteWorkspaceStore::open_for_test(path).unwrap();
|
||||
store.with_conn(|conn| {
|
||||
assert_eq!(ticket_graph(conn), ticket_before);
|
||||
assert_eq!(stored_fixture_rows(conn, "backend_job_deliveries"), before);
|
||||
check_guards(conn);
|
||||
Ok(())
|
||||
}).unwrap();
|
||||
let pending = store.list_pending_job_notifications("w", 100).unwrap();
|
||||
assert_eq!(pending.len(), 4);
|
||||
for (id,stage) in [(&accepted,server_api::NotificationDiagnosticStage::RuntimeAccepted),(&clean,server_api::NotificationDiagnosticStage::Suppressed),(&stale,server_api::NotificationDiagnosticStage::Suppressed),(&unknown,server_api::NotificationDiagnosticStage::AcceptanceUnknown),(&ambiguous,server_api::NotificationDiagnosticStage::Deferred)] {
|
||||
@@ -573,7 +781,10 @@ fn job_notification_migration_retains_legacy_acks_suppressions_unknown_bytes_and
|
||||
assert!(pending.iter().all(|s| s.delivery_id != accepted
|
||||
&& s.delivery_id != clean
|
||||
&& s.delivery_id != stale
|
||||
&& s.delivery_id != ambiguous));
|
||||
&& s.delivery_id != ambiguous
|
||||
&& s.delivery_id != multi_ack
|
||||
&& s.delivery_id != multi_unknown_retained
|
||||
&& s.delivery_id != multi_unknown));
|
||||
assert_eq!(
|
||||
store
|
||||
.job_notification_request("w", &unknown)
|
||||
|
||||
@@ -123,6 +123,85 @@ async fn job_notification_real_worker_ack_loss_reopen_and_parallel_recovery_comm
|
||||
delivery_stop(&reopened, worker);
|
||||
}
|
||||
|
||||
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
||||
async fn job_notification_checker_read_failure_reopen_recovers_same_source_to_real_receipt() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
let (api, execution, _) = delivery_real_fixture(dir.path(), false).await;
|
||||
let target = spawn_ticket_check_source(&api, "checker-retry-target");
|
||||
let backend = browser_ticket_backend(&api).unwrap();
|
||||
let ticket_ref = backend.create(ticket::NewTicket::new("Writer-only scope in requirements")).unwrap();
|
||||
let ticket = backend.show(ticket_ref.id.into()).unwrap();
|
||||
let request = ticket_item_checker::request(
|
||||
&ticket,
|
||||
ticket_item_checker::TicketItemCheckEdit::Create,
|
||||
target.clone(),
|
||||
"Check saved Ticket content.".into(),
|
||||
).unwrap();
|
||||
let dispatched = api.dispatch_backend_job(&request).unwrap();
|
||||
let worker = dispatched.attempt.worker.as_ref().unwrap();
|
||||
let submission = BackendJobResultSubmission {
|
||||
job_id: request.job_id.clone(),
|
||||
attempt_id: dispatched.attempt.attempt_id.clone(),
|
||||
input_digest: request.input_digest().unwrap(),
|
||||
result: json!({"findings":[{"category":"writer_scope","quote":"Writer-only scope","reason":"Writer scope is mixed into requirements.","suggestion":"Remove the writer-specific sentence."}]}),
|
||||
};
|
||||
let accepted = api.store.accept_backend_job_result(TEST_WORKSPACE_ID, worker, &submission, TEST_CREATED_AT).unwrap();
|
||||
let source = api.store.job_notification_source(TEST_WORKSPACE_ID, &request.job_id, &accepted.attempt.attempt_id).unwrap();
|
||||
let conn = rusqlite::Connection::open(&api.config.database_path).unwrap();
|
||||
let source_snapshot = || conn.query_row("SELECT sequence,facts_json FROM notification_sources WHERE workspace_id=?1 AND input_event_id=?2", rusqlite::params![TEST_WORKSPACE_ID,source.delivery_id], |r| Ok((r.get::<_,i64>(0)?,r.get::<_,String>(1)?))).unwrap();
|
||||
let original_source = source_snapshot();
|
||||
// Fail exactly one checker read through the real adapter's error branch.
|
||||
// Its independent Ticket connection cannot open a directory as SQLite, while
|
||||
// the already-open result store remains available to persist failure evidence.
|
||||
let mut unavailable = api.clone();
|
||||
unavailable.config.database_path = dir.path().to_path_buf();
|
||||
notification_delivery::recover_job_notifications(&unavailable);
|
||||
let failures = || {
|
||||
let mut query = conn.prepare("SELECT sequence,source_sequence,code,recorded_at FROM notification_source_preparation_failures WHERE source_sequence=?1 ORDER BY sequence").unwrap();
|
||||
query.query_map([original_source.0], |r| Ok((r.get::<_,i64>(0)?,r.get::<_,i64>(1)?,r.get::<_,String>(2)?,r.get::<_,String>(3)?))).unwrap().collect::<rusqlite::Result<Vec<_>>>().unwrap()
|
||||
};
|
||||
let failed_once = failures();
|
||||
assert_eq!(failed_once.len(), 1);
|
||||
assert_eq!(failed_once[0].2, "source_lookup_failed");
|
||||
assert_eq!(conn.query_row("SELECT COUNT(*) FROM notification_decisions WHERE source_sequence=?1", [original_source.0], |r| r.get::<_,i64>(0)).unwrap(), 0);
|
||||
assert!(api.store.job_notification_request(TEST_WORKSPACE_ID, &source.delivery_id).unwrap().is_none());
|
||||
assert!(delivery_receipts(&api, &target).is_empty());
|
||||
assert!(!execution.attempts.lock().unwrap().iter().any(|attempt| attempt.0 == source.delivery_id));
|
||||
drop(unavailable);
|
||||
|
||||
// Reopen Backend persistence and run the unchanged pending-source scan. It
|
||||
// rereads the real Ticket, obtains findings, then reaches the actual Runtime.
|
||||
let mut reopened = api.clone();
|
||||
reopened.store = Arc::new(SqliteWorkspaceStore::open_for_test(&api.config.database_path).unwrap());
|
||||
let pending = reopened.store.list_pending_job_notifications(TEST_WORKSPACE_ID, 100).unwrap();
|
||||
assert_eq!(pending.len(), 1);
|
||||
assert_eq!(pending[0].delivery_id, source.delivery_id);
|
||||
assert_eq!(pending[0].job_id, source.job_id);
|
||||
assert_eq!(pending[0].attempt_id, source.attempt_id);
|
||||
notification_delivery::recover_job_notifications(&reopened);
|
||||
delivery_wait_idle(&reopened, &target, 1).await;
|
||||
assert_eq!(job_notification_evidence(&reopened, &source.delivery_id), ("deliver".into(), "result".into(), true, None));
|
||||
let pinned = reopened.store.job_notification_request(TEST_WORKSPACE_ID, &source.delivery_id).unwrap().unwrap();
|
||||
assert_eq!(pinned.1, target);
|
||||
assert!(pinned.0.contains("Quote: \"Writer-only scope\""));
|
||||
assert_eq!(delivery_commits(&reopened, &target), vec![pinned.0.clone()]);
|
||||
assert_eq!(delivery_receipts(&reopened, &target).len(), 1);
|
||||
assert!(execution.attempts.lock().unwrap().iter().any(|attempt| attempt.0 == source.delivery_id && attempt.1 == pinned.0));
|
||||
assert_eq!(source_snapshot(), original_source);
|
||||
assert_eq!(failures(), failed_once, "recovery must retain the original failure observation");
|
||||
let job = reopened.store.get_backend_job(TEST_WORKSPACE_ID, &request.job_id).unwrap().unwrap();
|
||||
assert_eq!(job.state, BackendJobState::Completed);
|
||||
assert_eq!(job.current_attempt, accepted.job.current_attempt);
|
||||
assert_eq!(job.result, accepted.job.result);
|
||||
assert_eq!(job.result_digest, accepted.job.result_digest);
|
||||
assert!(reopened.store.accept_backend_job_result(TEST_WORKSPACE_ID, worker, &submission, TEST_CREATED_AT).unwrap().replayed);
|
||||
notification_delivery::recover_job_notifications(&reopened);
|
||||
assert_eq!(delivery_receipts(&reopened, &target).len(), 1);
|
||||
assert!(reopened.store.list_pending_job_notifications(TEST_WORKSPACE_ID, 100).unwrap().is_empty());
|
||||
delivery_stop(&reopened, &target);
|
||||
delivery_stop(&reopened, worker);
|
||||
}
|
||||
|
||||
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
||||
async fn job_notification_stopped_fixed_recipient_is_not_replaced_or_restored() {
|
||||
let dir = tempfile::tempdir().unwrap();
|
||||
|
||||
Reference in New Issue
Block a user