spi-conf: Spi conf add more nacos error codes (#484)

* add more error code to nacos grpc protocal

* fix test

Author: 4t145

services/spi-conf/src/main.rs

@@ -11,7 +11,7 @@ mod initializer;
 async fn main() -> TardisResult<()> {
     TardisFuns::init(Some("config")).await?;
     let web_server = TardisFuns::web_server();
-    initializer::init(web_server).await?;
+    initializer::init(&web_server).await?;
     web_server.start().await?;
     web_server.await;
     Ok(())

spi/spi-conf/src/api/nacos/grpc/mod.rs

@@ -106,6 +106,26 @@ impl NaocsGrpcResponse {
             request_id: None,
         }
     }
+    pub const fn not_found() -> Self {
+        Self {
+            result_code: 500,
+            error_code: Some(300),
+            message: None,
+            request_id: None,
+        }
+    }
+    pub const fn unregister() -> Self {
+        Self {
+            result_code: 500,
+            error_code: Some(301),
+            message: None,
+            request_id: None,
+        }
+    }
+}
+
+impl AsPayload for NaocsGrpcResponse {
+    const TYPE_NAME: &'static str = "Response";
 }
 
 #[derive(Clone, Debug, Serialize, Deserialize)]
@@ -170,6 +190,24 @@ pub struct ConfigQueryResponse {
     pub response: NaocsGrpcResponse,
 }
 
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub struct ConfigQueryResponseNotFound {
+    #[serde(flatten)]
+    pub response: NaocsGrpcResponse,
+}
+
+impl Default for ConfigQueryResponseNotFound {
+    fn default() -> Self {
+        ConfigQueryResponseNotFound {
+            response: NaocsGrpcResponse::not_found(),
+        }
+    }
+}
+
+impl AsPayload for ConfigQueryResponseNotFound {
+    const TYPE_NAME: &'static str = "ConfigQueryResponse";
+}
+
 impl From<ConfigItem> for ConfigQueryResponse {
     fn from(item: ConfigItem) -> Self {
         Self {
@@ -238,16 +276,20 @@ pub async fn dispatch_request(type_info: &str, value: &str, access_token: Option
         "ServerCheckRequest" => ServerCheckResponse::success(None).as_payload(),
         "HealthCheckRequest" => HealthCheckResponse::success().as_payload(),
         "ConfigQueryRequest" => {
-            let ctx = get_ctx.await?;
+            let Ok(ctx) = get_ctx.await else {
+                return Ok(NaocsGrpcResponse::unregister().as_payload())
+            };
             let ConfigQueryRequest { data_id, group, tenant } = serde_json::from_str(value).map_err(|_e| TardisError::bad_request("expect a ConfigQueryRequest", ""))?;
             let mut descriptor = ConfigDescriptor {
                 namespace_id: tenant.unwrap_or("public".into()),
                 data_id,
                 group,
                 ..Default::default()
             };
-            let result: ConfigQueryResponse = get_config_detail(&mut descriptor, &funs, &ctx).await?.into();
-            result.as_payload()
+            match get_config_detail(&mut descriptor, &funs, &ctx).await {
+                Ok(data) => ConfigQueryResponse::from(data).as_payload(),
+                Err(_) => ConfigQueryResponseNotFound::default().as_payload(),
+            }
         }
         "ConfigBatchListenRequest" => {
             let ctx = get_ctx.await?;

spi/spi-conf/tests/spi_conf_test_common.rs

@@ -40,7 +40,7 @@ pub async fn init_tardis(docker: &Cli) -> TardisResult<Holder> {
     bios_basic::rbum::rbum_initializer::init(DOMAIN_CODE, RbumConfig::default()).await?;
     let web_server = TardisFuns::web_server();
     rbum_initializer::init("bios-spi", RbumConfig::default()).await?;
-    conf_initializer::init(web_server).await?;
+    conf_initializer::init(&web_server).await?;
     Ok(holder)
 }