proksi 0.1.4

A batteries-included reverse proxy with automatic HTTPS using Cloudflare Pingora and Let's Encrypt.
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

use std::{collections::HashMap, sync::Arc};

use pingora_load_balancing::{selection::RoundRobin, LoadBalancer};

/// A store for routes that is updated in a background thread
#[derive(Clone)]
pub struct RouteStore {
    routes: HashMap<String, Arc<LoadBalancer<RoundRobin>>>,
}

impl RouteStore {
    pub fn new() -> Self {
        RouteStore {
            routes: HashMap::new(),
        }
    }

    pub fn get_route_keys(&self) -> Vec<String> {
        self.routes.keys().cloned().collect()
    }

    /// Adds a new route using a hostname and a LoadBalancer instance wrapped in an `Arc`
    pub fn add_route(&mut self, route: String, upstream: Arc<LoadBalancer<RoundRobin>>) {
        self.routes.insert(route, upstream);
    }

    /// Gets a route from the store
    pub fn get_route(&self, route: &str) -> Option<Arc<LoadBalancer<RoundRobin>>> {
        self.routes.get(route).cloned()
    }

    /// Optimistically removes a route from the store
    pub fn _remove_route(&mut self, route: &str) -> bool {
        self.routes.remove_entry(route);

        true
    }
}