ic_web3_rs/transports/
either.rs

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
//! A strongly-typed transport alternative.

use crate::{api, error, rpc, BatchTransport, DuplexTransport, RequestId, Transport};
use futures::{
    future::{BoxFuture, FutureExt},
    stream::{BoxStream, StreamExt},
};
use ic_cdk::api::management_canister::http_request::TransformContext;

use super::ic_http_client::CallOptions;

/// A wrapper over two possible transports.
///
/// This type can be used to write semi-generic
/// code without the hassle of making all functions generic.
///
/// See the `examples` folder for an example how to use it.
#[derive(Debug, Clone)]
pub enum Either<A, B> {
    /// First possible transport.
    Left(A),
    /// Second possible transport.
    Right(B),
}

impl<A, B, AOut, BOut> Transport for Either<A, B>
where
    A: Transport<Out = AOut>,
    B: Transport<Out = BOut>,
    AOut: futures::Future<Output = error::Result<rpc::Value>> + 'static + Send,
    BOut: futures::Future<Output = error::Result<rpc::Value>> + 'static + Send,
{
    type Out = BoxFuture<'static, error::Result<rpc::Value>>;

    fn prepare(&self, method: &str, params: Vec<rpc::Value>) -> (RequestId, rpc::Call) {
        match *self {
            Self::Left(ref a) => a.prepare(method, params),
            Self::Right(ref b) => b.prepare(method, params),
        }
    }

    fn send(&self, id: RequestId, request: rpc::Call, options: CallOptions) -> Self::Out {
        match *self {
            Self::Left(ref a) => a.send(id, request, options).boxed(),
            Self::Right(ref b) => b.send(id, request, options).boxed(),
        }
    }

    fn set_max_response_bytes(&mut self, v: u64) {
        match *self {
            Self::Left(ref mut a) => a.set_max_response_bytes(v),
            Self::Right(ref mut b) => b.set_max_response_bytes(v),
        }
    }
}

impl<A, B, ABatch, BBatch> BatchTransport for Either<A, B>
where
    A: BatchTransport<Batch = ABatch>,
    B: BatchTransport<Batch = BBatch>,
    A::Out: 'static + Send,
    B::Out: 'static + Send,
    ABatch: futures::Future<Output = error::Result<Vec<error::Result<rpc::Value>>>> + 'static + Send,
    BBatch: futures::Future<Output = error::Result<Vec<error::Result<rpc::Value>>>> + 'static + Send,
{
    type Batch = BoxFuture<'static, error::Result<Vec<error::Result<rpc::Value>>>>;

    fn send_batch<T>(&self, requests: T) -> Self::Batch
    where
        T: IntoIterator<Item = (RequestId, rpc::Call)>,
    {
        match *self {
            Self::Left(ref a) => a.send_batch(requests).boxed(),
            Self::Right(ref b) => b.send_batch(requests).boxed(),
        }
    }
}

impl<A, B, AStream, BStream> DuplexTransport for Either<A, B>
where
    A: DuplexTransport<NotificationStream = AStream>,
    B: DuplexTransport<NotificationStream = BStream>,
    A::Out: 'static + Send,
    B::Out: 'static + Send,
    AStream: futures::Stream<Item = rpc::Value> + 'static + Send,
    BStream: futures::Stream<Item = rpc::Value> + 'static + Send,
{
    type NotificationStream = BoxStream<'static, rpc::Value>;

    fn subscribe(&self, id: api::SubscriptionId) -> error::Result<Self::NotificationStream> {
        Ok(match *self {
            Self::Left(ref a) => a.subscribe(id)?.boxed(),
            Self::Right(ref b) => b.subscribe(id)?.boxed(),
        })
    }

    fn unsubscribe(&self, id: api::SubscriptionId) -> error::Result {
        match *self {
            Self::Left(ref a) => a.unsubscribe(id),
            Self::Right(ref b) => b.unsubscribe(id),
        }
    }
}