examples/diagnostics/inspect/codelab/rust/part_3/src/main.rs

// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use {
    crate::reverser::ReverserServerFactory,
    anyhow::{Context, Error},
    fidl_fuchsia_examples_inspect::FizzBuzzMarker,
    fuchsia_component::{client, server::ServiceFs},
    fuchsia_inspect::{component, health::Reporter},
    futures::{FutureExt, StreamExt},
    tracing::info,
};

mod reverser;

#[fuchsia::component(logging_tags = ["inspect_rust_codelab", "part3"])]
async fn main() -> Result<(), Error> {
    let mut fs = ServiceFs::new();

    info!("starting up...");

    inspect_runtime::serve(component::inspector(), &mut fs)?;

    // ComponentInspector has built-in health checking. Set it to "starting up" so snapshots show
    // we may still be initializing.
    component::health().set_starting_up();

    // Create a version string. We use record_ rather than create_ to tie the lifecyle of the
    // inspector root with the string property.
    // It is an error to not retain the created property.
    component::inspector().root().record_string("version", "part3");

    // Create a new Reverser Server factory. The factory holds global stats for the reverser
    // server.
    let reverser_factory =
        ReverserServerFactory::new(component::inspector().root().create_child("reverser_service"));

    // Serve the reverser service
    fs.dir("svc").add_fidl_service(move |stream| reverser_factory.spawn_new(stream));
    fs.take_and_serve_directory_handle()?;

    // Send a request to the FizzBuzz service and print the response when it arrives.
    let fizzbuzz =
        client::connect_to_protocol::<FizzBuzzMarker>().context("failed to connect to fizzbuzz")?;
    match fizzbuzz.execute(30u32).await {
        Ok(result) => {
            component::health().set_ok();
            info!(%result, "Got FizzBuzz");
        }
        Err(_) => {
            component::health().set_unhealthy("FizzBuzz connection closed");
        }
    };

    fs.collect::<()>().map(Ok).await
}