1  use arroyo_state::checkpoint_state::CheckpointState;
 2  use arroyo_state::committing_state::CommittingState;
 3  
 4  pub enum CheckpointingOrCommittingState {
 5      Checkpointing(CheckpointState),
 6      Committing(CommittingState),
 7  }
 8  
 9  impl CheckpointingOrCommittingState {
10      pub(crate) fn done(&self) -> bool {
11          match self {
12              CheckpointingOrCommittingState::Checkpointing(checkpointing) => checkpointing.done(),
13              CheckpointingOrCommittingState::Committing(committing) => committing.done(),
14          }
15      }
16  }