[rocketmq-client-rust] branch develop updated: Fix use statements

[li...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

lizhanhui pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/rocketmq-client-rust.git


The following commit(s) were added to refs/heads/develop by this push:
     new 0e1e3dd  Fix use statements
0e1e3dd is described below

commit 0e1e3ddef84f6f0c35f7b2f1e4a2c0aee0a56537
Author: Li Zhanhui <li...@gmail.com>
AuthorDate: Sat Apr 2 08:27:01 2022 +0000

    Fix use statements
---
 src/rocketmq.rs | 12 +++++++++++-
 src/service.rs  | 24 +++++++++++++-----------
 2 files changed, 24 insertions(+), 12 deletions(-)

diff --git a/src/rocketmq.rs b/src/rocketmq.rs
index 8c4f440..7ee369d 100644
--- a/src/rocketmq.rs
+++ b/src/rocketmq.rs
@@ -1,6 +1,5 @@
 use crate::pb::{
     messaging_service_client::MessagingServiceClient, QueryRouteRequest, QueryRouteResponse,
-    Resource,
 };
 use rustls::client::ServerCertVerifier;
 use tonic::{
@@ -66,6 +65,7 @@ impl RpcClient {
 #[cfg(test)]
 mod test {
     use super::*;
+    use crate::pb::{Resource, Code};
 
     #[tokio::test]
     async fn test_connect() {
@@ -73,6 +73,14 @@ mod test {
         let mut rpc_client = RpcClient::new(target)
             .await
             .expect("Should be able to connect");
+    }
+
+    #[tokio::test]
+    async fn test_query_route() {
+        let target = "http://127.0.0.1:5001";
+        let mut rpc_client = RpcClient::new(target)
+            .await
+            .expect("Should be able to connect");
         let topic = Resource {
             resource_namespace: String::from("arn"),
             name: String::from("TestTopic"),
@@ -86,5 +94,7 @@ mod test {
             .query_route(request)
             .await
             .expect("Failed to query route");
+        let route_response = reply.into_inner();
+        assert_eq!(route_response.status.unwrap().code, Code::Ok as i32);
     }
 }
diff --git a/src/service.rs b/src/service.rs
index 08c2c8f..e39a17b 100644
--- a/src/service.rs
+++ b/src/service.rs
@@ -1,16 +1,15 @@
 use crate::pb::{
-    messaging_service_server::MessagingService, messaging_service_server::MessagingServiceServer,
-    AckMessageRequest, AckMessageResponse, ChangeInvisibleDurationRequest,
-    ChangeInvisibleDurationResponse, EndTransactionRequest, EndTransactionResponse,
-    ForwardMessageToDeadLetterQueueRequest, ForwardMessageToDeadLetterQueueResponse,
-    HeartbeatRequest, HeartbeatResponse, NotifyClientTerminationRequest,
-    NotifyClientTerminationResponse, PullMessageRequest, PullMessageResponse,
-    QueryAssignmentRequest, QueryAssignmentResponse, QueryOffsetRequest, QueryOffsetResponse,
-    QueryRouteRequest, QueryRouteResponse, ReceiveMessageRequest, ReceiveMessageResponse,
-    SendMessageRequest, SendMessageResponse, TelemetryCommand,
+    messaging_service_server::MessagingService, AckMessageRequest, AckMessageResponse,
+    ChangeInvisibleDurationRequest, ChangeInvisibleDurationResponse, EndTransactionRequest,
+    EndTransactionResponse, ForwardMessageToDeadLetterQueueRequest,
+    ForwardMessageToDeadLetterQueueResponse, HeartbeatRequest, HeartbeatResponse,
+    NotifyClientTerminationRequest, NotifyClientTerminationResponse, PullMessageRequest,
+    PullMessageResponse, QueryAssignmentRequest, QueryAssignmentResponse, QueryOffsetRequest,
+    QueryOffsetResponse, QueryRouteRequest, QueryRouteResponse, ReceiveMessageRequest,
+    ReceiveMessageResponse, SendMessageRequest, SendMessageResponse, TelemetryCommand,
 };
 use futures::Stream;
-use tonic::{transport::Server, Request, Response, Status, Streaming};
+use tonic::{Request, Response, Status, Streaming};
 
 #[derive(Default)]
 pub struct ServerService {}
@@ -27,7 +26,10 @@ impl MessagingService for ServerService {
     ) -> Result<Response<QueryRouteResponse>, Status> {
         println!("{:?}", request);
         let reply = QueryRouteResponse {
-            status: None,
+            status: Some(crate::pb::Status {
+                code: crate::pb::Code::Ok as i32,
+                message: String::from("OK"),
+            }),
             message_queues: vec![],
         };
         Ok(Response::new(reply))