Struct LeaseManager 

pub struct LeaseManager<S: LeaseStorage> { /* private fields */ }

Implementations

impl<S: LeaseStorage + 'static> LeaseManager<S>

pub fn new(storage: Arc<S>, worker_id: String, config: LeaseConfig) -> Self

pub fn worker_id(&self) -> &str

pub async fn ensure_lease(&self, key: &str) -> Result<()>

Ensures a lease row exists for this key. Idempotent — safe to call from multiple workers simultaneously.

pub async fn get_my_lease_keys(&self) -> Result<Vec<String>>

Returns lease keys currently owned by this worker.

pub async fn rebalance(&self) -> Result<Vec<String>>

Core rebalance algorithm. Tries to reach a fair share of leases for this worker by first claiming unowned leases, then stealing expired ones.

pub async fn renew_my_leases(&self) -> Result<()>

Renew all leases owned by this worker. Uses concurrent requests to minimize total latency when holding many leases.

pub async fn get_checkpoint(&self, lease_key: &str) -> Result<Option<String>>

pub async fn checkpoint(&self, lease_key: &str, checkpoint: &str) -> Result<()>

pub fn start_background_tasks(self: Arc<Self>)

Spawns background tokio tasks for periodic renewal and rebalancing. Call this once after creating the manager.