重新开始rust学习

Author: chuntaojun

.gitignore

@@ -1,3 +1,5 @@
 /target
 Cargo.lock
-.idea
+.idea
+.zed
+.vscode

crates/polaris/src/cfg.rs

@@ -0,0 +1 @@
+

src/core/config/config.rs

@@ -1,25 +1,25 @@
 // Tencent is pleased to support the open source community by making Polaris available.
-// 
+//
 // Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
-// 
+//
 // Licensed under the BSD 3-Clause License (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
-// 
+//
 // https://opensource.org/licenses/BSD-3-Clause
-// 
+//
 // Unless required by applicable law or agreed to in writing, software distributed
 // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 // CONDITIONS OF ANY KIND, either express or implied. See the License for the
 // specific language governing permissions and limitations under the License.
 
-use std::{env, fs, io};
-use std::path::Path;
-use serde::Deserialize;
 use crate::core::config::config_file::ConfigFileConfig;
 use crate::core::config::consumer::ConsumerConfig;
 use crate::core::config::global::GlobalConfig;
 use crate::core::config::provider::ProviderConfig;
+use serde::Deserialize;
+use std::path::Path;
+use std::{env, fs, io};
 
 #[derive(Deserialize, Debug)]
 #[serde(rename_all = "camelCase", deny_unknown_fields)]
@@ -36,25 +36,26 @@ pub fn load_default<'a>() -> Result<Configuration, io::Error> {
         path = Path::new("./polaris.yml");
     }
     if env::var("POLARIS_RUST_CONFIG").is_ok() {
-        return load(env::var("POLARIS_RUST_CONFIG").unwrap())
+        return load(env::var("POLARIS_RUST_CONFIG").unwrap());
     }
