use crate::{
model::{
fence_acquisition::{
FenceAcquisitionAcknowledgement, FenceAcquisitionError, FenceAcquisitionPayload,
FenceAcquisitionRequest, original_authority,
},
fence_obligation::FenceObligationRecord,
operation_plan::OperationPlanError,
},
ops::persistence::{
AttemptJournalError, AttemptJournalGuard, ExecutionProgressPersistenceError,
ExecutionStageGuard, ExecutionWorkflowPersistenceError, FenceObligationPersistenceError,
FenceObligationRequirement, read_fence_obligation,
},
policy::fence_acquisition::{FenceAcquisitionAcknowledgementError, validate_acknowledgement},
ports::fence_acquisition::{FenceAcquisitionProvider, FenceAcquisitionProviderError},
};
use thiserror::Error;
pub async fn acquire_fence<E: std::error::Error + 'static>(
stage: &ExecutionStageGuard<'_>,
requirement: FenceObligationRequirement<'_>,
obligation: &FenceObligationRecord,
payload: &FenceAcquisitionPayload,
provider: &mut impl FenceAcquisitionProvider,
admit: impl AsyncFnOnce(&FenceAcquisitionRequest<'_>) -> Result<(), E>,
) -> Result<FenceAcquisitionAcknowledgement, FenceAcquisitionExecutionError<E>> {
let plan = stage.plan();
let layout = stage.layout()?;
read_fence_obligation(layout, plan, requirement, &obligation.digest())?;
let authority = plan.attempt_authority(obligation.acquisition_operation())?;
let mut journal = AttemptJournalGuard::open(layout, &authority)?;
original_authority(plan, obligation, journal.record()?, payload)?;
journal.reserve_planned_mutation(&plan.digest())?;
let request = FenceAcquisitionRequest::new(plan, obligation, journal.record()?, payload)?;
admit(&request)
.await
.map_err(FenceAcquisitionExecutionError::Admission)?;
read_fence_obligation(stage.layout()?, plan, requirement, &obligation.digest())?;
let acknowledgement = provider.acquire_fence(&request, journal.record()?).await?;
let admit_reply = || -> Result<(), FenceAcquisitionReplyError> {
validate_acknowledgement(&request, journal.record()?, &acknowledgement)?;
read_fence_obligation(stage.layout()?, plan, requirement, &obligation.digest())?;
Ok(())
};
match admit_reply() {
Ok(()) => Ok(acknowledgement),
Err(source) => Err(FenceAcquisitionExecutionError::AfterReply {
source,
acknowledgement: Box::new(acknowledgement),
}),
}
}
#[derive(Debug, Error)]
pub enum FenceAcquisitionExecutionError<E: std::error::Error + 'static> {
#[error(transparent)]
Plan(#[from] OperationPlanError),
#[error(transparent)]
Stage(#[from] ExecutionWorkflowPersistenceError),
#[error(transparent)]
Obligation(#[from] FenceObligationPersistenceError),
#[error(transparent)]
Journal(#[from] AttemptJournalError),
#[error(transparent)]
Progress(#[from] ExecutionProgressPersistenceError),
#[error(transparent)]
Request(#[from] FenceAcquisitionError),
#[error("fresh fence acquisition admission failed: {0}")]
Admission(#[source] E),
#[error(transparent)]
Provider(#[from] FenceAcquisitionProviderError),
#[error("fence acquisition reply rejected: {source}")]
AfterReply {
source: FenceAcquisitionReplyError,
acknowledgement: Box<FenceAcquisitionAcknowledgement>,
},
}
#[derive(Debug, Error)]
pub enum FenceAcquisitionReplyError {
#[error(transparent)]
Journal(#[from] AttemptJournalError),
#[error(transparent)]
Association(#[from] FenceAcquisitionAcknowledgementError),
#[error(transparent)]
Stage(#[from] ExecutionWorkflowPersistenceError),
#[error(transparent)]
Obligation(#[from] FenceObligationPersistenceError),
}
#[cfg(all(test, unix))]
mod tests;