refactor(config): add provider config

Author: yang20150702

config/src/config.rs

@@ -21,6 +21,7 @@ use lazy_static::lazy_static;
 use serde::{Deserialize, Serialize};
 
 use super::protocol::ProtocolConfig;
+use super::provider::ProviderConfig;
 use super::service::ServiceConfig;
 
 pub const DUBBO_CONFIG_PATH: &str = "./dubbo.yaml";
@@ -35,10 +36,14 @@ lazy_static! {
 #[derive(Debug, Default, Serialize, Deserialize, Clone)]
 pub struct RootConfig {
     pub name: String,
+
+    #[serde(skip_serializing, skip_deserializing)]
     pub service: HashMap<String, ServiceConfig>,
     pub protocols: HashMap<String, ProtocolConfig>,
     pub registries: HashMap<String, String>,
 
+    pub provider: ProviderConfig,
+
     #[serde(skip_serializing, skip_deserializing)]
     pub data: HashMap<String, String>,
 }
@@ -65,6 +70,7 @@ impl RootConfig {
             service: HashMap::new(),
             protocols: HashMap::new(),
             registries: HashMap::new(),
+            provider: ProviderConfig::new(),
             data: HashMap::new(),
         }
     }
@@ -96,6 +102,10 @@ impl RootConfig {
     }
 
     pub fn test_config(&mut self) {
+        let mut provider = ProviderConfig::new();
+        provider.protocol_ids = vec!["triple".to_string()];
+        provider.registry_ids = vec![];
+
         let service_config = ServiceConfig::default()
             .group("test".to_string())
             .serializer("json".to_string())
@@ -127,6 +137,10 @@ impl RootConfig {
                 .ip("0.0.0.0".to_string())
                 .port("8889".to_string()),
         );
+
+        provider.services = self.service.clone();
+        self.provider = provider.clone();
+        println!("provider config: {:?}", provider);
         // 通过环境变量读取某个文件。加在到内存中
         self.data.insert(
             "dubbo.provider.url".to_string(),
@@ -172,6 +186,12 @@ pub trait Config {
 mod tests {
     use super::*;
 
+    #[test]
+    fn test_config() {
+        let mut r = RootConfig::new();
+        r.test_config();
+    }
+
     #[test]
     fn test_load() {
         // case 1: read config yaml from default path

config/src/lib.rs

@@ -17,6 +17,7 @@
 
 pub mod config;
 pub mod protocol;
+pub mod provider;
 pub mod service;
 
 pub use config::*;

config/src/provider.rs

@@ -0,0 +1,56 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * 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 serde::{Deserialize, Serialize};
+
+use super::service::ServiceConfig;
+
+#[derive(Debug, Default, Serialize, Deserialize, Clone)]
+pub struct ProviderConfig {
+    pub registry_ids: Vec<String>,
+
+    pub protocol_ids: Vec<String>,
+
+    pub services: HashMap<String, ServiceConfig>,
+}
+
+impl ProviderConfig {
+    pub fn new() -> Self {
+        ProviderConfig {
+            registry_ids: vec![],
+            protocol_ids: vec![],
+            services: HashMap::new(),
+        }
+    }
+
+    pub fn with_registry_ids(mut self, registry_ids: Vec<String>) -> Self {
+        self.registry_ids = registry_ids;
+        self
+    }
+
+    pub fn with_protocol_ids(mut self, protocol_ids: Vec<String>) -> Self {
+        self.protocol_ids = protocol_ids;
+        self
+    }
+
+    pub fn with_services(mut self, services: HashMap<String, ServiceConfig>) -> Self {
+        self.services = services;
+        self
+    }
+}