add status check grpc (#13)

0glabs · Jul 4, 2024 · 8be1b87 · 8be1b87
1 parent f39a765
commit 8be1b87
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/grpc/proto/signer.proto b/grpc/proto/signer.proto
@@ -7,6 +7,7 @@ service Signer {
   rpc BatchSign(BatchSignRequest) returns (BatchSignReply) {}
   // This retrieves the requested encoded rows from the DA node database.
   rpc BatchRetrieve(BatchRetrieveRequest) returns (BatchRetrieveReply) {}
+  rpc GetStatus(Empty) returns (StatusReply) {}
 }
 
 message SignRequest {
@@ -53,4 +54,10 @@ message Slices {
 message BatchRetrieveReply {
   // required encoded slices
   repeated Slices encoded_slice = 1;
-}
+}
+
+message StatusReply {
+  uint64 status_code = 1;
+}
+
+message Empty {}
diff --git a/grpc/src/service.rs b/grpc/src/service.rs
@@ -13,7 +13,7 @@ use ethers::types::{Res, U256};
 use ethers::utils::keccak256;
 use prost::Message;
 use rayon::iter::{IndexedParallelIterator, IntoParallelRefIterator, ParallelIterator};
-use signer::{BatchRetrieveReply, BatchRetrieveRequest, Slices};
+use signer::{BatchRetrieveReply, BatchRetrieveRequest, Empty, Slices};
 use std::sync::Arc;
 use std::time::{Duration, Instant};
 use storage::blob_status_db::{BlobStatus, BlobStatusDB};
@@ -246,6 +246,16 @@ impl Signer for SignerService {
     ) -> Result<Response<BatchRetrieveReply>, Status> {
         self.batch_retrieve_inner(request).await
     }
+
+    async fn get_status(
+        &self,
+        request: Request<Empty>,
+    ) -> Result<Response<signer::StatusReply>, Status> {
+        let status = signer::StatusReply {
+            status_code: 200
+        };
+        Ok(Response::new(status))
+    }
 }
 
 pub enum VerificationError {