[Indexer-Grpc-V2] Add GrpcManagerService.

ecosystem/indexer-grpc/indexer-grpc-manager/Cargo.toml

@@ -15,10 +15,17 @@ rust-version = { workspace = true }
 [dependencies]
 anyhow = { workspace = true }
 aptos-indexer-grpc-server-framework = { workspace = true }
+aptos-protos = { workspace = true }
 async-trait = { workspace = true }
 clap = { workspace = true }
 serde = { workspace = true }
 tokio = { workspace = true }
+tokio-scoped = { workspace = true }
+tonic = { workspace = true }
+tracing = { workspace = true }
+
+[dev-dependencies]
+aptos-config = { workspace = true }
 
 [target.'cfg(unix)'.dependencies]
 jemallocator = { workspace = true }

ecosystem/indexer-grpc/indexer-grpc-manager/src/config.rs

@@ -1,18 +1,34 @@
 // Copyright © Aptos Foundation
 // SPDX-License-Identifier: Apache-2.0
 
+use crate::grpc_manager::GrpcManager;
 use anyhow::Result;
 use aptos_indexer_grpc_server_framework::RunnableConfig;
 use serde::{Deserialize, Serialize};
+use std::net::SocketAddr;
+use tokio::sync::OnceCell;
+
+static GRPC_MANAGER: OnceCell<GrpcManager> = OnceCell::const_new();
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct ServiceConfig {
+    pub(crate) listen_address: SocketAddr,
+}
 
 #[derive(Clone, Debug, Deserialize, Serialize)]
 #[serde(deny_unknown_fields)]
