Struct AllocationManager 

pub struct AllocationManager { /* private fields */ }

Manages allocations for a supervisor.

Implementations

impl AllocationManager

pub fn new(supervisor_id: impl Into<String>) -> Self

Create a new allocation manager.

pub fn max_lend(self, n: usize) -> Self

Set max lend.

pub fn max_borrow(self, n: usize) -> Self

Set max borrow.

pub async fn record_lent(&self, grant: AllocationGrant)

Record a grant where we are the lender.

pub async fn record_borrowed(&self, grant: AllocationGrant)

Record a grant where we are the borrower.

pub async fn remove_grant(&self, lease_id: &str)

Remove a grant.

pub async fn lent_count(&self) -> usize

Get count of lent executors.

pub async fn borrowed_count(&self) -> usize

Get count of borrowed executors.

pub async fn can_lend(&self, count: usize) -> bool

Can we lend more?

pub async fn can_borrow(&self, count: usize) -> bool

Can we borrow more?

pub async fn get_expired_grants(&self) -> Vec<AllocationGrant>

Get expired grants.

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

Clean up expired grants.

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

Get all borrowed executor IDs.

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

Get all lent executor IDs.

pub fn create_offer( &self, request: &AllocationRequest, executor_ids: Vec<String>, ) -> AllocationOffer

Create an offer for a request. Used by supervisors to respond to allocation requests.

pub fn create_release( lease_id: impl Into<String>, reason: ReleaseReason, ) -> AllocationRelease

Create a release notification when returning executors.

Trait Implementations

impl Debug for AllocationManager

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.