Method inner_get_worker

components/scheduler/src/vmworkerpool.rs:84–165

(
        &self,
        guild_id: Id<GuildMarker>,
        premium_tier: Option<PremiumSlotTier>,
    )

    }

    async fn inner_get_worker(
        &self,
        guild_id: Id<GuildMarker>,
        premium_tier: Option<PremiumSlotTier>,
    ) -> WorkerHandle {
        let started_at = Instant::now();
        let priority_index = premium_tier_index(premium_tier);

        let rx = {
            let mut w = self.inner.lock().unwrap();

            // try to find one with identical guild id, avoids us having to reload all scripts
            let mut i = priority_index;
            loop {
                let pool = &mut w.pools[i];
                let pref_worker = pool
                    .iter()
                    .enumerate()
                    .find(|(_, v)| matches!(v.last_active_guild, Some(g) if g == guild_id))
                    .map(|(i, _)| i);

                if let Some(pref_worker) = pref_worker {
                    metrics::gauge!("bl.scheduler.workerpool_available_workers", "priority_index" => i.to_string()).decrement(1.0);
                    let worker = pool.remove(pref_worker);
                    w.track_claimed_worker(ClaimedWorker::new_claim(guild_id, &worker));
                    info!("found worker in preferred search");
                    record_claim_wait("preferred", priority_index, started_at);
                    return worker;
                }

                if i == 0 {
                    break;
                }

                i -= 1;
            }

            // take the least recently used worker
            let mut i = priority_index;
            loop {
                let pool = &mut w.pools[i];

                let mut candidate = None;
                let mut canditate_age = Duration::MAX;
                for (i, worker) in pool.iter().enumerate() {
                    let elapsed = Instant::elapsed(&worker.returned_at);
                    if candidate.is_none() || elapsed > canditate_age {
                        candidate = Some(i);
                        canditate_age = elapsed;
                    }
                }

                if let Some(can) = candidate {
                    metrics::gauge!("bl.scheduler.workerpool_available_workers", "priority_index" => i.to_string()).decrement(1.0);
                    let worker = pool.remove(can);
                    w.track_claimed_worker(ClaimedWorker::new_claim(guild_id, &worker));
                    info!("found worker in least recently used search");
                    record_claim_wait("lru", priority_index, started_at);