1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

// 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(())
    }
}