deepslate 0.1.0

A high-performance Minecraft server proxy written in Rust.
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

//! gRPC service implementation for the Deepslate control plane.

use std::sync::Arc;

use tonic::{Request, Response, Status};

use crate::server::{Server, ServerRegistry};

/// Include the generated protobuf code.
pub mod proto {
    #![allow(clippy::doc_markdown)]
    #![allow(clippy::default_trait_access)]
    #![allow(clippy::missing_const_for_fn)]
    #![allow(clippy::too_many_lines)]
    #![allow(clippy::derive_partial_eq_without_eq)]
    #![allow(clippy::needless_pass_by_value)]
    #![allow(clippy::missing_errors_doc)]

    tonic::include_proto!("deepslate");

    /// Encoded file descriptor set for gRPC server reflection.
    pub const FILE_DESCRIPTOR_SET: &[u8] =
        tonic::include_file_descriptor_set!("deepslate_descriptor");
}

use proto::deepslate_server::Deepslate;
use proto::{
    DeregisterServerRequest, DeregisterServerResponse, ListServersRequest, ListServersResponse,
    RegisterServerRequest, RegisterServerResponse, SetTryOrderRequest, SetTryOrderResponse,
};

/// gRPC service implementation backed by a shared server registry.
pub struct DeepslateService {
    registry: Arc<ServerRegistry>,
}

impl DeepslateService {
    /// Create a new gRPC service with the given server registry.
    #[must_use]
    pub const fn new(registry: Arc<ServerRegistry>) -> Self {
        Self { registry }
    }
}

#[tonic::async_trait]
impl Deepslate for DeepslateService {
    async fn register_server(
        &self,
        request: Request<RegisterServerRequest>,
    ) -> Result<Response<RegisterServerResponse>, Status> {
        let req = request.into_inner();
        let server = Server::new(req.id.clone(), req.address.clone());
        if self.registry.register(&server) {
            tracing::info!(id = %req.id, addr = %req.address, "server registered via gRPC");
            Ok(Response::new(RegisterServerResponse {
                success: true,
                error: String::new(),
            }))
        } else {
            Ok(Response::new(RegisterServerResponse {
                success: false,
                error: format!("server with ID '{}' already exists", req.id),
            }))
        }
    }

    async fn deregister_server(
        &self,
        request: Request<DeregisterServerRequest>,
    ) -> Result<Response<DeregisterServerResponse>, Status> {
        let req = request.into_inner();
        if self.registry.deregister(&req.id).is_some() {
            tracing::info!(id = %req.id, "server deregistered via gRPC");
            Ok(Response::new(DeregisterServerResponse {
                success: true,
                error: String::new(),
            }))
        } else {
            Ok(Response::new(DeregisterServerResponse {
                success: false,
                error: format!("server with ID '{}' not found", req.id),
            }))
        }
    }

    async fn list_servers(
        &self,
        _request: Request<ListServersRequest>,
    ) -> Result<Response<ListServersResponse>, Status> {
        let servers = self
            .registry
            .list()
            .into_iter()
            .map(|s| proto::Server {
                id: s.id,
                address: s.addr,
            })
            .collect();
        let try_order = self.registry.try_order();
        Ok(Response::new(ListServersResponse { servers, try_order }))
    }

    async fn set_try_order(
        &self,
        request: Request<SetTryOrderRequest>,
    ) -> Result<Response<SetTryOrderResponse>, Status> {
        let req = request.into_inner();
        tracing::info!(order = ?req.ids, "try order updated via gRPC");
        self.registry.set_try_order(req.ids);
        Ok(Response::new(SetTryOrderResponse { success: true }))
    }
}