codex-proxy-rs/backend/src/dispatch/stream/trace.rs

//! Request-level trace state for Responses dispatch.

#[derive(Debug, Default)]
pub(in crate::dispatch) struct ResponseDispatchTrace {
    next_attempt_index: i64,
    attempts: Vec<ResponseDispatchAttempt>,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub(in crate::dispatch) struct ResponseDispatchAttempt {
    index: i64,
    account_id: String,
}

impl ResponseDispatchTrace {
    pub(in crate::dispatch) fn start_attempt(
        &mut self,
        account_id: &str,
    ) -> ResponseDispatchAttempt {
        let attempt = ResponseDispatchAttempt {
            index: self.next_attempt_index,
            account_id: account_id.to_string(),
        };
        self.next_attempt_index += 1;
        self.attempts.push(attempt.clone());
        attempt
    }

    pub(in crate::dispatch) fn attempts(&self) -> &[ResponseDispatchAttempt] {
        &self.attempts
    }
}

impl ResponseDispatchAttempt {
    pub(in crate::dispatch) fn index(&self) -> i64 {
        self.index
    }

    pub(in crate::dispatch) fn account_id(&self) -> &str {
        &self.account_id
    }
}