Function start_or_reconnect

atomic-cli/src/commands/agent/owner.rs:786–833

Start the repository owner or reconnect when another process won election.

(repository: &Path)

/// Start the repository owner or reconnect when another process won election.
pub(crate) fn start_or_reconnect(repository: &Path) -> anyhow::Result<OwnerResponse> {
    if let Ok(response) = request(repository, OwnerRequest::Ping) {
        return Ok(response);
    }

    let executable = std::env::current_exe().context("failed to locate atomic executable")?;
    let mut command = ProcessCommand::new(executable);
    command
        .arg("agent")
        .arg("database-owner")
        .arg("serve")
        .arg("--repository")
        .arg(repository)
        .stdin(Stdio::null())
        .stdout(Stdio::null())
        .stderr(Stdio::null());
    configure_detached(&mut command)?;
    let mut child = command
        .spawn()
        .context("failed to spawn repository database owner")?;

    for _ in 0..START_ATTEMPTS {
        if let Ok(response) = request(repository, OwnerRequest::Ping) {
            return Ok(response);
        }
        if let Some(status) = child
            .try_wait()
            .context("failed to inspect database-owner bootstrap")?
        {
            // A concurrent bootstrap may have won election. Give its endpoint
            // the remainder of the retry window before reporting our exit.
            if !status.success() {
                for _ in 0..START_ATTEMPTS {
                    if let Ok(response) = request(repository, OwnerRequest::Ping) {
                        return Ok(response);
                    }
                    std::thread::sleep(START_RETRY_DELAY);
                }
                return Err(anyhow!("database owner exited during bootstrap: {status}"));
            }
        }
        std::thread::sleep(START_RETRY_DELAY);
    }

    Err(anyhow!(
        "database owner did not become healthy before timeout"
    ))
}

fn request_with_reconnect(
    repository: &Path,