Move AC to inner functions

TraceMachina · Dec 30, 2020 · 5452f4b · 5452f4b
1 parent 3e8c210
commit 5452f4b
Showing 1 changed file with 25 additions and 5 deletions.
diff --git a/cas/grpc_service/ac_server.rs b/cas/grpc_service/ac_server.rs
@@ -32,11 +32,8 @@ impl AcServer {
     pub fn into_service(self) -> Server<AcServer> {
         Server::new(self)
     }
-}
 
-#[tonic::async_trait]
-impl ActionCache for AcServer {
-    async fn get_action_result(
+    async fn inner_get_action_result(
         &self,
         grpc_request: Request<GetActionResultRequest>,
     ) -> Result<Response<ActionResult>, Status> {
@@ -71,7 +68,7 @@ impl ActionCache for AcServer {
         Ok(Response::new(action_result))
     }
 
-    async fn update_action_result(
+    async fn inner_update_action_result(
         &self,
         grpc_request: Request<UpdateActionResultRequest>,
     ) -> Result<Response<ActionResult>, Status> {
@@ -116,3 +113,26 @@ impl ActionCache for AcServer {
         Ok(Response::new(action_result))
     }
 }
+
+#[tonic::async_trait]
+impl ActionCache for AcServer {
+    async fn get_action_result(
+        &self,
+        grpc_request: Request<GetActionResultRequest>,
+    ) -> Result<Response<ActionResult>, Status> {
+        println!("get_action_result Req: {:?}", grpc_request);
+        let resp = self.inner_get_action_result(grpc_request).await;
+        println!("get_action_result Resp: {:?}", resp);
+        return resp;
+    }
+
+    async fn update_action_result(
+        &self,
+        grpc_request: Request<UpdateActionResultRequest>,
+    ) -> Result<Response<ActionResult>, Status> {
+        println!("update_action_result Req: {:?}", grpc_request);
+        let resp = self.inner_update_action_result(grpc_request).await;
+        println!("update_action_result Resp: {:?}", resp);
+        return resp;
+    }
+}