mirror of
https://github.com/fluencelabs/rust-libp2p
synced 2025-06-29 09:41:34 +00:00
Clean up directory structure (#426)
* Remove unused circular-buffer crate * Move transports into subdirectory * Move misc into subdirectory * Move stores into subdirectory * Move multiplexers * Move protocols * Move libp2p top layer * Fix Test: skip doctest if secio isn't enabled
This commit is contained in:
committed by
GitHub
parent
f5ce93c730
commit
2ea49718f3
86
src/simple.rs
Normal file
86
src/simple.rs
Normal file
@ -0,0 +1,86 @@
|
||||
// Copyright 2017 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 bytes::Bytes;
|
||||
use core::upgrade::{ConnectionUpgrade, Endpoint};
|
||||
use futures::prelude::*;
|
||||
use std::{iter, io::Error as IoError, sync::Arc};
|
||||
use tokio_io::{AsyncRead, AsyncWrite};
|
||||
|
||||
/// Implementation of `ConnectionUpgrade`. Convenient to use with small protocols.
|
||||
#[derive(Debug)]
|
||||
pub struct SimpleProtocol<F> {
|
||||
name: Bytes,
|
||||
// Note: we put the closure `F` in an `Arc` because Rust closures aren't automatically clonable
|
||||
// yet.
|
||||
upgrade: Arc<F>,
|
||||
}
|
||||
|
||||
impl<F> SimpleProtocol<F> {
|
||||
/// Builds a `SimpleProtocol`.
|
||||
#[inline]
|
||||
pub fn new<N>(name: N, upgrade: F) -> SimpleProtocol<F>
|
||||
where
|
||||
N: Into<Bytes>,
|
||||
{
|
||||
SimpleProtocol {
|
||||
name: name.into(),
|
||||
upgrade: Arc::new(upgrade),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<F> Clone for SimpleProtocol<F> {
|
||||
#[inline]
|
||||
fn clone(&self) -> Self {
|
||||
SimpleProtocol {
|
||||
name: self.name.clone(),
|
||||
upgrade: self.upgrade.clone(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<C, F, O, Maf> ConnectionUpgrade<C, Maf> for SimpleProtocol<F>
|
||||
where
|
||||
C: AsyncRead + AsyncWrite,
|
||||
F: Fn(C) -> O,
|
||||
O: IntoFuture<Error = IoError>,
|
||||
O::Future: 'static,
|
||||
Maf: 'static,
|
||||
{
|
||||
type NamesIter = iter::Once<(Bytes, ())>;
|
||||
type UpgradeIdentifier = ();
|
||||
|
||||
#[inline]
|
||||
fn protocol_names(&self) -> Self::NamesIter {
|
||||
iter::once((self.name.clone(), ()))
|
||||
}
|
||||
|
||||
type Output = O::Item;
|
||||
type MultiaddrFuture = Maf;
|
||||
type Future = Box<Future<Item = (O::Item, Self::MultiaddrFuture), Error = IoError>>;
|
||||
|
||||
#[inline]
|
||||
fn upgrade(self, socket: C, _: (), _: Endpoint, client_addr: Maf) -> Self::Future {
|
||||
let upgrade = &self.upgrade;
|
||||
let fut = upgrade(socket).into_future().from_err().map(move |out| (out, client_addr));
|
||||
Box::new(fut) as Box<_>
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user