gway 0.1.1

proxy
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

use std::{
  collections::{HashMap, HashSet},
  net::SocketAddr,
  sync::Arc,
};

use dashmap::{DashMap, mapref::one::Ref};
use faststr::FastStr;

#[derive(Debug, Clone)]
pub struct SiteConf {
  pub upstream: Arc<Upstream>,
  pub cert_host: FastStr,
}

impl SiteConf {
  pub fn new(upstream: Arc<Upstream>, cert_host: FastStr) -> Self {
    Self {
      upstream,
      cert_host,
    }
  }
}

#[derive(PartialEq, Eq, Debug)]
pub enum Protocol {
  H1,
}

#[derive(Debug)]
pub struct Upstream {
  pub addr_li: Box<[SocketAddr]>,
  pub connect_timeout_sec: u64,
  pub request_timeout_sec: u64,
  pub max_retry: usize,
  pub protocol: Protocol,
}

#[derive(Debug)]
pub struct UpstreamSiteSet {
  pub upstream: Arc<Upstream>,
  pub host_set: HashSet<FastStr>,
}

#[derive(Debug, Default)]
pub struct Route {
  pub host_conf: DashMap<FastStr, SiteConf>,
  pub upstream_site: HashMap<FastStr, UpstreamSiteSet>,
}

impl Route {
  /// 设置 域名 到 服务器 的映射
  pub fn set(
    &mut self,
    host: impl Into<FastStr>,
    cert_host: impl Into<FastStr>,
    upstream_name: impl Into<FastStr>,
  ) -> &mut Self {
    let upstream_name = upstream_name.into();
    let host = host.into();
    if let Some(t) = self.upstream_site.get_mut(&upstream_name) {
      t.host_set.insert(host.clone());
      self
        .host_conf
        .insert(host, SiteConf::new(t.upstream.clone(), cert_host.into()));
    }
    self
  }

  pub fn conf_by_host(&self, host: &str) -> Option<Ref<'_, FastStr, SiteConf>> {
    self.host_conf.get(host)
  }
}