Public release release-e7c2b3ac175d

Source commit: e7c2b3ac175db426791c02779841c5df1d0ffdff

Public tree identity: sha256:4f0b7cd828932c06d56f2406788f89b2050ef56c1e1a4f76f55341d387d78e46
This commit is contained in:
Clusterflux release 2026-07-29 02:53:56 +02:00
parent 9f43c6276a
commit cfd4f19da2
16 changed files with 1386 additions and 313 deletions

View file

@ -1,6 +1,7 @@
use std::collections::{BTreeMap, VecDeque}; use std::collections::{BTreeMap, VecDeque};
use std::future::Future; use std::future::Future;
use std::pin::Pin; use std::pin::Pin;
use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::{Arc, Mutex}; use std::sync::{Arc, Mutex};
use std::time::Duration; use std::time::Duration;
@ -33,13 +34,20 @@ pub trait ClientTransport: Send + Sync + 'static {
fn send(&self, request: TransportRequest) -> TransportFuture; fn send(&self, request: TransportRequest) -> TransportFuture;
} }
type ControlSessionPool = Arc<Vec<Mutex<Option<ControlSession>>>>;
pub struct ControlTransport { pub struct ControlTransport {
endpoint: String, endpoint: String,
connect_timeout: Duration, connect_timeout: Duration,
io_timeout: Duration, io_timeout: Duration,
sessions: Arc<Mutex<BTreeMap<String, ControlSession>>>, sessions: Arc<Mutex<BTreeMap<String, ControlSessionPool>>>,
next_session: Arc<AtomicU64>,
} }
// A small pool allows independent HTMX reads to progress concurrently while
// keeping connection and blocking-worker use strictly bounded.
const SESSIONS_PER_API_PATH: usize = 4;
impl ControlTransport { impl ControlTransport {
pub fn new(endpoint: impl Into<String>) -> Result<Self, ClientTransportError> { pub fn new(endpoint: impl Into<String>) -> Result<Self, ClientTransportError> {
Self::with_timeouts(endpoint, Duration::from_secs(10), Duration::from_secs(30)) Self::with_timeouts(endpoint, Duration::from_secs(10), Duration::from_secs(30))
@ -58,6 +66,7 @@ impl ControlTransport {
connect_timeout, connect_timeout,
io_timeout, io_timeout,
sessions: Arc::new(Mutex::new(BTreeMap::new())), sessions: Arc::new(Mutex::new(BTreeMap::new())),
next_session: Arc::new(AtomicU64::new(0)),
}) })
} }
} }
@ -68,35 +77,89 @@ impl ClientTransport for ControlTransport {
let connect_timeout = self.connect_timeout; let connect_timeout = self.connect_timeout;
let io_timeout = self.io_timeout; let io_timeout = self.io_timeout;
let sessions = Arc::clone(&self.sessions); let sessions = Arc::clone(&self.sessions);
let next_session = Arc::clone(&self.next_session);
Box::pin(async move { Box::pin(async move {
let pool = {
let mut sessions = sessions.lock().map_err(|_| {
ClientTransportError::Failed(
"client transport pool lock was poisoned".to_owned(),
)
})?;
Arc::clone(sessions.entry(request.api_path.clone()).or_insert_with(|| {
Arc::new(
(0..SESSIONS_PER_API_PATH)
.map(|_| Mutex::new(None))
.collect(),
)
}))
};
let slot_index =
next_session.fetch_add(1, Ordering::Relaxed) as usize % SESSIONS_PER_API_PATH;
tokio::task::spawn_blocking(move || { tokio::task::spawn_blocking(move || {
let value = serde_json::from_slice(&request.body) let value = serde_json::from_slice(&request.body)
.map_err(|error| ClientTransportError::Failed(error.to_string()))?; .map_err(|error| ClientTransportError::Failed(error.to_string()))?;
let mut sessions = sessions.lock().map_err(|_| { let mut selected = None;
for offset in 0..SESSIONS_PER_API_PATH {
let index = (slot_index + offset) % SESSIONS_PER_API_PATH;
match pool[index].try_lock() {
Ok(session) if session.is_some() => {
selected = Some(session);
break;
}
Ok(_) | Err(std::sync::TryLockError::WouldBlock) => {}
Err(std::sync::TryLockError::Poisoned(_)) => {
return Err(ClientTransportError::Failed(
"client transport session lock was poisoned".to_owned(),
));
}
}
}
if selected.is_none() {
for offset in 0..SESSIONS_PER_API_PATH {
let index = (slot_index + offset) % SESSIONS_PER_API_PATH;
match pool[index].try_lock() {
Ok(session) => {
selected = Some(session);
break;
}
Err(std::sync::TryLockError::WouldBlock) => {}
Err(std::sync::TryLockError::Poisoned(_)) => {
return Err(ClientTransportError::Failed(
"client transport session lock was poisoned".to_owned(),
));
}
}
}
}
let mut session = match selected {
Some(session) => session,
None => pool[slot_index].lock().map_err(|_| {
ClientTransportError::Failed( ClientTransportError::Failed(
"client transport session lock was poisoned".to_owned(), "client transport session lock was poisoned".to_owned(),
) )
})?; })?,
if !sessions.contains_key(&request.api_path) { };
let session = ControlSession::connect_to_api_path_with_timeouts( if session.is_none() {
*session = Some(
ControlSession::connect_to_api_path_with_timeouts(
&endpoint, &endpoint,
&request.api_path, &request.api_path,
connect_timeout, connect_timeout,
io_timeout, io_timeout,
) )
.map_err(|error| ClientTransportError::Failed(error.to_string()))?; .map_err(|error| ClientTransportError::Failed(error.to_string()))?,
sessions.insert(request.api_path.clone(), session); );
} }
let response = sessions let response = session
.get_mut(&request.api_path) .as_mut()
.expect("session was inserted for the requested API path") .expect("session was initialized for the requested API path")
.request(&value); .request(&value);
match response { match response {
Ok(response) => serde_json::to_vec(&response) Ok(response) => serde_json::to_vec(&response)
.map(|body| TransportResponse { body }) .map(|body| TransportResponse { body })
.map_err(|error| ClientTransportError::Failed(error.to_string())), .map_err(|error| ClientTransportError::Failed(error.to_string())),
Err(error) => { Err(error) => {
sessions.remove(&request.api_path); *session = None;
Err(ClientTransportError::Failed(error.to_string())) Err(ClientTransportError::Failed(error.to_string()))
} }
} }

View file

@ -1,9 +1,10 @@
use std::net::TcpListener; use std::net::TcpListener;
use std::time::Duration;
use clusterflux_client::{ use clusterflux_client::{
ApiErrorCategory, ApiErrorCode, ArtifactDownloadPoll, ArtifactId, ClientError, ApiErrorCategory, ApiErrorCode, ArtifactDownloadPoll, ArtifactId, ClientError,
ClusterfluxClient, MockTransport, ProjectId, SessionCredential, TenantId, UserId, ClusterfluxClient, ControlTransport, MockTransport, ProjectId, SessionCredential, TenantId,
CLIENT_API_VERSION, UserId, CLIENT_API_VERSION,
}; };
use clusterflux_control::CONTROL_API_PATH; use clusterflux_control::CONTROL_API_PATH;
use clusterflux_coordinator::service::CoordinatorService; use clusterflux_coordinator::service::CoordinatorService;
@ -181,3 +182,47 @@ async fn typed_client_runs_against_the_real_strict_control_endpoint() {
drop(client); drop(client);
server.join().unwrap(); server.join().unwrap();
} }
#[tokio::test]
async fn concurrent_requests_expand_the_bounded_pool_without_serializing() {
let listener = TcpListener::bind("127.0.0.1:0").unwrap();
let address = listener.local_addr().unwrap();
let server = std::thread::spawn(move || {
let handlers = (0..2)
.map(|_| {
let (stream, _) = listener.accept().unwrap();
std::thread::spawn(move || {
let mut service = CoordinatorService::new(7);
service
.issue_cli_session(
TenantId::from("tenant-one"),
ProjectId::from("project-one"),
UserId::from("user-one"),
"concurrent-session",
None,
)
.unwrap();
service.handle_stream(stream).unwrap();
})
})
.collect::<Vec<_>>();
for handler in handlers {
handler.join().unwrap();
}
});
let transport = ControlTransport::with_timeouts(
format!("clusterflux+tcp://{address}"),
Duration::from_secs(2),
Duration::from_secs(5),
)
.unwrap();
let client = ClusterfluxClient::with_transport(transport)
.with_session_credential(&SessionCredential::from_secret("concurrent-session"));
let (first, second) = tokio::join!(client.account_status(), client.account_status());
assert!(first.unwrap().authenticated);
assert!(second.unwrap().authenticated);
drop(client);
server.join().unwrap();
}

View file

@ -284,9 +284,10 @@ pub struct CoordinatorService {
process_summaries: BTreeMap<ProcessControlKey, summaries::StoredProcessSummary>, process_summaries: BTreeMap<ProcessControlKey, summaries::StoredProcessSummary>,
process_summary_order: VecDeque<ProcessControlKey>, process_summary_order: VecDeque<ProcessControlKey>,
next_process_summary_order: u64, next_process_summary_order: u64,
task_terminal_states: BTreeMap<TaskRestartKey, TaskTerminalState>,
recent_logs: BTreeMap<(TenantId, ProjectId), VecDeque<RecentLogEntry>>, recent_logs: BTreeMap<(TenantId, ProjectId), VecDeque<RecentLogEntry>>,
recent_log_dropped_through: BTreeMap<ProcessControlKey, u64>, recent_log_dropped_through: BTreeMap<ProcessControlKey, u64>,
recent_log_offsets: BTreeMap< recent_log_accounted_bytes: BTreeMap<
( (
TenantId, TenantId,
ProjectId, ProjectId,
@ -296,6 +297,13 @@ pub struct CoordinatorService {
), ),
u64, u64,
>, >,
recent_log_truncated_streams: BTreeSet<(
TenantId,
ProjectId,
ProcessId,
clusterflux_core::TaskInstanceId,
String,
)>,
next_recent_log_sequence: u64, next_recent_log_sequence: u64,
debug_audit_events: VecDeque<DebugAuditEvent>, debug_audit_events: VecDeque<DebugAuditEvent>,
debug_epochs: BTreeMap<ProcessControlKey, u64>, debug_epochs: BTreeMap<ProcessControlKey, u64>,
@ -566,9 +574,11 @@ impl CoordinatorService {
process_summaries: BTreeMap::new(), process_summaries: BTreeMap::new(),
process_summary_order: VecDeque::new(), process_summary_order: VecDeque::new(),
next_process_summary_order: 1, next_process_summary_order: 1,
task_terminal_states: BTreeMap::new(),
recent_logs: BTreeMap::new(), recent_logs: BTreeMap::new(),
recent_log_dropped_through: BTreeMap::new(), recent_log_dropped_through: BTreeMap::new(),
recent_log_offsets: BTreeMap::new(), recent_log_accounted_bytes: BTreeMap::new(),
recent_log_truncated_streams: BTreeSet::new(),
next_recent_log_sequence: 1, next_recent_log_sequence: 1,
debug_audit_events: VecDeque::new(), debug_audit_events: VecDeque::new(),
debug_epochs: BTreeMap::new(), debug_epochs: BTreeMap::new(),

View file

@ -39,40 +39,41 @@ impl CoordinatorService {
self.authorize_node_for_process_or_termination(&node, &tenant, &project, &process)?; self.authorize_node_for_process_or_termination(&node, &tenant, &project, &process)?;
validate_task_log_tail("stdout_tail", &stdout_tail)?; validate_task_log_tail("stdout_tail", &stdout_tail)?;
validate_task_log_tail("stderr_tail", &stderr_tail)?; validate_task_log_tail("stderr_tail", &stderr_tail)?;
let reported_bytes = checked_reported_log_bytes(stdout_bytes, stderr_bytes)?;
let now_epoch_seconds = self.current_epoch_seconds()?; let now_epoch_seconds = self.current_epoch_seconds()?;
let unaccounted_bytes = self.unaccounted_final_log_bytes(
&tenant,
&project,
&process,
&task,
stdout_bytes,
stderr_bytes,
)?;
self.quota self.quota
.can_charge_log_bytes(&tenant, &project, reported_bytes, now_epoch_seconds)?; .can_charge_log_bytes(&tenant, &project, unaccounted_bytes, now_epoch_seconds)?;
self.quota self.quota
.charge_log_bytes(&tenant, &project, reported_bytes, now_epoch_seconds)?; .charge_log_bytes(&tenant, &project, unaccounted_bytes, now_epoch_seconds)?;
let stdout_key = self.reconcile_final_log_stream(
recent_log_offset_key(&tenant, &project, &process, &task, &TaskLogStream::Stdout); &tenant,
let stderr_key = &project,
recent_log_offset_key(&tenant, &project, &process, &task, &TaskLogStream::Stderr); &process,
if !stdout_tail.is_empty() && !self.recent_log_offsets.contains_key(&stdout_key) { &task,
self.record_recent_log(
tenant.clone(),
project.clone(),
process.clone(),
task.clone(),
TaskLogStream::Stdout, TaskLogStream::Stdout,
stdout_tail.clone(), stdout_bytes,
&stdout_tail,
stdout_truncated, stdout_truncated,
now_epoch_seconds, now_epoch_seconds,
); );
} self.reconcile_final_log_stream(
if !stderr_tail.is_empty() && !self.recent_log_offsets.contains_key(&stderr_key) { &tenant,
self.record_recent_log( &project,
tenant.clone(), &process,
project.clone(), &task,
process.clone(),
task.clone(),
TaskLogStream::Stderr, TaskLogStream::Stderr,
stderr_tail.clone(), stderr_bytes,
&stderr_tail,
stderr_truncated, stderr_truncated,
now_epoch_seconds, now_epoch_seconds,
); );
}
Ok(CoordinatorResponse::TaskLogRecorded { Ok(CoordinatorResponse::TaskLogRecorded {
process, process,
task, task,
@ -129,13 +130,18 @@ impl CoordinatorService {
let task = TaskInstanceId::new(task); let task = TaskInstanceId::new(task);
self.authorize_node_for_process_or_termination(&node, &tenant, &project, &process)?; self.authorize_node_for_process_or_termination(&node, &tenant, &project, &process)?;
let key = recent_log_offset_key(&tenant, &project, &process, &task, &stream); let key = recent_log_offset_key(&tenant, &project, &process, &task, &stream);
let expected = self.recent_log_offsets.get(&key).copied().unwrap_or(0); let expected = self
.recent_log_accounted_bytes
.get(&key)
.copied()
.unwrap_or(0);
let end = offset.checked_add(source_bytes).ok_or_else(|| { let end = offset.checked_add(source_bytes).ok_or_else(|| {
CoordinatorServiceError::Protocol( CoordinatorServiceError::Protocol(
"live log chunk offset exceeds the supported range".to_owned(), "live log chunk offset exceeds the supported range".to_owned(),
) )
})?; })?;
if end <= expected { let state_marker = source_bytes == 0 && truncated;
if end < expected || (end == expected && !state_marker) {
return Ok(CoordinatorResponse::TaskLogChunkRecorded { return Ok(CoordinatorResponse::TaskLogChunkRecorded {
process, process,
task, task,
@ -144,6 +150,11 @@ impl CoordinatorService {
}); });
} }
let now_epoch_seconds = self.current_epoch_seconds()?; let now_epoch_seconds = self.current_epoch_seconds()?;
let newly_accounted = end.saturating_sub(expected);
self.quota
.can_charge_log_bytes(&tenant, &project, newly_accounted, now_epoch_seconds)?;
self.quota
.charge_log_bytes(&tenant, &project, newly_accounted, now_epoch_seconds)?;
if offset > expected { if offset > expected {
self.record_recent_log( self.record_recent_log(
tenant.clone(), tenant.clone(),
@ -156,14 +167,32 @@ impl CoordinatorService {
now_epoch_seconds, now_epoch_seconds,
); );
} else if offset < expected { } else if offset < expected {
return Ok(CoordinatorResponse::TaskLogChunkRecorded { self.record_recent_log(
process, tenant.clone(),
task, project.clone(),
sequence: None, process.clone(),
next_offset: expected, task.clone(),
}); stream.clone(),
format!(
"[log output overlap omitted: {} new source bytes]",
end - expected
),
true,
now_epoch_seconds,
);
} }
let sequence = (!text.is_empty()).then(|| { let marker_is_new = if truncated {
self.recent_log_truncated_streams.insert(key.clone())
} else {
false
};
let text = if state_marker && text.is_empty() {
"[log output truncated at source]".to_owned()
} else {
text
};
let sequence = (!text.is_empty() && offset >= expected && (!state_marker || marker_is_new))
.then(|| {
self.record_recent_log( self.record_recent_log(
tenant.clone(), tenant.clone(),
project.clone(), project.clone(),
@ -175,7 +204,9 @@ impl CoordinatorService {
now_epoch_seconds, now_epoch_seconds,
) )
}); });
self.recent_log_offsets.insert(key, end); if end > expected {
self.recent_log_accounted_bytes.insert(key, end);
}
Ok(CoordinatorResponse::TaskLogChunkRecorded { Ok(CoordinatorResponse::TaskLogChunkRecorded {
process, process,
task, task,
@ -302,20 +333,49 @@ impl CoordinatorService {
artifact_size_bytes, artifact_size_bytes,
result, result,
}; };
let reported_bytes = checked_reported_log_bytes(event.stdout_bytes, event.stderr_bytes)?;
let now_epoch_seconds = self.current_epoch_seconds()?; let now_epoch_seconds = self.current_epoch_seconds()?;
let unaccounted_bytes = self.unaccounted_final_log_bytes(
&event.tenant,
&event.project,
&event.process,
&event.task,
event.stdout_bytes,
event.stderr_bytes,
)?;
self.quota.can_charge_log_bytes( self.quota.can_charge_log_bytes(
&event.tenant, &event.tenant,
&event.project, &event.project,
reported_bytes, unaccounted_bytes,
now_epoch_seconds, now_epoch_seconds,
)?; )?;
self.quota.charge_log_bytes( self.quota.charge_log_bytes(
&event.tenant, &event.tenant,
&event.project, &event.project,
reported_bytes, unaccounted_bytes,
now_epoch_seconds, now_epoch_seconds,
)?; )?;
self.reconcile_final_log_stream(
&event.tenant,
&event.project,
&event.process,
&event.task,
TaskLogStream::Stdout,
event.stdout_bytes,
&event.stdout_tail,
event.stdout_truncated,
now_epoch_seconds,
);
self.reconcile_final_log_stream(
&event.tenant,
&event.project,
&event.process,
&event.task,
TaskLogStream::Stderr,
event.stderr_bytes,
&event.stderr_tail,
event.stderr_truncated,
now_epoch_seconds,
);
let task_key = task_control_key( let task_key = task_control_key(
&event.tenant, &event.tenant,
&event.project, &event.project,
@ -648,6 +708,22 @@ impl CoordinatorService {
pub(super) fn record_task_completion_event(&mut self, mut event: TaskCompletionEvent) { pub(super) fn record_task_completion_event(&mut self, mut event: TaskCompletionEvent) {
event.stdout_tail = bounded_log_tail(event.stdout_tail, &mut event.stdout_truncated); event.stdout_tail = bounded_log_tail(event.stdout_tail, &mut event.stdout_truncated);
event.stderr_tail = bounded_log_tail(event.stderr_tail, &mut event.stderr_truncated); event.stderr_tail = bounded_log_tail(event.stderr_tail, &mut event.stderr_truncated);
match event.executor {
super::TaskExecutor::CoordinatorMain => self.record_main_terminal_state(
&event.tenant,
&event.project,
&event.process,
event.task_definition.clone(),
event.task.clone(),
event.terminal_state.clone(),
),
super::TaskExecutor::Node => {
self.task_terminal_states.insert(
task_restart_key(&event.tenant, &event.project, &event.process, &event.task),
event.terminal_state.clone(),
);
}
}
let process_scope = ( let process_scope = (
event.tenant.clone(), event.tenant.clone(),
event.project.clone(), event.project.clone(),
@ -756,6 +832,171 @@ impl CoordinatorService {
sequence sequence
} }
#[allow(clippy::too_many_arguments)]
fn unaccounted_final_log_bytes(
&self,
tenant: &TenantId,
project: &ProjectId,
process: &ProcessId,
task: &TaskInstanceId,
stdout_bytes: u64,
stderr_bytes: u64,
) -> Result<u64, CoordinatorServiceError> {
let stdout_accounted = self
.recent_log_accounted_bytes
.get(&recent_log_offset_key(
tenant,
project,
process,
task,
&TaskLogStream::Stdout,
))
.copied()
.unwrap_or(0);
let stderr_accounted = self
.recent_log_accounted_bytes
.get(&recent_log_offset_key(
tenant,
project,
process,
task,
&TaskLogStream::Stderr,
))
.copied()
.unwrap_or(0);
let stdout_remaining = stdout_bytes.checked_sub(stdout_accounted).ok_or_else(|| {
CoordinatorServiceError::Protocol(format!(
"final stdout byte count {stdout_bytes} is below the {stdout_accounted} live bytes already accounted"
))
})?;
let stderr_remaining = stderr_bytes.checked_sub(stderr_accounted).ok_or_else(|| {
CoordinatorServiceError::Protocol(format!(
"final stderr byte count {stderr_bytes} is below the {stderr_accounted} live bytes already accounted"
))
})?;
checked_reported_log_bytes(stdout_remaining, stderr_remaining)
}
#[allow(clippy::too_many_arguments)]
fn reconcile_final_log_stream(
&mut self,
tenant: &TenantId,
project: &ProjectId,
process: &ProcessId,
task: &TaskInstanceId,
stream: TaskLogStream,
total_source_bytes: u64,
final_tail: &str,
source_truncated: bool,
now_epoch_seconds: u64,
) {
let key = recent_log_offset_key(tenant, project, process, task, &stream);
let accounted = self
.recent_log_accounted_bytes
.get(&key)
.copied()
.unwrap_or(0);
let mut visible_truncation = false;
if total_source_bytes > accounted {
let missing = total_source_bytes - accounted;
if final_tail.is_empty() {
self.record_recent_log(
tenant.clone(),
project.clone(),
process.clone(),
task.clone(),
stream.clone(),
format!("[log output unavailable: {missing} source bytes]"),
true,
now_epoch_seconds,
);
visible_truncation = true;
} else if (final_tail.len() as u64) <= total_source_bytes {
let tail_source_start = total_source_bytes - final_tail.len() as u64;
if accounted < tail_source_start {
self.record_recent_log(
tenant.clone(),
project.clone(),
process.clone(),
task.clone(),
stream.clone(),
format!(
"[log output lost before final tail: {} source bytes]",
tail_source_start - accounted
),
true,
now_epoch_seconds,
);
visible_truncation = true;
}
let source_start = accounted.max(tail_source_start) - tail_source_start;
let mut byte_start = usize::try_from(source_start)
.unwrap_or(final_tail.len())
.min(final_tail.len());
while byte_start < final_tail.len() && !final_tail.is_char_boundary(byte_start) {
byte_start += 1;
}
let suffix = &final_tail[byte_start..];
if !suffix.is_empty() {
self.record_recent_log(
tenant.clone(),
project.clone(),
process.clone(),
task.clone(),
stream.clone(),
suffix.to_owned(),
source_truncated || visible_truncation,
now_epoch_seconds,
);
visible_truncation |= source_truncated;
}
} else if accounted == 0 {
self.record_recent_log(
tenant.clone(),
project.clone(),
process.clone(),
task.clone(),
stream.clone(),
final_tail.to_owned(),
source_truncated,
now_epoch_seconds,
);
visible_truncation |= source_truncated;
} else {
self.record_recent_log(
tenant.clone(),
project.clone(),
process.clone(),
task.clone(),
stream.clone(),
format!(
"[{missing} additional source bytes could not be merged without duplicating redacted output]"
),
true,
now_epoch_seconds,
);
visible_truncation = true;
}
self.recent_log_accounted_bytes
.insert(key.clone(), total_source_bytes);
}
let marker_is_new = (source_truncated || visible_truncation)
&& self.recent_log_truncated_streams.insert(key);
if marker_is_new && !visible_truncation {
self.record_recent_log(
tenant.clone(),
project.clone(),
process.clone(),
task.clone(),
stream,
"[log output truncated at source]".to_owned(),
true,
now_epoch_seconds,
);
}
}
pub(super) fn clear_recent_log_offsets_for_task( pub(super) fn clear_recent_log_offsets_for_task(
&mut self, &mut self,
tenant: &TenantId, tenant: &TenantId,
@ -763,7 +1004,7 @@ impl CoordinatorService {
process: &ProcessId, process: &ProcessId,
task: &TaskInstanceId, task: &TaskInstanceId,
) { ) {
self.recent_log_offsets.retain( self.recent_log_accounted_bytes.retain(
|(entry_tenant, entry_project, entry_process, entry_task, _), _| { |(entry_tenant, entry_project, entry_process, entry_task, _), _| {
entry_tenant != tenant entry_tenant != tenant
|| entry_project != project || entry_project != project
@ -771,6 +1012,14 @@ impl CoordinatorService {
|| entry_task != task || entry_task != task
}, },
); );
self.recent_log_truncated_streams.retain(
|(entry_tenant, entry_project, entry_process, entry_task, _)| {
entry_tenant != tenant
|| entry_project != project
|| entry_process != process
|| entry_task != task
},
);
} }
fn finish_task_attempt(&mut self, event: &mut TaskCompletionEvent) -> bool { fn finish_task_attempt(&mut self, event: &mut TaskCompletionEvent) -> bool {
@ -856,13 +1105,11 @@ impl CoordinatorService {
return Ok(false); return Ok(false);
} }
let main_completed = self.task_events.iter().rev().any(|event| { let main_completed = self
&event.tenant == tenant .process_summaries
&& &event.project == project .get(&process_key)
&& &event.process == process .and_then(|summary| summary.main_terminal_state.as_ref())
&& matches!(event.executor, super::TaskExecutor::CoordinatorMain) .is_some_and(|state| matches!(state, TaskTerminalState::Completed));
&& matches!(event.terminal_state, TaskTerminalState::Completed)
});
let cancellation_completed = self.process_cancellations.contains(&process_key); let cancellation_completed = self.process_cancellations.contains(&process_key);
if !main_completed && !cancellation_completed { if !main_completed && !cancellation_completed {
return Ok(false); return Ok(false);
@ -879,13 +1126,24 @@ impl CoordinatorService {
} }
let final_result = if cancellation_completed { let final_result = if cancellation_completed {
super::ProcessFinalResult::Cancelled super::ProcessFinalResult::Cancelled
} else if self.task_events.iter().any(|event| { } else if self.task_terminal_states.iter().any(
&event.tenant == tenant |((task_tenant, task_project, task_process, _), terminal_state)| {
&& &event.project == project task_tenant == tenant
&& &event.process == process && task_project == project
&& event.terminal_state == TaskTerminalState::Failed && task_process == process
}) { && terminal_state == &TaskTerminalState::Failed
},
) {
super::ProcessFinalResult::Failed super::ProcessFinalResult::Failed
} else if self.task_terminal_states.iter().any(
|((task_tenant, task_project, task_process, _), terminal_state)| {
task_tenant == tenant
&& task_project == project
&& task_process == process
&& terminal_state == &TaskTerminalState::Cancelled
},
) {
super::ProcessFinalResult::Cancelled
} else { } else {
super::ProcessFinalResult::Completed super::ProcessFinalResult::Completed
}; };

View file

@ -671,7 +671,9 @@ impl CoordinatorService {
)); ));
}; };
self.task_attempts.remove(&removable); self.task_attempts.remove(&removable);
self.task_terminal_states.remove(&removable);
} }
self.task_terminal_states.remove(&key);
let attempts = self.task_attempts.entry(key).or_default(); let attempts = self.task_attempts.entry(key).or_default();
for attempt in attempts.iter_mut() { for attempt in attempts.iter_mut() {
attempt.current = false; attempt.current = false;

View file

@ -382,6 +382,10 @@ impl CoordinatorService {
|| attempt_project != &project || attempt_project != &project
|| attempt_process != &process || attempt_process != &process
}); });
self.task_terminal_states
.retain(|(task_tenant, task_project, task_process, _), _| {
task_tenant != &tenant || task_project != &project || task_process != &process
});
self.restart_launches self.restart_launches
.retain(|(attempt_tenant, attempt_project, attempt_process, _)| { .retain(|(attempt_tenant, attempt_project, attempt_process, _)| {
attempt_tenant != &tenant attempt_tenant != &tenant
@ -671,10 +675,18 @@ impl CoordinatorService {
.map(|active| { .map(|active| {
let process_key = process_control_key(&active.tenant, &active.project, &active.id); let process_key = process_control_key(&active.tenant, &active.project, &active.id);
let main = self.main_runtime.controls.get(&process_key); let main = self.main_runtime.controls.get(&process_key);
let stored = self.process_summaries.get(&process_key);
let stored_main_state = stored
.and_then(|summary| summary.main_terminal_state.as_ref())
.map(|state| match state {
super::TaskTerminalState::Completed => "completed",
super::TaskTerminalState::Failed => "failed",
super::TaskTerminalState::Cancelled => "cancelled",
});
let state = if self.process_cancellations.contains(&process_key) { let state = if self.process_cancellations.contains(&process_key) {
"cancelling" "cancelling"
} else { } else {
main.map_or("running", |main| main.state.as_str()) "running"
}; };
let main_wait_state = main.and_then(|main| { let main_wait_state = main.and_then(|main| {
if main.state != "running" { if main.state != "running" {
@ -699,9 +711,15 @@ impl CoordinatorService {
VirtualProcessStatus { VirtualProcessStatus {
process: active.id, process: active.id,
state: state.to_owned(), state: state.to_owned(),
main_task_definition: main.map(|main| main.task_definition.clone()), main_task_definition: main.map(|main| main.task_definition.clone()).or_else(
main_task_instance: main.map(|main| main.task_instance.clone()), || stored.and_then(|summary| summary.main_task_definition.clone()),
main_state: main.map(|main| main.state.clone()), ),
main_task_instance: main
.map(|main| main.task_instance.clone())
.or_else(|| stored.and_then(|summary| summary.main_task_instance.clone())),
main_state: main
.map(|main| main.state.clone())
.or_else(|| stored_main_state.map(str::to_owned)),
main_wait_state, main_wait_state,
main_debug_epoch: main.and_then(|main| main.debug.requested_epoch()), main_debug_epoch: main.and_then(|main| main.debug.requested_epoch()),
connected_nodes: active.connected_nodes.into_iter().collect(), connected_nodes: active.connected_nodes.into_iter().collect(),

View file

@ -2,7 +2,8 @@ use std::collections::BTreeSet;
use std::time::Instant; use std::time::Instant;
use clusterflux_core::{ use clusterflux_core::{
ArtifactId, ArtifactMetadata, NodeId, ProcessId, ProjectId, TenantId, UserId, ArtifactId, ArtifactMetadata, NodeId, ProcessId, ProjectId, TaskDefinitionId, TaskInstanceId,
TenantId, UserId,
}; };
use super::keys::{process_control_key, ProcessControlKey}; use super::keys::{process_control_key, ProcessControlKey};
@ -19,6 +20,9 @@ pub(super) struct StoredProcessSummary {
pub(super) ended_at_epoch_seconds: Option<u64>, pub(super) ended_at_epoch_seconds: Option<u64>,
pub(super) final_result: Option<ProcessFinalResult>, pub(super) final_result: Option<ProcessFinalResult>,
pub(super) connected_nodes: Vec<NodeId>, pub(super) connected_nodes: Vec<NodeId>,
pub(super) main_task_definition: Option<TaskDefinitionId>,
pub(super) main_task_instance: Option<TaskInstanceId>,
pub(super) main_terminal_state: Option<super::TaskTerminalState>,
pub(super) order: u64, pub(super) order: u64,
} }
@ -84,10 +88,16 @@ impl CoordinatorService {
} }
} }
self.recent_log_dropped_through.remove(&key); self.recent_log_dropped_through.remove(&key);
self.recent_log_offsets self.recent_log_accounted_bytes.retain(
.retain(|(entry_tenant, entry_project, entry_process, _, _), _| { |(entry_tenant, entry_project, entry_process, _, _), _| {
entry_tenant != tenant || entry_project != project || entry_process != process entry_tenant != tenant || entry_project != project || entry_process != process
}); },
);
self.recent_log_truncated_streams.retain(
|(entry_tenant, entry_project, entry_process, _, _)| {
entry_tenant != tenant || entry_project != project || entry_process != process
},
);
self.process_summary_order self.process_summary_order
.retain(|retained| retained != &key); .retain(|retained| retained != &key);
self.evict_process_summaries_for_project(tenant, project); self.evict_process_summaries_for_project(tenant, project);
@ -101,6 +111,9 @@ impl CoordinatorService {
ended_at_epoch_seconds: None, ended_at_epoch_seconds: None,
final_result: None, final_result: None,
connected_nodes: Vec::new(), connected_nodes: Vec::new(),
main_task_definition: None,
main_task_instance: None,
main_terminal_state: None,
order, order,
}, },
); );
@ -133,6 +146,9 @@ impl CoordinatorService {
ended_at_epoch_seconds: None, ended_at_epoch_seconds: None,
final_result: None, final_result: None,
connected_nodes: Vec::new(), connected_nodes: Vec::new(),
main_task_definition: None,
main_task_instance: None,
main_terminal_state: None,
order, order,
} }
}); });
@ -141,6 +157,31 @@ impl CoordinatorService {
entry.connected_nodes = connected_nodes; entry.connected_nodes = connected_nodes;
} }
pub(super) fn record_main_terminal_state(
&mut self,
tenant: &TenantId,
project: &ProjectId,
process: &ProcessId,
task_definition: TaskDefinitionId,
task_instance: TaskInstanceId,
terminal_state: super::TaskTerminalState,
) {
let key = process_control_key(tenant, project, process);
if !self.process_summaries.contains_key(&key) {
self.record_process_started(
tenant,
project,
process,
self.liveness_now_epoch_seconds(),
);
}
if let Some(summary) = self.process_summaries.get_mut(&key) {
summary.main_task_definition = Some(task_definition);
summary.main_task_instance = Some(task_instance);
summary.main_terminal_state = Some(terminal_state);
}
}
pub(super) fn handle_list_process_summaries( pub(super) fn handle_list_process_summaries(
&mut self, &mut self,
tenant: String, tenant: String,
@ -449,10 +490,7 @@ impl CoordinatorService {
let Some(candidate) = candidate.cloned() else { let Some(candidate) = candidate.cloned() else {
break; break;
}; };
self.process_summaries.remove(&candidate); self.remove_process_summary_state(&candidate);
self.recent_log_dropped_through.remove(&candidate);
self.process_summary_order
.retain(|retained| retained != &candidate);
} }
} }
@ -466,12 +504,30 @@ impl CoordinatorService {
let Some(candidate) = candidate.cloned() else { let Some(candidate) = candidate.cloned() else {
break; break;
}; };
self.process_summaries.remove(&candidate); self.remove_process_summary_state(&candidate);
self.recent_log_dropped_through.remove(&candidate);
self.process_summary_order
.retain(|retained| retained != &candidate);
} }
} }
fn remove_process_summary_state(&mut self, key: &super::ProcessControlKey) {
self.process_summaries.remove(key);
self.recent_log_dropped_through.remove(key);
if let Some(logs) = self.recent_logs.get_mut(&(key.0.clone(), key.1.clone())) {
logs.retain(|entry| entry.process != key.2);
if logs.is_empty() {
self.recent_logs.remove(&(key.0.clone(), key.1.clone()));
}
}
self.recent_log_accounted_bytes
.retain(|(tenant, project, process, _, _), _| {
tenant != &key.0 || project != &key.1 || process != &key.2
});
self.recent_log_truncated_streams
.retain(|(tenant, project, process, _, _)| {
tenant != &key.0 || project != &key.1 || process != &key.2
});
self.process_summary_order
.retain(|retained| retained != key);
}
} }
fn parse_order_cursor( fn parse_order_cursor(

View file

@ -2501,10 +2501,10 @@ fn signed_node_log_ingestion_checks_scoped_quota_before_accepting_bytes() {
process: "process".to_owned(), process: "process".to_owned(),
node: "node".to_owned(), node: "node".to_owned(),
task: "task".to_owned(), task: "task".to_owned(),
stdout_bytes: 1, stdout_bytes: 4,
stderr_bytes: 0, stderr_bytes: 1,
stdout_tail: "x".to_owned(), stdout_tail: "outx".to_owned(),
stderr_tail: String::new(), stderr_tail: "e".to_owned(),
stdout_truncated: false, stdout_truncated: false,
stderr_truncated: false, stderr_truncated: false,
backpressured: true, backpressured: true,
@ -4474,6 +4474,63 @@ fn completed_main_unpolled_final_assignment_completion_retires_process() {
.expect("unpolled terminal completion must release the one-process slot"); .expect("unpolled terminal completion must release the one-process slot");
} }
#[test]
fn completed_main_retires_from_authoritative_state_after_event_history_rotates() {
let mut service =
service_with_completed_main_and_final_child(clusterflux_core::TaskFailurePolicy::FailFast);
let tenant = TenantId::from("tenant");
let project = ProjectId::from("project");
let process = ProcessId::from("terminal-matrix");
for index in 0..=MAX_TASK_EVENTS_PER_PROCESS {
service.record_task_completion_event(TaskCompletionEvent {
tenant: tenant.clone(),
project: project.clone(),
process: process.clone(),
node: NodeId::from("worker"),
executor: TaskExecutor::Node,
task_definition: TaskDefinitionId::from("historical"),
task: TaskInstanceId::new(format!("historical-{index}")),
attempt_id: None,
placement: None,
terminal_state: TaskTerminalState::Completed,
status_code: Some(0),
stdout_bytes: 0,
stderr_bytes: 0,
stdout_tail: String::new(),
stderr_tail: String::new(),
stdout_truncated: false,
stderr_truncated: false,
artifact_path: None,
artifact_digest: None,
artifact_size_bytes: None,
result: None,
});
}
assert!(
service.task_events.iter().all(|event| {
event.process != process || event.executor != TaskExecutor::CoordinatorMain
}),
"the regression requires bounded history to have rotated the main event"
);
complete_terminal_matrix_child(&mut service, TaskTerminalState::Completed);
assert!(service
.coordinator
.active_process(&tenant, &project, &process)
.is_none());
let summary = service
.process_summaries
.get(&process_control_key(&tenant, &project, &process))
.expect("the terminal process summary must remain authoritative");
assert_eq!(summary.final_result, Some(ProcessFinalResult::Completed));
assert_eq!(
summary.main_terminal_state,
Some(TaskTerminalState::Completed)
);
}
#[test] #[test]
fn completed_main_await_operator_blocks_retirement_until_each_resolution() { fn completed_main_await_operator_blocks_retirement_until_each_resolution() {
for resolution in [ for resolution in [
@ -4634,6 +4691,14 @@ fn completed_main_failed_child_restarted_successfully_retires_with_successful_cu
assert!(attempts assert!(attempts
.iter() .iter()
.any(|attempt| attempt.current && attempt.state == TaskAttemptState::Completed)); .any(|attempt| attempt.current && attempt.state == TaskAttemptState::Completed));
assert_eq!(
service
.process_summaries
.get(&process_control_key(&tenant, &project, &process))
.and_then(|summary| summary.final_result.clone()),
Some(ProcessFinalResult::Completed),
"a successful current retry must override the superseded failed attempt"
);
assert_eq!( assert_eq!(
service service
.task_join_result(tenant, project, process, task) .task_join_result(tenant, project, process, task)
@ -8182,6 +8247,54 @@ fn web_process_summaries_are_scoped_paginated_and_retain_authoritative_terminal_
assert!(oversized.to_string().contains("100")); assert!(oversized.to_string().contains("100"));
} }
#[test]
fn process_summary_eviction_releases_live_log_accounting_state() {
let mut service = CoordinatorService::new(7);
let tenant = TenantId::from("tenant-summary-bound");
let project = ProjectId::from("project-summary-bound");
let task = TaskInstanceId::from("task");
for index in 0..MAX_RECENT_PROCESS_SUMMARIES_PER_PROJECT {
let process = ProcessId::new(format!("process-{index:03}"));
service.record_process_started(&tenant, &project, &process, index as u64);
service.record_process_terminal(
&tenant,
&project,
&process,
ProcessFinalResult::Completed,
index as u64 + 1,
);
let key = (
tenant.clone(),
project.clone(),
process,
task.clone(),
"stdout".to_owned(),
);
service.recent_log_accounted_bytes.insert(key.clone(), 10);
service.recent_log_truncated_streams.insert(key);
}
let evicted = ProcessId::from("process-000");
service.record_process_started(&tenant, &project, &ProcessId::from("process-next"), 1_000);
assert!(!service.process_summaries.contains_key(&(
tenant.clone(),
project.clone(),
evicted.clone()
)));
assert!(!service.recent_log_accounted_bytes.keys().any(
|(entry_tenant, entry_project, process, _, _)| {
entry_tenant == &tenant && entry_project == &project && process == &evicted
}
));
assert!(!service.recent_log_truncated_streams.iter().any(
|(entry_tenant, entry_project, process, _, _)| {
entry_tenant == &tenant && entry_project == &project && process == &evicted
}
));
}
#[test] #[test]
fn web_node_summaries_are_scoped_paginated_and_hard_bounded() { fn web_node_summaries_are_scoped_paginated_and_hard_bounded() {
let mut service = CoordinatorService::new(7); let mut service = CoordinatorService::new(7);
@ -8560,6 +8673,15 @@ fn web_recent_logs_are_signed_scoped_cursor_safe_and_memory_bounded() {
else { else {
panic!("expected first live log sequence"); panic!("expected first live log sequence");
}; };
assert_eq!(
service.quota.used_log_bytes(
&TenantId::from("tenant-a"),
&ProjectId::from("project-a"),
100,
),
5,
"live bytes must be charged when accepted"
);
let retry = service let retry = service
.handle_signed_node_request_auto(CoordinatorRequest::ReportTaskLogChunk { .handle_signed_node_request_auto(CoordinatorRequest::ReportTaskLogChunk {
tenant: "tenant-a".to_owned(), tenant: "tenant-a".to_owned(),
@ -8578,6 +8700,15 @@ fn web_recent_logs_are_signed_scoped_cursor_safe_and_memory_bounded() {
retry, retry,
CoordinatorResponse::TaskLogChunkRecorded { sequence: None, .. } CoordinatorResponse::TaskLogChunkRecorded { sequence: None, .. }
)); ));
assert_eq!(
service.quota.used_log_bytes(
&TenantId::from("tenant-a"),
&ProjectId::from("project-a"),
100,
),
5,
"a retried chunk must not be charged twice"
);
service service
.handle_signed_node_request_auto(CoordinatorRequest::ReportTaskLogChunk { .handle_signed_node_request_auto(CoordinatorRequest::ReportTaskLogChunk {
tenant: "tenant-a".to_owned(), tenant: "tenant-a".to_owned(),
@ -8592,6 +8723,15 @@ fn web_recent_logs_are_signed_scoped_cursor_safe_and_memory_bounded() {
truncated: false, truncated: false,
}) })
.unwrap(); .unwrap();
assert_eq!(
service.quota.used_log_bytes(
&TenantId::from("tenant-a"),
&ProjectId::from("project-a"),
100,
),
10,
"a gap and the delivered bytes must both count toward source-byte usage"
);
let CoordinatorResponse::RecentLogs { let CoordinatorResponse::RecentLogs {
entries, entries,
@ -8633,6 +8773,113 @@ fn web_recent_logs_are_signed_scoped_cursor_safe_and_memory_bounded() {
assert_eq!(entries.len(), 1); assert_eq!(entries.len(), 1);
assert_eq!(entries[0].text, "ok"); assert_eq!(entries[0].text, "ok");
service
.handle_report_task_log(
"tenant-a".to_owned(),
"project-a".to_owned(),
"process-shared".to_owned(),
"node-a".to_owned(),
"task-one".to_owned(),
12,
0,
"hello???okZZ".to_owned(),
String::new(),
false,
false,
false,
)
.unwrap();
assert_eq!(
service.quota.used_log_bytes(
&TenantId::from("tenant-a"),
&ProjectId::from("project-a"),
100,
),
12,
"the final summary must charge only source bytes not already charged live"
);
assert_eq!(
service
.recent_logs
.get(&(TenantId::from("tenant-a"), ProjectId::from("project-a")))
.unwrap()
.back()
.unwrap()
.text,
"ZZ",
"final-tail reconciliation must append only the nonduplicating suffix"
);
service
.handle_report_task_log(
"tenant-a".to_owned(),
"project-a".to_owned(),
"process-shared".to_owned(),
"node-a".to_owned(),
"task-one".to_owned(),
12,
0,
"hello???okZZ".to_owned(),
String::new(),
false,
false,
false,
)
.unwrap();
assert_eq!(
service.quota.used_log_bytes(
&TenantId::from("tenant-a"),
&ProjectId::from("project-a"),
100,
),
12,
"replayed final accounting must be idempotent"
);
let marker = service
.handle_report_task_log_chunk(
"tenant-a".to_owned(),
"project-a".to_owned(),
"process-shared".to_owned(),
"node-a".to_owned(),
"task-one".to_owned(),
TaskLogStream::Stdout,
12,
0,
"[log output truncated at node capture limit]".to_owned(),
true,
)
.unwrap();
assert!(matches!(
marker,
CoordinatorResponse::TaskLogChunkRecorded {
sequence: Some(_),
next_offset: 12,
..
}
));
let repeated_marker = service
.handle_report_task_log_chunk(
"tenant-a".to_owned(),
"project-a".to_owned(),
"process-shared".to_owned(),
"node-a".to_owned(),
"task-one".to_owned(),
TaskLogStream::Stdout,
12,
0,
"[log output truncated at node capture limit]".to_owned(),
true,
)
.unwrap();
assert!(matches!(
repeated_marker,
CoordinatorResponse::TaskLogChunkRecorded {
sequence: None,
next_offset: 12,
..
}
));
let cross_tenant = service let cross_tenant = service
.handle_request(CoordinatorRequest::Authenticated { .handle_request(CoordinatorRequest::Authenticated {
session_secret: "session-b".to_owned(), session_secret: "session-b".to_owned(),

View file

@ -195,7 +195,34 @@ impl ArtifactRegistry {
protected_processes: &BTreeSet<ProcessId>, protected_processes: &BTreeSet<ProcessId>,
) -> Result<ArtifactMetadata, String> { ) -> Result<ArtifactMetadata, String> {
let key = ArtifactScopeKey::from_refs(&flush.tenant, &flush.project, &flush.id); let key = ArtifactScopeKey::from_refs(&flush.tenant, &flush.project, &flush.id);
self.next_epoch += 1; let replacing = self.artifacts.contains_key(&key);
let retained_for_project = self
.artifacts
.values()
.filter(|metadata| metadata.tenant == flush.tenant && metadata.project == flush.project)
.count();
let eviction = if replacing || retained_for_project < MAX_ARTIFACT_METADATA_PER_PROJECT {
None
} else {
let mut protected_processes = protected_processes.clone();
protected_processes.insert(flush.process.clone());
Some(
self.project_metadata_eviction_candidate(
&flush.tenant,
&flush.project,
pinned,
&protected_processes,
)
.ok_or_else(|| {
format!(
"artifact metadata capacity of {MAX_ARTIFACT_METADATA_PER_PROJECT} is \
exhausted by active or retained artifacts"
)
})?,
)
};
self.next_epoch = self.next_epoch.saturating_add(1);
let metadata = ArtifactMetadata { let metadata = ArtifactMetadata {
id: flush.id.clone(), id: flush.id.clone(),
tenant: flush.tenant, tenant: flush.tenant,
@ -210,15 +237,10 @@ impl ArtifactRegistry {
explicit_locations: Vec::new(), explicit_locations: Vec::new(),
coordinator_has_large_bytes: false, coordinator_has_large_bytes: false,
}; };
if let Some(eviction) = eviction {
self.artifacts.remove(&eviction);
}
self.artifacts.insert(key, metadata.clone()); self.artifacts.insert(key, metadata.clone());
let mut protected_processes = protected_processes.clone();
protected_processes.insert(metadata.process.clone());
self.enforce_project_metadata_limit(
&metadata.tenant,
&metadata.project,
pinned,
&protected_processes,
);
Ok(metadata) Ok(metadata)
} }
@ -237,8 +259,29 @@ impl ArtifactRegistry {
.count() .count()
> MAX_ARTIFACT_METADATA_PER_PROJECT > MAX_ARTIFACT_METADATA_PER_PROJECT
{ {
let candidate = self let candidate = self.project_metadata_eviction_candidate(
.artifacts tenant,
project,
pinned,
protected_processes,
);
let Some(candidate) = candidate else {
break;
};
self.artifacts.remove(&candidate);
evicted += 1;
}
evicted
}
fn project_metadata_eviction_candidate(
&self,
tenant: &TenantId,
project: &ProjectId,
pinned: &BTreeSet<ArtifactScopeKey>,
protected_processes: &BTreeSet<ProcessId>,
) -> Option<ArtifactScopeKey> {
self.artifacts
.values() .values()
.filter(|metadata| { .filter(|metadata| {
&metadata.tenant == tenant &metadata.tenant == tenant
@ -260,14 +303,7 @@ impl ArtifactRegistry {
.min_by_key(|metadata| metadata.flushed_epoch) .min_by_key(|metadata| metadata.flushed_epoch)
.map(|metadata| { .map(|metadata| {
ArtifactScopeKey::from_refs(&metadata.tenant, &metadata.project, &metadata.id) ArtifactScopeKey::from_refs(&metadata.tenant, &metadata.project, &metadata.id)
}); })
let Some(candidate) = candidate else {
break;
};
self.artifacts.remove(&candidate);
evicted += 1;
}
evicted
} }
pub fn sync_to_explicit_store( pub fn sync_to_explicit_store(
@ -1485,6 +1521,87 @@ mod tests {
); );
} }
#[test]
fn artifact_metadata_capacity_rejects_atomically_when_every_entry_is_protected() {
let mut registry = ArtifactRegistry::default();
let tenant = TenantId::from("tenant");
let project = ProjectId::from("project");
let active_process = ProcessId::from("active-process");
for index in 0..MAX_ARTIFACT_METADATA_PER_PROJECT {
registry.flush_metadata(ArtifactFlush {
id: ArtifactId::new(format!("artifact-{index}")),
tenant: tenant.clone(),
project: project.clone(),
process: active_process.clone(),
producer_task: TaskInstanceId::new(format!("task-{index}")),
retaining_node: NodeId::from("node"),
digest: Digest::sha256(format!("content-{index}")),
size: 1,
});
}
let replacement = ArtifactFlush {
id: ArtifactId::from("artifact-0"),
tenant: tenant.clone(),
project: project.clone(),
process: active_process.clone(),
producer_task: TaskInstanceId::from("replacement-task"),
retaining_node: NodeId::from("node"),
digest: Digest::sha256("replacement"),
size: 2,
};
registry
.flush_metadata_with_protected_processes(
replacement,
&BTreeSet::new(),
&BTreeSet::from([active_process.clone()]),
)
.expect("replacement must remain possible at the metadata bound");
assert_eq!(
registry
.metadata(&tenant, &project, &ArtifactId::from("artifact-0"))
.unwrap()
.digest,
Digest::sha256("replacement")
);
let epoch_before_rejection = registry.next_epoch;
let result = registry.flush_metadata_with_protected_processes(
ArtifactFlush {
id: ArtifactId::from("artifact-rejected"),
tenant: tenant.clone(),
project: project.clone(),
process: active_process.clone(),
producer_task: TaskInstanceId::from("rejected-task"),
retaining_node: NodeId::from("node"),
digest: Digest::sha256("rejected"),
size: 3,
},
&BTreeSet::new(),
&BTreeSet::from([active_process]),
);
assert!(result
.unwrap_err()
.contains("exhausted by active or retained artifacts"));
assert_eq!(registry.next_epoch, epoch_before_rejection);
assert_eq!(
registry.metadata_for_project(&tenant, &project).count(),
MAX_ARTIFACT_METADATA_PER_PROJECT
);
assert!(registry
.metadata(&tenant, &project, &ArtifactId::from("artifact-rejected"))
.is_none());
assert_eq!(
registry
.metadata(&tenant, &project, &ArtifactId::from("artifact-0"))
.unwrap()
.digest,
Digest::sha256("replacement"),
"rejection must not modify existing metadata"
);
}
#[test] #[test]
fn download_stream_accounts_usage_before_and_during_streaming() { fn download_stream_accounts_usage_before_and_during_streaming() {
let mut registry = registry_with_artifact(); let mut registry = registry_with_artifact();

View file

@ -742,14 +742,16 @@ fn fetch_current_process_status(
client_user_request( client_user_request(
state, state,
json!({ json!({
"type": "list_processes", "type": "list_process_summaries",
"tenant": state.tenant, "tenant": state.tenant,
"project": state.project_id, "project": state.project_id,
"actor_user": state.actor_user, "actor_user": state.actor_user,
"cursor": null,
"limit": 100,
}), }),
), ),
)?; )?;
let current = statuses let summary = statuses
.get("processes") .get("processes")
.and_then(Value::as_array) .and_then(Value::as_array)
.and_then(|processes| { .and_then(|processes| {
@ -758,6 +760,9 @@ fn fetch_current_process_status(
}) })
}) })
.cloned(); .cloned();
let current = merge_active_process_status(state, summary, |request| {
coordinator_request(coordinator, request)
})?;
Ok((statuses, current)) Ok((statuses, current))
} }
@ -768,13 +773,15 @@ fn fetch_current_process_status_in(
let statuses = session.request(client_user_request( let statuses = session.request(client_user_request(
state, state,
json!({ json!({
"type": "list_processes", "type": "list_process_summaries",
"tenant": state.tenant, "tenant": state.tenant,
"project": state.project_id, "project": state.project_id,
"actor_user": state.actor_user, "actor_user": state.actor_user,
"cursor": null,
"limit": 100,
}), }),
))?; ))?;
let current = statuses let summary = statuses
.get("processes") .get("processes")
.and_then(Value::as_array) .and_then(Value::as_array)
.and_then(|processes| { .and_then(|processes| {
@ -783,9 +790,51 @@ fn fetch_current_process_status_in(
}) })
}) })
.cloned(); .cloned();
let current = merge_active_process_status(state, summary, |request| session.request(request))?;
Ok((statuses, current)) Ok((statuses, current))
} }
fn merge_active_process_status<F>(
state: &AdapterState,
summary: Option<Value>,
mut request: F,
) -> Result<Option<Value>>
where
F: FnMut(Value) -> Result<Value>,
{
let Some(mut summary) = summary else {
return Ok(None);
};
if summary.get("lifecycle").and_then(Value::as_str) != Some("active") {
return Ok(Some(summary));
}
let active_statuses = request(client_user_request(
state,
json!({
"type": "list_processes",
"tenant": state.tenant,
"project": state.project_id,
"actor_user": state.actor_user,
}),
))?;
let active = active_statuses
.get("processes")
.and_then(Value::as_array)
.and_then(|processes| {
processes.iter().find(|process| {
process.get("process").and_then(Value::as_str) == Some(state.process.as_str())
})
});
if let (Some(summary), Some(active)) =
(summary.as_object_mut(), active.and_then(Value::as_object))
{
for (key, value) in active {
summary.entry(key.clone()).or_insert_with(|| value.clone());
}
}
Ok(Some(summary))
}
pub(crate) fn relaunch_services_main_runtime(state: &AdapterState) -> Result<RuntimeLaunchRecord> { pub(crate) fn relaunch_services_main_runtime(state: &AdapterState) -> Result<RuntimeLaunchRecord> {
let coordinator = let coordinator =
crate::view_state::normalize_coordinator_endpoint(&state.coordinator_endpoint); crate::view_state::normalize_coordinator_endpoint(&state.coordinator_endpoint);
@ -859,27 +908,7 @@ pub(crate) fn attach_services_runtime(state: &AdapterState) -> Result<RuntimeLau
.count() .count()
}) })
.unwrap_or(0); .unwrap_or(0);
let process_statuses = coordinator_request( let (process_statuses, process_status) = fetch_current_process_status(&coordinator, state)?;
&coordinator,
client_user_request(
state,
json!({
"type": "list_processes",
"tenant": state.tenant,
"project": state.project_id,
"actor_user": state.actor_user,
}),
),
)?;
let process_status = process_statuses
.get("processes")
.and_then(Value::as_array)
.and_then(|processes| {
processes.iter().find(|process| {
process.get("process").and_then(Value::as_str) == Some(state.process.as_str())
})
})
.cloned();
let node = process_status let node = process_status
.as_ref() .as_ref()
.and_then(|status| status.get("connected_nodes")) .and_then(|status| status.get("connected_nodes"))
@ -1079,9 +1108,8 @@ pub(crate) fn observe_services_runtime(
} }
} }
} }
// Terminal task events remain inspectable after the active process slot is // Events remain useful for output and attempt details, but terminal
// released. Read them before active-process debug state so a fast main exit // authority comes from the durable process summary read below.
// cannot turn a successful continuation into an authorization error.
let current_session = session.as_mut().expect("observer session connected"); let current_session = session.as_mut().expect("observer session connected");
let events = match current_session.request(client_user_request( let events = match current_session.request(client_user_request(
state, state,
@ -1118,65 +1146,6 @@ pub(crate) fn observe_services_runtime(
std::thread::sleep(reconnect_delay); std::thread::sleep(reconnect_delay);
continue; continue;
} }
if let Some(mut outcome) = terminal_runtime_outcome(&coordinator, state, &events) {
let snapshot_request = if inject_snapshot_failure && !snapshot_failure_injected {
snapshot_failure_injected = true;
Err(anyhow!("injected task snapshot transport failure"))
} else {
fetch_task_snapshots_in(current_session, state)
};
let task_snapshots = match snapshot_request {
Ok(snapshots) => snapshots,
Err(error) => {
if !emit(RuntimeContinuationOutcome::Diagnostic(format!(
"runtime terminal snapshot observation failed: {error:#}; reconnecting in {} ms",
reconnect_delay.as_millis()
))) {
return Ok(());
}
session = None;
std::thread::sleep(reconnect_delay);
reconnect_delay = (reconnect_delay * 2).min(Duration::from_secs(5));
continue;
}
};
let process_status_request =
if inject_process_status_failure && !process_status_failure_injected {
process_status_failure_injected = true;
Err(anyhow!("injected process-status transport failure"))
} else {
fetch_current_process_status_in(current_session, state)
};
let (process_statuses, process_status) = match process_status_request {
Ok(status) => status,
Err(error) => {
if !emit(RuntimeContinuationOutcome::Diagnostic(format!(
"runtime terminal process observation failed: {error:#}; reconnecting in {} ms",
reconnect_delay.as_millis()
))) {
return Ok(());
}
session = None;
std::thread::sleep(reconnect_delay);
reconnect_delay = (reconnect_delay * 2).min(Duration::from_secs(5));
continue;
}
};
if !has_current_runtime_task(&task_snapshots) {
if let RuntimeContinuationOutcome::Terminal(record) = &mut outcome {
record.status_code =
whole_process_status_code(record.status_code, &task_snapshots);
record.node_report = json!({
"terminal_event": record.node_report.get("terminal_event"),
"task_snapshots": task_snapshots,
"process_status": process_status,
"process_statuses": process_statuses,
});
}
emit(outcome);
return Ok(());
}
}
let snapshot_request = if inject_snapshot_failure && !snapshot_failure_injected { let snapshot_request = if inject_snapshot_failure && !snapshot_failure_injected {
snapshot_failure_injected = true; snapshot_failure_injected = true;
Err(anyhow!("injected task snapshot transport failure")) Err(anyhow!("injected task snapshot transport failure"))
@ -1221,6 +1190,22 @@ pub(crate) fn observe_services_runtime(
} }
}; };
reconnect_delay = Duration::from_millis(100); reconnect_delay = Duration::from_millis(100);
if !has_current_runtime_task(&task_snapshots) {
if let Some(mut outcome) =
terminal_runtime_outcome(&coordinator, state, &events, process_status.as_ref())
{
if let RuntimeContinuationOutcome::Terminal(record) = &mut outcome {
record.node_report = json!({
"terminal_event": record.node_report.get("terminal_event"),
"task_snapshots": task_snapshots,
"process_status": process_status,
"process_statuses": process_statuses,
});
}
emit(outcome);
return Ok(());
}
}
if let Some((failed_task, _attempt_id)) = failed_awaiting_action_snapshot(&task_snapshots) { if let Some((failed_task, _attempt_id)) = failed_awaiting_action_snapshot(&task_snapshots) {
let failed_task = failed_task.to_owned(); let failed_task = failed_task.to_owned();
let failed_event = events let failed_event = events
@ -1352,7 +1337,9 @@ pub(crate) fn observe_services_runtime(
continue; continue;
} }
}; };
if let Some(mut outcome) = terminal_runtime_outcome(&coordinator, state, &events) { if let Some(mut outcome) =
terminal_runtime_outcome(&coordinator, state, &events, process_status.as_ref())
{
let task_snapshots = match fetch_task_snapshots_in(current_session, state) { let task_snapshots = match fetch_task_snapshots_in(current_session, state) {
Ok(snapshots) => snapshots, Ok(snapshots) => snapshots,
Err(snapshot_error) => { Err(snapshot_error) => {
@ -1370,8 +1357,6 @@ pub(crate) fn observe_services_runtime(
}; };
if !has_current_runtime_task(&task_snapshots) { if !has_current_runtime_task(&task_snapshots) {
if let RuntimeContinuationOutcome::Terminal(record) = &mut outcome { if let RuntimeContinuationOutcome::Terminal(record) = &mut outcome {
record.status_code =
whole_process_status_code(record.status_code, &task_snapshots);
record.node_report = json!({ record.node_report = json!({
"terminal_event": record.node_report.get("terminal_event"), "terminal_event": record.node_report.get("terminal_event"),
"task_snapshots": task_snapshots, "task_snapshots": task_snapshots,
@ -1547,6 +1532,7 @@ fn has_current_runtime_task(task_snapshots: &Value) -> bool {
}) })
} }
#[cfg(test)]
pub(crate) fn whole_process_status_code( pub(crate) fn whole_process_status_code(
main_status_code: Option<i32>, main_status_code: Option<i32>,
task_snapshots: &Value, task_snapshots: &Value,
@ -1582,66 +1568,75 @@ pub(crate) fn whole_process_status_code(
fn terminal_runtime_outcome( fn terminal_runtime_outcome(
coordinator: &str, coordinator: &str,
state: &AdapterState, _state: &AdapterState,
events: &Value, events: &Value,
process_status: Option<&Value>,
) -> Option<RuntimeContinuationOutcome> { ) -> Option<RuntimeContinuationOutcome> {
let final_result = process_status?
.get("final_result")
.and_then(Value::as_str)?;
let status_code = match final_result {
"completed" => 0,
"failed" | "cancelled" => 1,
_ => return None,
};
let event = events let event = events
.get("events") .get("events")
.and_then(Value::as_array)? .and_then(Value::as_array)
.get(state.runtime_event_count..)? .and_then(|events| {
.iter() events.iter().rev().find(|event| {
.rev() event.get("executor").and_then(Value::as_str) == Some("coordinator_main")
.find(|event| event.get("executor").and_then(Value::as_str) == Some("coordinator_main"))?; })
let status_code = event });
.get("status_code")
.and_then(Value::as_i64)
.map(|status| status as i32)
.or_else(
|| match event.get("terminal_state").and_then(Value::as_str) {
Some("completed") => Some(0),
Some("failed" | "cancelled") => Some(1),
_ => None,
},
);
Some(RuntimeContinuationOutcome::Terminal(RuntimeLaunchRecord { Some(RuntimeContinuationOutcome::Terminal(RuntimeLaunchRecord {
coordinator: coordinator.to_owned(), coordinator: coordinator.to_owned(),
node: event node: event
.get("node") .and_then(|event| event.get("node"))
.and_then(Value::as_str) .and_then(Value::as_str)
.or_else(|| {
process_status
.and_then(|status| status.get("connected_nodes"))
.and_then(Value::as_array)
.and_then(|nodes| nodes.first())
.and_then(Value::as_str)
})
.unwrap_or("coordinator-main") .unwrap_or("coordinator-main")
.to_owned(), .to_owned(),
node_report: json!({ "terminal_event": event }), node_report: json!({
"terminal_event": event,
"process_status": process_status,
}),
task_events: events.clone(), task_events: events.clone(),
placed_task_launched: true, placed_task_launched: true,
status_code, status_code: Some(status_code),
stdout_bytes: event stdout_bytes: event
.get("stdout_bytes") .and_then(|event| event.get("stdout_bytes"))
.and_then(Value::as_u64) .and_then(Value::as_u64)
.unwrap_or(0), .unwrap_or(0),
stderr_bytes: event stderr_bytes: event
.get("stderr_bytes") .and_then(|event| event.get("stderr_bytes"))
.and_then(Value::as_u64) .and_then(Value::as_u64)
.unwrap_or(0), .unwrap_or(0),
stdout_tail: event stdout_tail: event
.get("stdout_tail") .and_then(|event| event.get("stdout_tail"))
.and_then(Value::as_str) .and_then(Value::as_str)
.unwrap_or_default() .unwrap_or_default()
.to_owned(), .to_owned(),
stderr_tail: event stderr_tail: event
.get("stderr_tail") .and_then(|event| event.get("stderr_tail"))
.and_then(Value::as_str) .and_then(Value::as_str)
.unwrap_or_default() .unwrap_or_default()
.to_owned(), .to_owned(),
stdout_truncated: event stdout_truncated: event
.get("stdout_truncated") .and_then(|event| event.get("stdout_truncated"))
.and_then(Value::as_bool) .and_then(Value::as_bool)
.unwrap_or(false), .unwrap_or(false),
stderr_truncated: event stderr_truncated: event
.get("stderr_truncated") .and_then(|event| event.get("stderr_truncated"))
.and_then(Value::as_bool) .and_then(Value::as_bool)
.unwrap_or(false), .unwrap_or(false),
artifact_path: event artifact_path: event
.get("artifact_path") .and_then(|event| event.get("artifact_path"))
.and_then(Value::as_str) .and_then(Value::as_str)
.map(str::to_owned), .map(str::to_owned),
event_count: events event_count: events
@ -1727,6 +1722,53 @@ mod transactional_launch_tests {
assert!(error.contains("no virtual process was created")); assert!(error.contains("no virtual process was created"));
} }
#[test]
fn durable_process_summary_is_terminal_authority_after_event_rotation() {
let state = AdapterState {
runtime_event_count: 10_000,
..AdapterState::default()
};
let completed = terminal_runtime_outcome(
"127.0.0.1:1",
&state,
&json!({ "events": [] }),
Some(&json!({
"process": state.process.as_str(),
"lifecycle": "recent_terminal",
"final_result": "completed",
"connected_nodes": []
})),
)
.expect("the durable summary should terminate observation");
let RuntimeContinuationOutcome::Terminal(completed) = completed else {
panic!("expected terminal outcome");
};
assert_eq!(completed.status_code, Some(0));
let failed = terminal_runtime_outcome(
"127.0.0.1:1",
&state,
&json!({
"events": [{
"executor": "coordinator_main",
"terminal_state": "completed",
"status_code": 0
}]
}),
Some(&json!({
"process": state.process.as_str(),
"lifecycle": "recent_terminal",
"final_result": "failed",
"connected_nodes": []
})),
)
.expect("the aggregate summary should override a successful main event");
let RuntimeContinuationOutcome::Terminal(failed) = failed else {
panic!("expected terminal outcome");
};
assert_eq!(failed.status_code, Some(1));
}
#[test] #[test]
fn failed_debug_launch_reconnects_and_aborts_the_process() { fn failed_debug_launch_reconnects_and_aborts_the_process() {
let listener = TcpListener::bind("127.0.0.1:0").unwrap(); let listener = TcpListener::bind("127.0.0.1:0").unwrap();

View file

@ -2,7 +2,7 @@ use std::collections::{BTreeMap, BTreeSet, HashMap};
use std::io::Read; use std::io::Read;
use std::path::PathBuf; use std::path::PathBuf;
use std::process::Stdio; use std::process::Stdio;
use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
use std::sync::{Arc, Mutex}; use std::sync::{Arc, Mutex};
use std::thread; use std::thread;
use std::time::{Duration, Instant}; use std::time::{Duration, Instant};
@ -46,6 +46,40 @@ use validation::{
resolve_task_export, task_descriptors, verify_environment_digest, verify_source_snapshot, resolve_task_export, task_descriptors, verify_environment_digest, verify_source_snapshot,
}; };
#[derive(Debug)]
struct AssignmentExecutionError {
message: String,
stdout_source_bytes: u64,
stderr_source_bytes: u64,
}
impl std::fmt::Display for AssignmentExecutionError {
fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
formatter.write_str(&self.message)
}
}
impl std::error::Error for AssignmentExecutionError {}
fn execution_error_with_log_bytes(
message: impl Into<String>,
stdout_source_bytes: &AtomicU64,
stderr_source_bytes: &AtomicU64,
) -> Box<dyn std::error::Error> {
Box::new(AssignmentExecutionError {
message: message.into(),
stdout_source_bytes: stdout_source_bytes.load(Ordering::Relaxed),
stderr_source_bytes: stderr_source_bytes.load(Ordering::Relaxed),
})
}
pub(crate) fn assignment_error_log_bytes(error: &(dyn std::error::Error + 'static)) -> (u64, u64) {
error
.downcast_ref::<AssignmentExecutionError>()
.map(|error| (error.stdout_source_bytes, error.stderr_source_bytes))
.unwrap_or((0, 0))
}
pub(crate) fn run_verified_wasmtime_assignment( pub(crate) fn run_verified_wasmtime_assignment(
args: &Args, args: &Args,
task: &RuntimeTask, task: &RuntimeTask,
@ -95,6 +129,8 @@ pub(crate) fn run_verified_wasmtime_assignment(
return Err("Wasm entrypoint assignment omitted its descriptor export".into()); return Err("Wasm entrypoint assignment omitted its descriptor export".into());
} }
}; };
let command_stdout_source_bytes = Arc::new(AtomicU64::new(0));
let command_stderr_source_bytes = Arc::new(AtomicU64::new(0));
let (stdout, boundary_result) = match abi { let (stdout, boundary_result) = match abi {
WasmExportAbi::EntrypointV1 | WasmExportAbi::TaskV1 => { WasmExportAbi::EntrypointV1 | WasmExportAbi::TaskV1 => {
let invocation = WasmTaskInvocation::new( let invocation = WasmTaskInvocation::new(
@ -102,7 +138,8 @@ pub(crate) fn run_verified_wasmtime_assignment(
task_spec.task_instance.clone(), task_spec.task_instance.clone(),
task_spec.args.clone(), task_spec.args.clone(),
); );
let result = WasmtimeTaskRuntime::new()?.run_task_export_verified_with_task_host( let result = WasmtimeTaskRuntime::new()?
.run_task_export_verified_with_task_host(
&module, &module,
expected_bundle_digest, expected_bundle_digest,
export, export,
@ -112,8 +149,17 @@ pub(crate) fn run_verified_wasmtime_assignment(
task, task,
node_private_key, node_private_key,
&module, &module,
Arc::clone(&command_stdout_source_bytes),
Arc::clone(&command_stderr_source_bytes),
)?), )?),
)?; )
.map_err(|error| {
execution_error_with_log_bytes(
error.to_string(),
&command_stdout_source_bytes,
&command_stderr_source_bytes,
)
})?;
if std::env::var_os("CLUSTERFLUX_DEBUG_CONTROL_TRACE").is_some() { if std::env::var_os("CLUSTERFLUX_DEBUG_CONTROL_TRACE").is_some() {
eprintln!( eprintln!(
"clusterflux debug control: Wasm assignment returned for task {}", "clusterflux debug control: Wasm assignment returned for task {}",
@ -122,17 +168,35 @@ pub(crate) fn run_verified_wasmtime_assignment(
} }
match result.outcome { match result.outcome {
WasmTaskOutcome::Completed => { WasmTaskOutcome::Completed => {
let boundary = result.result.ok_or("completed Wasm task omitted result")?; let boundary = result.result.ok_or_else(|| {
execution_error_with_log_bytes(
"completed Wasm task omitted result",
&command_stdout_source_bytes,
&command_stderr_source_bytes,
)
})?;
( (
format!("{}\n", serde_json::to_string(&boundary)?), format!(
"{}\n",
serde_json::to_string(&boundary).map_err(|error| {
execution_error_with_log_bytes(
error.to_string(),
&command_stdout_source_bytes,
&command_stderr_source_bytes,
)
})?
),
Some(boundary), Some(boundary),
) )
} }
WasmTaskOutcome::Failed => { WasmTaskOutcome::Failed => {
return Err(result return Err(execution_error_with_log_bytes(
result
.error .error
.unwrap_or_else(|| "Wasm task failed without an error".to_owned()) .unwrap_or_else(|| "Wasm task failed without an error".to_owned()),
.into()) &command_stdout_source_bytes,
&command_stderr_source_bytes,
))
} }
} }
} }
@ -140,12 +204,18 @@ pub(crate) fn run_verified_wasmtime_assignment(
let task_id = TaskInstanceId::new(task.task.clone()); let task_id = TaskInstanceId::new(task.task.clone());
let artifacts = TaskArtifactStore::new(task_id.clone(), NodeId::new(args.node.clone())); let artifacts = TaskArtifactStore::new(task_id.clone(), NodeId::new(args.node.clone()));
let manifest = artifacts.flush(); let manifest = artifacts.flush();
let stdout_source_bytes = command_stdout_source_bytes
.load(Ordering::Relaxed)
.saturating_add(stdout.len() as u64);
let stderr_source_bytes = command_stderr_source_bytes.load(Ordering::Relaxed);
Ok(( Ok((
CommandOutput { CommandOutput {
virtual_thread: task_id, virtual_thread: task_id,
status_code: Some(0), status_code: Some(0),
stdout, stdout,
stderr: String::new(), stderr: String::new(),
stdout_source_bytes,
stderr_source_bytes,
stdout_truncated: false, stdout_truncated: false,
stderr_truncated: false, stderr_truncated: false,
log_backpressured: false, log_backpressured: false,
@ -176,6 +246,8 @@ struct CoordinatorWasmTaskHost {
next_handle_id: u64, next_handle_id: u64,
handles: Arc<Mutex<HashMap<u64, TaskSpec>>>, handles: Arc<Mutex<HashMap<u64, TaskSpec>>>,
command_status: Arc<Mutex<Option<String>>>, command_status: Arc<Mutex<Option<String>>>,
command_stdout_source_bytes: Arc<AtomicU64>,
command_stderr_source_bytes: Arc<AtomicU64>,
cancellation_requested: Arc<AtomicBool>, cancellation_requested: Arc<AtomicBool>,
abort_requested: Arc<AtomicBool>, abort_requested: Arc<AtomicBool>,
debug_control: Arc<WasmDebugControl>, debug_control: Arc<WasmDebugControl>,
@ -188,6 +260,8 @@ impl CoordinatorWasmTaskHost {
parent: &RuntimeTask, parent: &RuntimeTask,
node_private_key: &str, node_private_key: &str,
module: &[u8], module: &[u8],
command_stdout_source_bytes: Arc<AtomicU64>,
command_stderr_source_bytes: Arc<AtomicU64>,
) -> Result<Self, Box<dyn std::error::Error>> { ) -> Result<Self, Box<dyn std::error::Error>> {
let task_spec = parent let task_spec = parent
.task_spec .task_spec
@ -268,6 +342,8 @@ impl CoordinatorWasmTaskHost {
next_handle_id: 1, next_handle_id: 1,
handles, handles,
command_status, command_status,
command_stdout_source_bytes,
command_stderr_source_bytes,
cancellation_requested, cancellation_requested,
abort_requested, abort_requested,
debug_control, debug_control,
@ -293,7 +369,16 @@ impl CoordinatorWasmTaskHost {
runtime_task_from_assignment(assignment).map_err(|error| error.to_string())?; runtime_task_from_assignment(assignment).map_err(|error| error.to_string())?;
let execution = let execution =
run_verified_wasmtime_assignment(&self.args, &runtime_task, &self.node_private_key); run_verified_wasmtime_assignment(&self.args, &runtime_task, &self.node_private_key);
let (terminal_state, status_code, stdout, stderr, result, retained) = match execution { let (
terminal_state,
status_code,
stdout,
stderr,
stdout_source_bytes,
stderr_source_bytes,
result,
retained,
) = match execution {
Ok((output, _manifest, result)) => { Ok((output, _manifest, result)) => {
let retained = retained_result_artifact( let retained = retained_result_artifact(
self.args.project_root.as_deref(), self.args.project_root.as_deref(),
@ -306,20 +391,40 @@ impl CoordinatorWasmTaskHost {
output.status_code, output.status_code,
output.stdout, output.stdout,
output.stderr, output.stderr,
output.stdout_source_bytes,
output.stderr_source_bytes,
result, result,
retained, retained,
), ),
Err(error) => ("failed", Some(1), String::new(), error, None, None),
}
}
Err(error) => ( Err(error) => (
"failed", "failed",
Some(1), Some(1),
String::new(), String::new(),
error.to_string(), error.clone(),
output.stdout_source_bytes,
output
.stderr_source_bytes
.saturating_add(error.len() as u64),
None, None,
None, None,
), ),
}
}
Err(error) => {
let (stdout_source_bytes, stderr_source_bytes) =
assignment_error_log_bytes(error.as_ref());
let error = error.to_string();
(
"failed",
Some(1),
String::new(),
error.clone(),
stdout_source_bytes,
stderr_source_bytes.saturating_add(error.len() as u64),
None,
None,
)
}
}; };
let artifact_path = retained let artifact_path = retained
.as_ref() .as_ref()
@ -339,8 +444,8 @@ impl CoordinatorWasmTaskHost {
"task": runtime_task.task, "task": runtime_task.task,
"terminal_state": terminal_state, "terminal_state": terminal_state,
"status_code": status_code, "status_code": status_code,
"stdout_bytes": stdout.len(), "stdout_bytes": stdout_source_bytes,
"stderr_bytes": stderr.len(), "stderr_bytes": stderr_source_bytes,
"stdout_tail": stdout, "stdout_tail": stdout,
"stderr_tail": stderr, "stderr_tail": stderr,
"stdout_truncated": false, "stdout_truncated": false,

View file

@ -66,6 +66,8 @@ pub(super) struct CoordinatorControlledProcessRunner {
pub(super) node_private_key: String, pub(super) node_private_key: String,
pub(super) debug_control: Arc<WasmDebugControl>, pub(super) debug_control: Arc<WasmDebugControl>,
pub(super) command_status: Arc<Mutex<Option<String>>>, pub(super) command_status: Arc<Mutex<Option<String>>>,
pub(super) stdout_source_bytes: Arc<AtomicU64>,
pub(super) stderr_source_bytes: Arc<AtomicU64>,
pub(super) timeout: Duration, pub(super) timeout: Duration,
pub(super) configured_secrets: Vec<String>, pub(super) configured_secrets: Vec<String>,
} }
@ -85,6 +87,8 @@ impl CoordinatorControlledProcessRunner {
node_private_key: host.node_private_key.clone(), node_private_key: host.node_private_key.clone(),
debug_control: Arc::clone(&host.debug_control), debug_control: Arc::clone(&host.debug_control),
command_status: Arc::clone(&host.command_status), command_status: Arc::clone(&host.command_status),
stdout_source_bytes: Arc::clone(&host.command_stdout_source_bytes),
stderr_source_bytes: Arc::clone(&host.command_stderr_source_bytes),
timeout, timeout,
configured_secrets, configured_secrets,
} }
@ -269,12 +273,13 @@ impl CoordinatorControlledProcessRunner {
stream: &'static str, stream: &'static str,
sender: SyncSender<LiveLogChunk>, sender: SyncSender<LiveLogChunk>,
configured_secrets: Vec<String>, configured_secrets: Vec<String>,
source_bytes_total: Arc<AtomicU64>,
) -> thread::JoinHandle<Result<Vec<u8>, String>> { ) -> thread::JoinHandle<Result<Vec<u8>, String>> {
thread::spawn(move || { thread::spawn(move || {
let mut captured = Vec::new(); let mut captured = Vec::new();
let mut buffer = [0_u8; 16 * 1024]; let mut buffer = [0_u8; 16 * 1024];
let mut source_offset = 0_u64; let stream_base = source_bytes_total.load(Ordering::Relaxed);
let mut pending_offset = 0_u64; let mut pending_offset = stream_base;
let mut pending = Vec::new(); let mut pending = Vec::new();
let mut discarded = false; let mut discarded = false;
loop { loop {
@ -284,6 +289,11 @@ impl CoordinatorControlledProcessRunner {
if count == 0 { if count == 0 {
break; break;
} }
let _ = source_bytes_total.fetch_update(
Ordering::Relaxed,
Ordering::Relaxed,
|current| Some(current.saturating_add(count as u64)),
);
let remaining = maximum.saturating_sub(captured.len()); let remaining = maximum.saturating_sub(captured.len());
let retained = count.min(remaining); let retained = count.min(remaining);
if retained > 0 { if retained > 0 {
@ -306,7 +316,6 @@ impl CoordinatorControlledProcessRunner {
if retained < count { if retained < count {
discarded = true; discarded = true;
} }
source_offset = source_offset.saturating_add(count as u64);
} }
if let Some((consumed, text)) = if let Some((consumed, text)) =
redact_safe_live_log_prefix(&pending, &configured_secrets, true) redact_safe_live_log_prefix(&pending, &configured_secrets, true)
@ -322,7 +331,7 @@ impl CoordinatorControlledProcessRunner {
if discarded { if discarded {
let _ = sender.try_send(LiveLogChunk { let _ = sender.try_send(LiveLogChunk {
stream, stream,
offset: maximum as u64, offset: stream_base.saturating_add(maximum as u64),
source_bytes: 0, source_bytes: 0,
bytes: b"[log output truncated at node capture limit]".to_vec(), bytes: b"[log output truncated at node capture limit]".to_vec(),
truncated: true, truncated: true,
@ -445,6 +454,7 @@ impl ProcessRunner for CoordinatorControlledProcessRunner {
"stdout", "stdout",
live_log_sender.clone(), live_log_sender.clone(),
self.configured_secrets.clone(), self.configured_secrets.clone(),
Arc::clone(&self.stdout_source_bytes),
); );
let stderr = Self::drain_bounded( let stderr = Self::drain_bounded(
child child
@ -455,6 +465,7 @@ impl ProcessRunner for CoordinatorControlledProcessRunner {
"stderr", "stderr",
live_log_sender, live_log_sender,
self.configured_secrets.clone(), self.configured_secrets.clone(),
Arc::clone(&self.stderr_source_bytes),
); );
let live_log_reporter = self.spawn_live_log_reporter(live_log_receiver); let live_log_reporter = self.spawn_live_log_reporter(live_log_receiver);
let mut session = match CoordinatorSession::connect(&self.args.coordinator) { let mut session = match CoordinatorSession::connect(&self.args.coordinator) {
@ -584,7 +595,10 @@ impl ProcessRunner for CoordinatorControlledProcessRunner {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::redact_safe_live_log_prefix; use super::{redact_safe_live_log_prefix, CoordinatorControlledProcessRunner};
use std::io::Cursor;
use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::{mpsc, Arc};
#[test] #[test]
fn live_log_redaction_holds_boundaries_until_split_secrets_are_complete() { fn live_log_redaction_holds_boundaries_until_split_secrets_are_complete() {
@ -600,4 +614,30 @@ mod tests {
assert!(!combined.contains("correct-")); assert!(!combined.contains("correct-"));
assert!(!combined.contains("horse")); assert!(!combined.contains("horse"));
} }
#[test]
fn bounded_capture_retains_the_complete_source_byte_count() {
let source_bytes = Arc::new(AtomicU64::new(5));
let (sender, receiver) = mpsc::sync_channel(8);
let reader = CoordinatorControlledProcessRunner::drain_bounded(
Cursor::new(vec![b'x'; 32]),
8,
"stdout",
sender,
Vec::new(),
Arc::clone(&source_bytes),
);
let captured = reader.join().unwrap().unwrap();
let chunks = receiver.into_iter().collect::<Vec<_>>();
assert_eq!(captured, vec![b'x'; 8]);
assert_eq!(source_bytes.load(Ordering::Relaxed), 37);
assert_eq!(
chunks.iter().map(|chunk| chunk.source_bytes).sum::<u64>(),
8
);
assert_eq!(chunks[0].offset, 5);
assert_eq!(chunks.last().unwrap().offset, 13);
assert!(chunks.iter().any(|chunk| chunk.truncated));
}
} }

View file

@ -48,6 +48,8 @@ fn test_controlled_runner(
), ),
debug_control: Arc::new(WasmDebugControl::default()), debug_control: Arc::new(WasmDebugControl::default()),
command_status: Arc::new(Mutex::new(None)), command_status: Arc::new(Mutex::new(None)),
stdout_source_bytes: Arc::new(AtomicU64::new(0)),
stderr_source_bytes: Arc::new(AtomicU64::new(0)),
timeout, timeout,
configured_secrets: Vec::new(), configured_secrets: Vec::new(),
} }
@ -90,6 +92,8 @@ fn controlled_process_runner_kills_running_group_when_abort_is_polled() {
), ),
debug_control: Arc::new(WasmDebugControl::default()), debug_control: Arc::new(WasmDebugControl::default()),
command_status: Arc::new(Mutex::new(None)), command_status: Arc::new(Mutex::new(None)),
stdout_source_bytes: Arc::new(AtomicU64::new(0)),
stderr_source_bytes: Arc::new(AtomicU64::new(0)),
timeout: Duration::from_secs(30), timeout: Duration::from_secs(30),
configured_secrets: Vec::new(), configured_secrets: Vec::new(),
}; };

View file

@ -42,6 +42,8 @@ pub struct CommandOutput {
pub status_code: Option<i32>, pub status_code: Option<i32>,
pub stdout: String, pub stdout: String,
pub stderr: String, pub stderr: String,
pub stdout_source_bytes: u64,
pub stderr_source_bytes: u64,
pub stdout_truncated: bool, pub stdout_truncated: bool,
pub stderr_truncated: bool, pub stderr_truncated: bool,
pub log_backpressured: bool, pub log_backpressured: bool,
@ -83,6 +85,8 @@ impl LocalCommandExecutor {
&output.stderr, &output.stderr,
max_log_bytes, max_log_bytes,
); );
let stdout_source_bytes = output.stdout.len() as u64;
let stderr_source_bytes = output.stderr.len() as u64;
let staged_artifact = if let Some(path) = command.stage_stdout_as { let staged_artifact = if let Some(path) = command.stage_stdout_as {
Some(overlay.write( Some(overlay.write(
path, path,
@ -98,6 +102,8 @@ impl LocalCommandExecutor {
status_code: output.status.code(), status_code: output.status.code(),
stdout: logs.stdout, stdout: logs.stdout,
stderr: logs.stderr, stderr: logs.stderr,
stdout_source_bytes,
stderr_source_bytes,
stdout_truncated: logs.stdout_truncated, stdout_truncated: logs.stdout_truncated,
stderr_truncated: logs.stderr_truncated, stderr_truncated: logs.stderr_truncated,
log_backpressured: logs.backpressured, log_backpressured: logs.backpressured,

View file

@ -11,7 +11,7 @@ use clusterflux_core::{
}; };
use serde_json::{json, Value}; use serde_json::{json, Value};
use crate::assignment_runner::run_verified_wasmtime_assignment; use crate::assignment_runner::{assignment_error_log_bytes, run_verified_wasmtime_assignment};
#[cfg(test)] #[cfg(test)]
use crate::coordinator_session::control_endpoint_identity; use crate::coordinator_session::control_endpoint_identity;
use crate::coordinator_session::CoordinatorSession; use crate::coordinator_session::CoordinatorSession;
@ -464,6 +464,8 @@ fn run_runtime_task(
capability_report, capability_report,
debug_command, debug_command,
node_private_key, node_private_key,
0,
0,
); );
} }
@ -498,9 +500,13 @@ fn run_runtime_task(
debug_command, debug_command,
node_private_key, node_private_key,
&error, &error,
output.stdout_source_bytes,
output.stderr_source_bytes,
), ),
}, },
Err(error) => { Err(error) => {
let (stdout_source_bytes, stderr_source_bytes) =
assignment_error_log_bytes(error.as_ref());
let error = error.to_string(); let error = error.to_string();
if error.contains("task execution cancelled:") { if error.contains("task execution cancelled:") {
record_cancelled_task( record_cancelled_task(
@ -512,6 +518,8 @@ fn run_runtime_task(
capability_report, capability_report,
debug_command, debug_command,
node_private_key, node_private_key,
stdout_source_bytes,
stderr_source_bytes,
) )
} else { } else {
record_failed_task( record_failed_task(
@ -524,6 +532,8 @@ fn run_runtime_task(
debug_command, debug_command,
node_private_key, node_private_key,
&error, &error,
stdout_source_bytes,
stderr_source_bytes,
) )
} }
} }

View file

@ -47,8 +47,8 @@ pub(crate) fn record_completed_task(
"process": &task.process, "process": &task.process,
"node": &args.node, "node": &args.node,
"task": &task.task, "task": &task.task,
"stdout_bytes": output.stdout.len(), "stdout_bytes": output.stdout_source_bytes,
"stderr_bytes": output.stderr.len(), "stderr_bytes": output.stderr_source_bytes,
"stdout_tail": &output.stdout, "stdout_tail": &output.stdout,
"stderr_tail": &output.stderr, "stderr_tail": &output.stderr,
"stdout_truncated": output.stdout_truncated, "stdout_truncated": output.stdout_truncated,
@ -85,8 +85,8 @@ pub(crate) fn record_completed_task(
"node": &args.node, "node": &args.node,
"task": &task.task, "task": &task.task,
"status_code": output.status_code, "status_code": output.status_code,
"stdout_bytes": output.stdout.len(), "stdout_bytes": output.stdout_source_bytes,
"stderr_bytes": output.stderr.len(), "stderr_bytes": output.stderr_source_bytes,
"stdout_tail": &output.stdout, "stdout_tail": &output.stdout,
"stderr_tail": &output.stderr, "stderr_tail": &output.stderr,
"stdout_truncated": output.stdout_truncated, "stdout_truncated": output.stdout_truncated,
@ -123,8 +123,11 @@ pub(crate) fn record_failed_task(
debug_command: Value, debug_command: Value,
node_private_key: &str, node_private_key: &str,
error: &str, error: &str,
stdout_source_bytes: u64,
command_stderr_source_bytes: u64,
) -> Result<Value, Box<dyn std::error::Error>> { ) -> Result<Value, Box<dyn std::error::Error>> {
let error = bounded_runtime_error(error); let error = bounded_runtime_error(error);
let stderr_source_bytes = command_stderr_source_bytes.saturating_add(error.len() as u64);
let log_event = session.request(signed_node_request_json( let log_event = session.request(signed_node_request_json(
args, args,
node_private_key, node_private_key,
@ -136,8 +139,8 @@ pub(crate) fn record_failed_task(
"process": &task.process, "process": &task.process,
"node": &args.node, "node": &args.node,
"task": &task.task, "task": &task.task,
"stdout_bytes": 0, "stdout_bytes": stdout_source_bytes,
"stderr_bytes": error.len(), "stderr_bytes": stderr_source_bytes,
"stdout_tail": "", "stdout_tail": "",
"stderr_tail": &error, "stderr_tail": &error,
"stdout_truncated": false, "stdout_truncated": false,
@ -175,8 +178,8 @@ pub(crate) fn record_failed_task(
"task": &task.task, "task": &task.task,
"terminal_state": "failed", "terminal_state": "failed",
"status_code": -1, "status_code": -1,
"stdout_bytes": 0, "stdout_bytes": stdout_source_bytes,
"stderr_bytes": error.len(), "stderr_bytes": stderr_source_bytes,
"stdout_tail": "", "stdout_tail": "",
"stderr_tail": &error, "stderr_tail": &error,
"stdout_truncated": false, "stdout_truncated": false,
@ -189,6 +192,8 @@ pub(crate) fn record_failed_task(
Ok(failed_node_report( Ok(failed_node_report(
task, task,
&error, &error,
stdout_source_bytes,
stderr_source_bytes,
registration, registration,
heartbeat, heartbeat,
capability_report, capability_report,
@ -210,6 +215,8 @@ pub(crate) fn record_cancelled_task(
capability_report: Value, capability_report: Value,
debug_command: Value, debug_command: Value,
node_private_key: &str, node_private_key: &str,
stdout_source_bytes: u64,
stderr_source_bytes: u64,
) -> Result<Value, Box<dyn std::error::Error>> { ) -> Result<Value, Box<dyn std::error::Error>> {
let recorded = session.request(signed_node_request_json( let recorded = session.request(signed_node_request_json(
args, args,
@ -224,12 +231,12 @@ pub(crate) fn record_cancelled_task(
"task": &task.task, "task": &task.task,
"terminal_state": "cancelled", "terminal_state": "cancelled",
"status_code": null, "status_code": null,
"stdout_bytes": 0, "stdout_bytes": stdout_source_bytes,
"stderr_bytes": 0, "stderr_bytes": stderr_source_bytes,
"stdout_tail": "", "stdout_tail": "",
"stderr_tail": "", "stderr_tail": "",
"stdout_truncated": false, "stdout_truncated": stdout_source_bytes > 0,
"stderr_truncated": false, "stderr_truncated": stderr_source_bytes > 0,
"artifact_path": null, "artifact_path": null,
"artifact_digest": null, "artifact_digest": null,
"artifact_size_bytes": null, "artifact_size_bytes": null,
@ -238,6 +245,8 @@ pub(crate) fn record_cancelled_task(
)?)?; )?)?;
Ok(cancelled_node_report( Ok(cancelled_node_report(
task, task,
stdout_source_bytes,
stderr_source_bytes,
registration, registration,
heartbeat, heartbeat,
capability_report, capability_report,
@ -281,8 +290,8 @@ pub(crate) fn completed_node_report(
"virtual_thread": output.virtual_thread, "virtual_thread": output.virtual_thread,
"terminal_state": if output.status_code == Some(0) { "completed" } else { "failed" }, "terminal_state": if output.status_code == Some(0) { "completed" } else { "failed" },
"status_code": output.status_code, "status_code": output.status_code,
"stdout_bytes": output.stdout.len(), "stdout_bytes": output.stdout_source_bytes,
"stderr_bytes": output.stderr.len(), "stderr_bytes": output.stderr_source_bytes,
"stdout_tail": &output.stdout, "stdout_tail": &output.stdout,
"stderr_tail": &output.stderr, "stderr_tail": &output.stderr,
"stdout_truncated": output.stdout_truncated, "stdout_truncated": output.stdout_truncated,
@ -305,6 +314,8 @@ pub(crate) fn completed_node_report(
#[allow(clippy::too_many_arguments)] #[allow(clippy::too_many_arguments)]
pub(crate) fn cancelled_node_report( pub(crate) fn cancelled_node_report(
task: &RuntimeTask, task: &RuntimeTask,
stdout_source_bytes: u64,
stderr_source_bytes: u64,
registration_response: Value, registration_response: Value,
heartbeat_response: Value, heartbeat_response: Value,
capability_response: Value, capability_response: Value,
@ -318,12 +329,12 @@ pub(crate) fn cancelled_node_report(
"virtual_thread": &task.task, "virtual_thread": &task.task,
"terminal_state": "cancelled", "terminal_state": "cancelled",
"status_code": null, "status_code": null,
"stdout_bytes": 0, "stdout_bytes": stdout_source_bytes,
"stderr_bytes": 0, "stderr_bytes": stderr_source_bytes,
"stdout_tail": "", "stdout_tail": "",
"stderr_tail": "", "stderr_tail": "",
"stdout_truncated": false, "stdout_truncated": stdout_source_bytes > 0,
"stderr_truncated": false, "stderr_truncated": stderr_source_bytes > 0,
"log_backpressured": false, "log_backpressured": false,
"staged_artifact": null, "staged_artifact": null,
"large_bytes_uploaded": false, "large_bytes_uploaded": false,
@ -343,6 +354,8 @@ pub(crate) fn cancelled_node_report(
pub(crate) fn failed_node_report( pub(crate) fn failed_node_report(
task: &RuntimeTask, task: &RuntimeTask,
error: &str, error: &str,
stdout_source_bytes: u64,
stderr_source_bytes: u64,
registration_response: Value, registration_response: Value,
heartbeat_response: Value, heartbeat_response: Value,
capability_response: Value, capability_response: Value,
@ -357,8 +370,8 @@ pub(crate) fn failed_node_report(
"virtual_thread": &task.task, "virtual_thread": &task.task,
"terminal_state": "failed", "terminal_state": "failed",
"status_code": -1, "status_code": -1,
"stdout_bytes": 0, "stdout_bytes": stdout_source_bytes,
"stderr_bytes": error.len(), "stderr_bytes": stderr_source_bytes,
"stdout_tail": "", "stdout_tail": "",
"stderr_tail": error, "stderr_tail": error,
"stdout_truncated": false, "stdout_truncated": false,
@ -377,3 +390,40 @@ pub(crate) fn failed_node_report(
"coordinator_response": coordinator_response, "coordinator_response": coordinator_response,
}) })
} }
#[cfg(test)]
mod tests {
use super::*;
use clusterflux_core::TaskInstanceId;
#[test]
fn completed_report_uses_source_counts_instead_of_bounded_tail_lengths() {
let report = completed_node_report(
CommandOutput {
virtual_thread: TaskInstanceId::from("task"),
status_code: Some(0),
stdout: "bounded tail".to_owned(),
stderr: String::new(),
stdout_source_bytes: 519,
stderr_source_bytes: 0,
stdout_truncated: true,
stderr_truncated: false,
log_backpressured: false,
staged_artifact: None,
},
false,
Value::Null,
Value::Null,
Value::Null,
Value::Null,
Value::Null,
Value::Null,
Value::Null,
Value::Null,
1,
);
assert_eq!(report["stdout_bytes"], 519);
assert_eq!(report["stdout_tail"], "bounded tail");
}
}