-pub struct IndexerGrpcManagerConfig {}
+pub struct IndexerGrpcManagerConfig {
+    pub(crate) chain_id: u64,
+    pub(crate) service_config: ServiceConfig,
+}
 
 #[async_trait::async_trait]
 impl RunnableConfig for IndexerGrpcManagerConfig {
     async fn run(&self) -> Result<()> {
-        Ok(())
+        GRPC_MANAGER
+            .get_or_init(|| async { GrpcManager::new(self).await })
+            .await
+            .start(&self.service_config)
     }
 
     fn get_server_name(&self) -> String {

ecosystem/indexer-grpc/indexer-grpc-manager/src/grpc_manager.rs

@@ -0,0 +1,46 @@
+// Copyright © Aptos Foundation
+// SPDX-License-Identifier: Apache-2.0
+
+use crate::{
+    config::{IndexerGrpcManagerConfig, ServiceConfig},
+    service::GrpcManagerService,
+};
+use anyhow::Result;
+use aptos_protos::indexer::v1::grpc_manager_server::GrpcManagerServer;
+use std::time::Duration;
+use tonic::{codec::CompressionEncoding, transport::Server};
+use tracing::info;
+
+const HTTP2_PING_INTERVAL_DURATION: Duration = Duration::from_secs(60);
+const HTTP2_PING_TIMEOUT_DURATION: Duration = Duration::from_secs(10);
+
+pub(crate) struct GrpcManager {
+    chain_id: u64,
+}
+
+impl GrpcManager {
+    pub(crate) async fn new(config: &IndexerGrpcManagerConfig) -> Self {
+        let chain_id = config.chain_id;
+
+        Self { chain_id }
+    }
+
+    pub(crate) fn start(&self, service_config: &ServiceConfig) -> Result<()> {
+        let service = GrpcManagerServer::new(GrpcManagerService::new(self.chain_id))
+            .send_compressed(CompressionEncoding::Zstd)
+            .accept_compressed(CompressionEncoding::Zstd);
+        let server = Server::builder()
+            .http2_keepalive_interval(Some(HTTP2_PING_INTERVAL_DURATION))
+            .http2_keepalive_timeout(Some(HTTP2_PING_TIMEOUT_DURATION))
+            .add_service(service);
+
+        tokio_scoped::scope(|s| {
+            s.spawn(async move {
+                info!("Starting GrpcManager at {}.", service_config.listen_address);
+                server.serve(service_config.listen_address).await.unwrap();
+            });
+        });
+
+        Ok(())
+    }
+}

ecosystem/indexer-grpc/indexer-grpc-manager/src/lib.rs

@@ -2,3 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 
 pub mod config;
+mod grpc_manager;
+mod service;
+#[cfg(test)]
+mod test;

ecosystem/indexer-grpc/indexer-grpc-manager/src/service.rs

@@ -0,0 +1,109 @@
+// Copyright (c) Aptos Foundation
+// SPDX-License-Identifier: Apache-2.0
+
+use aptos_protos::indexer::v1::{
+    grpc_manager_server::GrpcManager, service_info::Info, GetDataServiceForRequestRequest,
+    GetDataServiceForRequestResponse, GetTransactionsRequest, HeartbeatRequest, HeartbeatResponse,
+    TransactionsResponse,
+};
+use tonic::{Request, Response, Status};
+
+pub struct GrpcManagerService {
+    chain_id: u64,
+}
+
+impl GrpcManagerService {
+    pub(crate) fn new(chain_id: u64) -> Self {
+        Self { chain_id }
+    }
+
+    async fn handle_heartbeat(
+        &self,
+        _address: String,
+        _info: Info,
+    ) -> anyhow::Result<Response<HeartbeatResponse>> {
+        // TODO(grao): Implement.
+        todo!()
+    }
+
+    fn pick_live_data_service(&self, _starting_version: u64) -> Option<String> {
+        // TODO(grao): Implement.
+        todo!()
+    }
+
+    async fn pick_historical_data_service(&self, _starting_version: u64) -> Option<String> {
+        // TODO(grao): Implement.
+        todo!()
+    }
+}
+
+#[tonic::async_trait]
+impl GrpcManager for GrpcManagerService {
+    async fn heartbeat(
+        &self,
+        request: Request<HeartbeatRequest>,
+    ) -> Result<Response<HeartbeatResponse>, Status> {
+        let request = request.into_inner();
+        if let Some(service_info) = request.service_info {
+            if let Some(address) = service_info.address {
+                if let Some(info) = service_info.info {
+                    return self
+                        .handle_heartbeat(address, info)
+                        .await
+                        .map_err(|e| Status::internal(format!("Error handling heartbeat: {e}")));
+                }
+            }
+        }
+
+        Err(Status::invalid_argument("Bad request."))
+    }
+
+    async fn get_transactions(
+        &self,
+        request: Request<GetTransactionsRequest>,
+    ) -> Result<Response<TransactionsResponse>, Status> {
+        let _request = request.into_inner();
+        let transactions = vec![];
+        // TODO(grao): Implement.
+
+        Ok(Response::new(TransactionsResponse {
+            transactions,
+            chain_id: Some(self.chain_id),
+        }))
+    }
+
+    async fn get_data_service_for_request(
+        &self,
+        request: Request<GetDataServiceForRequestRequest>,
+    ) -> Result<Response<GetDataServiceForRequestResponse>, Status> {
+        let request = request.into_inner();
+
+        if request.user_request.is_none() {
+            return Err(Status::invalid_argument("Bad request."));
+        }
+
+        let user_request = request.user_request.unwrap();
+        if user_request.starting_version.is_none() {
+            return Err(Status::invalid_argument("Bad request."));
+        }
+
+        let starting_version = user_request.starting_version();
+
+        let data_service_address =
+            // TODO(grao): Use a simple strategy for now. Consider to make it smarter in the
+            // future.
+            if let Some(address) = self.pick_live_data_service(starting_version) {
+                address
+            } else if let Some(address) = self.pick_historical_data_service(starting_version).await {
+                address
+            } else {
+                return Err(Status::internal(
+                    "Cannot find a data service instance to serve the provided request.",
+                ));
+            };
+
+        Ok(Response::new(GetDataServiceForRequestResponse {
+            data_service_address,
+        }))
+    }
+}

ecosystem/indexer-grpc/indexer-grpc-manager/src/test.rs

@@ -0,0 +1,25 @@
+// Copyright © Aptos Foundation
+// SPDX-License-Identifier: Apache-2.0
+
+use crate::config::{IndexerGrpcManagerConfig, ServiceConfig};
+use aptos_config::utils::get_available_port;
+use aptos_indexer_grpc_server_framework::RunnableConfig;
+use std::time::Duration;
+
+#[tokio::test(flavor = "multi_thread", worker_threads = 16)]
+async fn test_run() {
+    let port = get_available_port();
+    let listen_address = format!("127.0.0.1:{port}").parse().unwrap();
+    let config = IndexerGrpcManagerConfig {
+        chain_id: 0,
+        service_config: ServiceConfig { listen_address },
+    };
+
+    let task = tokio::spawn(async move {
+        config.run().await.unwrap();
+    });
+
+    tokio::time::sleep(Duration::from_secs(10)).await;
+
+    assert!(!task.is_finished());
+}