translator_sv2 0.2.3

SV1 to SV2 translation proxy
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

use async_channel::{Receiver, Sender};
use stratum_apps::{stratum_core::sv1_api::json_rpc, utils::types::DownstreamId};
use tokio_util::sync::CancellationToken;
use tracing::debug;

#[derive(Clone, Debug)]
pub struct DownstreamChannelState {
    pub downstream_sv1_sender: Sender<json_rpc::Message>,
    pub downstream_sv1_receiver: Receiver<json_rpc::Message>,
    pub sv1_server_sender: Sender<(DownstreamId, json_rpc::Message)>,
    pub sv1_server_receiver: Receiver<json_rpc::Message>,
    /// Per-connection cancellation token (child of the global token).
    /// Cancelled when this downstream's task loop exits, causing
    /// the associated SV1 I/O task to shut down.
    pub connection_token: CancellationToken,
}

#[cfg_attr(not(test), hotpath::measure_all)]
impl DownstreamChannelState {
    pub fn new(
        downstream_sv1_sender: Sender<json_rpc::Message>,
        downstream_sv1_receiver: Receiver<json_rpc::Message>,
        sv1_server_sender: Sender<(DownstreamId, json_rpc::Message)>,
        sv1_server_receiver: Receiver<json_rpc::Message>,
        connection_token: CancellationToken,
    ) -> Self {
        Self {
            downstream_sv1_receiver,
            downstream_sv1_sender,
            sv1_server_receiver,
            sv1_server_sender,
            connection_token,
        }
    }

    pub fn drop(&self) {
        debug!("Dropping downstream channel state");
        self.connection_token.cancel();
        self.downstream_sv1_receiver.close();
        self.downstream_sv1_sender.close();
    }
}