tobira 0.3.3

A VMess relay written in Rust.
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

use std::sync::Arc;
use std::time::Duration;

use tokio::sync::RwLock;

use crate::relay::transport::grpc::GrpcPool;
use crate::vmess::validator::Validator;

#[derive(Clone)]
pub struct RelayRuntime {
    pub validator: Arc<RwLock<Validator>>,
    pub grpc_pool: Arc<GrpcPool>,
    pub relay_idle_timeout: Arc<RwLock<Option<Duration>>>,
}

impl RelayRuntime {
    pub fn new(
        validator: Arc<RwLock<Validator>>,
        grpc_pool: Arc<GrpcPool>,
        relay_idle_timeout: Option<Duration>,
    ) -> Self {
        Self {
            validator,
            grpc_pool,
            relay_idle_timeout: Arc::new(RwLock::new(relay_idle_timeout)),
        }
    }

    pub async fn set_relay_idle_timeout(&self, relay_idle_timeout: Option<Duration>) {
        *self.relay_idle_timeout.write().await = relay_idle_timeout;
    }
}