controller.rs

// Copyright 2022 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use anyhow::Result;
use base::ReadNotifier;
use base::WaitContext;

use crate::controller::MetricsController;
use crate::controller::MetricsControllerToken;

impl MetricsController {
    pub(crate) fn run_internal(&mut self) -> Result<()> {
        let wait_ctx: WaitContext<MetricsControllerToken> = WaitContext::new()?;
        self.closed_tubes = 0;

        for (agent_index, agent) in self.agents.iter().enumerate() {
            wait_ctx.add(
                agent.get_read_notifier(),
                MetricsControllerToken::Agent(agent_index),
            )?;
        }

        'listen: loop {
            let events = wait_ctx.wait()?;
            for event in events.iter().filter(|e| e.is_readable) {
                let MetricsControllerToken::Agent(client_index) = event.token;
                self.on_tube_readable(&self.agents[client_index]);
            }
            for event in events.iter().filter(|e| e.is_hungup) {
                let MetricsControllerToken::Agent(client_index) = event.token;
                wait_ctx.delete(self.agents[client_index].get_read_notifier())?;
                if self.on_connection_closed() {
                    break 'listen;
                }
            }
        }
        Ok(())
    }
}