-    return load(path)
+    return load(path);
 }
 
 pub fn load<P: AsRef<Path>>(path: P) -> Result<Configuration, io::Error> {
     let val = fs::read_to_string(path);
     if val.is_ok() {
         let data = val.ok().unwrap();
-        let config: Configuration = serde_yaml::from_str(&*data).expect(&format!("failure to format yaml str {}", &data));
-        return Ok(config)
+        let config: Configuration =
+            serde_yaml::from_str(&*data).expect(&format!("failure to format yaml str {}", &data));
+        return Ok(config);
     }
-    return Err(val.err().unwrap())
+    return Err(val.err().unwrap());
 }
 
 #[cfg(test)]
 mod tests {
-    use std::env;
     use crate::core::config::config::load;
+    use std::env;
 
     #[test]
     fn test_load() {

src/core/engine.rs

@@ -1,13 +1,13 @@
 // Tencent is pleased to support the open source community by making Polaris available.
-// 
+//
 // Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
-// 
+//
 // Licensed under the BSD 3-Clause License (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
-// 
+//
 // https://opensource.org/licenses/BSD-3-Clause
-// 
+//
 // Unless required by applicable law or agreed to in writing, software distributed
 // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 // CONDITIONS OF ANY KIND, either express or implied. See the License for the
@@ -19,14 +19,15 @@ use crate::core::model::error::PolarisError;
 use crate::core::model::naming::InstanceRequest;
 use crate::core::plugin::connector::Connector;
 use crate::core::plugin::plugins::Extensions;
-use crate::discovery::req::{InstanceDeregisterRequest, InstanceRegisterRequest, InstanceRegisterResponse};
+use crate::discovery::req::{
+    InstanceDeregisterRequest, InstanceRegisterRequest, InstanceRegisterResponse,
+};
 
 pub struct Engine {
-    extensions: Extensions
+    extensions: Extensions,
 }
 
 impl Engine {
-
     pub fn new() -> Self {
         Self {
             extensions: Extensions::default(),
@@ -41,40 +42,44 @@ impl Engine {
         return Ok(true);
     }
 
-    pub(crate) fn sync_register_instance(&mut self, req: InstanceRegisterRequest) -> Result<InstanceRegisterResponse, PolarisError> {
-        let mut connector = self.extensions.get_active_connector(DISCOVER_SERVER_CONNECTOR);
-        let rsp = connector.register_instance(InstanceRequest{
+    pub(crate) fn sync_register_instance(
+        &mut self,
+        req: InstanceRegisterRequest,
+    ) -> Result<InstanceRegisterResponse, PolarisError> {
+        let connector = self
+            .extensions
+            .get_active_connector(DISCOVER_SERVER_CONNECTOR);
+        let rsp = connector.register_instance(InstanceRequest {
             flow_id: req.flow_id.clone(),
             ttl: req.ttl.clone(),
             instance: req.convert_instance(),
         });
 
         return match rsp {
-            Ok(ins_rsp) => {
-                Ok(InstanceRegisterResponse{ instance_id: ins_rsp.instance.ip.clone(), exist: ins_rsp.exist.clone() })
-            }
-            Err(err) => {
-                Err(err)
-            }
-        }
+            Ok(ins_rsp) => Ok(InstanceRegisterResponse {
+                instance_id: ins_rsp.instance.ip.clone(),
+                exist: ins_rsp.exist.clone(),
+            }),
+            Err(err) => Err(err),
+        };
     }
 
-    pub(crate) fn sync_deregister_instance(&mut self, req: InstanceDeregisterRequest) -> Result<(), PolarisError> {
-        let mut connector = self.extensions.get_active_connector(DISCOVER_SERVER_CONNECTOR);
-        let rsp = connector.deregister_instance(InstanceRequest{
+    pub(crate) fn sync_deregister_instance(
+        &mut self,
+        req: InstanceDeregisterRequest,
+    ) -> Result<(), PolarisError> {
+        let mut connector = self
+            .extensions
+            .get_active_connector(DISCOVER_SERVER_CONNECTOR);
+        let rsp = connector.deregister_instance(InstanceRequest {
             flow_id: req.flow_id.clone(),
             ttl: 0,
             instance: req.convert_instance(),
         });
 
         return match rsp {
-            Ok(ins_rsp) => {
-                Ok(())
-            }
-            Err(err) => {
-                Err(err)
-            }
-        }
+            Ok(ins_rsp) => Ok(()),
+            Err(err) => Err(err),
+        };
     }
-
 }

src/core/model/cluster.rs

@@ -1,13 +1,13 @@
 // Tencent is pleased to support the open source community by making Polaris available.
-// 
+//
 // Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
-// 
+//
 // Licensed under the BSD 3-Clause License (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
-// 
+//
 // https://opensource.org/licenses/BSD-3-Clause
-// 
+//
 // Unless required by applicable law or agreed to in writing, software distributed
 // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 // CONDITIONS OF ANY KIND, either express or implied. See the License for the
@@ -20,13 +20,13 @@ use std::time::Duration;
 pub static BUILDIN_SERVER_NAMESPACE: &str = "Polaris";
 pub static BUILDIN_SERVER_SERVICE: &str = "polaris.builtin";
 
-#[derive(Default, Debug, PartialEq, Eq, Clone, Copy)]
+#[derive(Default, Debug, PartialEq, Eq, Clone, Copy, Hash)]
 pub enum ClusterType {
     #[default]
     BuildIn,
     Discover,
     Config,
-    HealthCheck
+    HealthCheck,
 }
 
 impl Display for ClusterType {

src/core/plugin/connector.rs

@@ -1,35 +1,34 @@
 // Tencent is pleased to support the open source community by making Polaris available.
-// 
+//
 // Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
-// 
+//
 // Licensed under the BSD 3-Clause License (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
-// 
+//
 // https://opensource.org/licenses/BSD-3-Clause
-// 
+//
 // Unless required by applicable law or agreed to in writing, software distributed
 // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 // CONDITIONS OF ANY KIND, either express or implied. See the License for the
 // specific language governing permissions and limitations under the License.
 
-use std::collections::HashMap;
-use std::sync::Arc;
 use crate::core::model::error::{ErrorCode, PolarisError};
 use crate::core::model::naming::{InstanceRequest, InstanceResponse};
 use crate::core::plugin::plugins::{Extensions, Plugin};
+use std::collections::HashMap;
+use std::sync::Arc;
 
 pub trait Connector: Plugin {
-
     // register_resource_handler 注册资源处理器
     fn register_resource_handler(&self) -> Result<bool, PolarisError>;
 
     // deregister_resource_handler
     fn deregister_resource_handler(&self) -> Result<bool, PolarisError>;
 
-    fn register_instance(&mut self, req: InstanceRequest) -> Result<InstanceResponse, PolarisError>;
+    fn register_instance(&self, req: InstanceRequest) -> Result<InstanceResponse, PolarisError>;
 
-    fn deregister_instance(&mut self, req: InstanceRequest) -> Result<bool, PolarisError>;
+    fn deregister_instance(&self, req: InstanceRequest) -> Result<bool, PolarisError>;
 
     fn heartbeat_instance(&self, req: InstanceRequest) -> Result<bool, PolarisError>;
 
@@ -48,18 +47,16 @@ pub trait Connector: Plugin {
     fn upsert_publish_config_file(&self) -> Result<bool, PolarisError>;
 }
 
-pub struct NoopConnector {
-
-}
+pub struct NoopConnector {}
 
 impl Default for NoopConnector {
     fn default() -> Self {
-        Self{}
+        Self {}
     }
 }
 
 impl Plugin for NoopConnector {
-    fn init(&self, options: HashMap<String, String>, extensions: Arc<Extensions>) {
+    fn init(&mut self, options: HashMap<String, String>, extensions: Arc<Extensions>) {
         todo!()
     }
 
@@ -81,11 +78,11 @@ impl Connector for NoopConnector {
         todo!()
     }
 
-    fn register_instance(&mut self, req: InstanceRequest) -> Result<InstanceResponse, PolarisError> {
+    fn register_instance(&self, req: InstanceRequest) -> Result<InstanceResponse, PolarisError> {
         todo!()
     }
 
-    fn deregister_instance(&mut self, req: InstanceRequest) -> Result<bool, PolarisError> {
+    fn deregister_instance(&self, req: InstanceRequest) -> Result<bool, PolarisError> {
         todo!()
     }