dubbo 0.4.0

dubbo
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

/*
 * 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::sync::Arc;

use crate::{
    cluster::NewCluster, directory::NewCachedDirectory, extension, loadbalancer::NewLoadBalancer,
    route::NewRoutes, utils::boxed_clone::BoxCloneService,
};

use crate::{
    registry::{registry::StaticRegistry, MkRegistryService},
    Url,
};
use aws_smithy_http::body::SdkBody;
use tower::ServiceBuilder;

pub type ClientBoxService =
    BoxCloneService<http::Request<SdkBody>, http::Response<crate::BoxBody>, crate::Error>;

pub type ServiceMK =
    Arc<NewCluster<NewLoadBalancer<NewRoutes<NewCachedDirectory<MkRegistryService>>>>>;

#[derive(Default)]
pub struct ClientBuilder {
    pub timeout: Option<u64>,
    pub connector: &'static str,
    registry_extension_url: Option<Url>,
    pub direct: bool,
}

impl ClientBuilder {
    pub fn new() -> ClientBuilder {
        ClientBuilder {
            timeout: None,
            connector: "",
            registry_extension_url: None,
            direct: false,
        }
    }

    pub fn from_static(host: &str) -> ClientBuilder {
        let registry_extension_url = StaticRegistry::to_extension_url(vec![host.parse().unwrap()]);
        Self {
            timeout: None,
            connector: "",
            registry_extension_url: Some(registry_extension_url),
            direct: true,
        }
    }

    pub fn with_timeout(self, timeout: u64) -> Self {
        Self {
            timeout: Some(timeout),
            ..self
        }
    }

    pub fn with_registry(self, registry: Url) -> Self {
        let registry_extension_url = extension::registry_extension::to_extension_url(registry);
        Self {
            registry_extension_url: Some(registry_extension_url),
            ..self
        }
    }

    pub fn with_host(self, host: &'static str) -> Self {
        let registry_extension_url = StaticRegistry::to_extension_url(vec![host.parse().unwrap()]);

        Self {
            registry_extension_url: Some(registry_extension_url),
            ..self
        }
    }

    pub fn with_connector(self, connector: &'static str) -> Self {
        Self { connector, ..self }
    }

    pub fn with_direct(self, direct: bool) -> Self {
        Self { direct, ..self }
    }

    pub fn build(mut self) -> ServiceMK {
        let registry = self
            .registry_extension_url
            .take()
            .expect("registry must not be empty");

        let mk_service = ServiceBuilder::new()
            .layer(NewCluster::layer())
            .layer(NewLoadBalancer::layer())
            .layer(NewRoutes::layer())
            .layer(NewCachedDirectory::layer())
            .service(MkRegistryService::new(registry));

        Arc::new(mk_service)
    }
}