mysten_service/
service.rs

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
// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use crate::health::HealthResponse;
use crate::DEFAULT_PORT;
use anyhow::Result;
use axum::routing::get;
use axum::Json;
use axum::Router;
use tracing::debug;

pub fn get_mysten_service<S>(app_name: &str, app_version: &str) -> Router<S>
where
    S: Send + Clone + Sync + 'static,
{
    // build our application with a single route
    Router::new().route(
        "/health",
        get(Json(HealthResponse::new(app_name, app_version))),
    )
}

pub async fn serve(app: Router) -> Result<()> {
    // run it with hyper on localhost:3000
    debug!("listening on http://localhost:{}", DEFAULT_PORT);

    let listener = tokio::net::TcpListener::bind(&format!("0.0.0.0:{}", DEFAULT_PORT))
        .await
        .unwrap();
    axum::serve(listener, app).await?;
    Ok(())
}