mirror of
https://github.com/fluencelabs/rust-libp2p
synced 2025-05-29 02:31:20 +00:00
fadcc43e69
This type can be replaced with std-lib components. Related: https://github.com/libp2p/rust-libp2p/pull/3746. Pull-Request: #3842.
106 lines
3.7 KiB
Rust
106 lines
3.7 KiB
Rust
// Copyright 2018 Parity Technologies (UK) Ltd.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
|
// copy of this software and associated documentation files (the "Software"),
|
|
// to deal in the Software without restriction, including without limitation
|
|
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
// and/or sell copies of the Software, and to permit persons to whom the
|
|
// Software is furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
// DEALINGS IN THE SOFTWARE.
|
|
|
|
use crate::either::EitherFuture;
|
|
use crate::{
|
|
either::EitherName,
|
|
upgrade::{InboundUpgrade, OutboundUpgrade, UpgradeInfo},
|
|
};
|
|
use either::Either;
|
|
use futures::future;
|
|
use std::iter::{Chain, Map};
|
|
|
|
/// Upgrade that combines two upgrades into one. Supports all the protocols supported by either
|
|
/// sub-upgrade.
|
|
///
|
|
/// The protocols supported by the first element have a higher priority.
|
|
#[derive(Debug, Clone)]
|
|
pub struct SelectUpgrade<A, B>(A, B);
|
|
|
|
impl<A, B> SelectUpgrade<A, B> {
|
|
/// Combines two upgrades into an `SelectUpgrade`.
|
|
///
|
|
/// The protocols supported by the first element have a higher priority.
|
|
pub fn new(a: A, b: B) -> Self {
|
|
SelectUpgrade(a, b)
|
|
}
|
|
}
|
|
|
|
impl<A, B> UpgradeInfo for SelectUpgrade<A, B>
|
|
where
|
|
A: UpgradeInfo,
|
|
B: UpgradeInfo,
|
|
{
|
|
type Info = EitherName<A::Info, B::Info>;
|
|
type InfoIter = Chain<
|
|
Map<<A::InfoIter as IntoIterator>::IntoIter, fn(A::Info) -> Self::Info>,
|
|
Map<<B::InfoIter as IntoIterator>::IntoIter, fn(B::Info) -> Self::Info>,
|
|
>;
|
|
|
|
fn protocol_info(&self) -> Self::InfoIter {
|
|
let a = self
|
|
.0
|
|
.protocol_info()
|
|
.into_iter()
|
|
.map(EitherName::A as fn(A::Info) -> _);
|
|
let b = self
|
|
.1
|
|
.protocol_info()
|
|
.into_iter()
|
|
.map(EitherName::B as fn(B::Info) -> _);
|
|
|
|
a.chain(b)
|
|
}
|
|
}
|
|
|
|
impl<C, A, B, TA, TB, EA, EB> InboundUpgrade<C> for SelectUpgrade<A, B>
|
|
where
|
|
A: InboundUpgrade<C, Output = TA, Error = EA>,
|
|
B: InboundUpgrade<C, Output = TB, Error = EB>,
|
|
{
|
|
type Output = future::Either<TA, TB>;
|
|
type Error = Either<EA, EB>;
|
|
type Future = EitherFuture<A::Future, B::Future>;
|
|
|
|
fn upgrade_inbound(self, sock: C, info: Self::Info) -> Self::Future {
|
|
match info {
|
|
EitherName::A(info) => EitherFuture::First(self.0.upgrade_inbound(sock, info)),
|
|
EitherName::B(info) => EitherFuture::Second(self.1.upgrade_inbound(sock, info)),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<C, A, B, TA, TB, EA, EB> OutboundUpgrade<C> for SelectUpgrade<A, B>
|
|
where
|
|
A: OutboundUpgrade<C, Output = TA, Error = EA>,
|
|
B: OutboundUpgrade<C, Output = TB, Error = EB>,
|
|
{
|
|
type Output = future::Either<TA, TB>;
|
|
type Error = Either<EA, EB>;
|
|
type Future = EitherFuture<A::Future, B::Future>;
|
|
|
|
fn upgrade_outbound(self, sock: C, info: Self::Info) -> Self::Future {
|
|
match info {
|
|
EitherName::A(info) => EitherFuture::First(self.0.upgrade_outbound(sock, info)),
|
|
EitherName::B(info) => EitherFuture::Second(self.1.upgrade_outbound(sock, info)),
|
|
}
|
|
}
|
|
}
|