blip 0.3.0

A crate for writing fast and highly resilient in-process gRPC service meshes.
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

// Copyright 2020 nytopop (Eric Izoita)
//
// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.
//! Shared code referred to by multiple test modules.
#![allow(unused_attributes)]
#![type_length_limit = "8388608"]

use blip::{MeshService, MultiNodeCut, Subscription};
use simplelog::{Config, LevelFilter, TestLogger};
use std::{
    net::SocketAddr,
    sync::atomic::{AtomicU32, Ordering::Relaxed},
};
use tokio::sync::mpsc;

// A quick NOTE about addressing in integration tests: each test should subnet a unique /20
// from the 127.128/9 loopback block to avoid collisions with other tests or os services.
//
// The port should always be 10000.

pub fn subnet() -> u32 {
    static SUBNET: AtomicU32 = AtomicU32::new(0);
    let s = SUBNET.fetch_add(1, Relaxed);
    assert!(s <= 2 ^ 11);
    s
}

pub fn addr_in(subnet: u32, host: u32) -> SocketAddr {
    let mut addr = (subnet & 0x1fff) << 12; // 11 bits of subnet
    addr |= host & 0xfff; // 12 bits of host
    addr |= 0x7f800000; // and the loopback stuff

    (addr.to_be_bytes(), 10000).into()
}

pub fn init_logger() {
    let _ = TestLogger::init(LevelFilter::Info, Config::default());
}

pub fn cfg_handle() -> (CfgHandle, CfgService) {
    let (tx, rx) = mpsc::channel(32);

    let h = CfgHandle { rx };
    let s = CfgService { tx };

    (h, s)
}

pub struct CfgService {
    tx: mpsc::Sender<MultiNodeCut>,
}

#[blip::async_trait]
impl MeshService for CfgService {
    async fn accept(mut self: Box<Self>, mut cuts: Subscription) {
        while let Ok(cut) = cuts.recv().await {
            self.tx.send(cut).await.unwrap();
        }
    }
}

pub struct CfgHandle {
    rx: mpsc::Receiver<MultiNodeCut>,
}

impl CfgHandle {
    /// Blocks until a view-change proposal with `n` peers is accepted.
    pub async fn cfg_change(&mut self, n: usize) -> MultiNodeCut {
        while let Some(cut) = self.rx.recv().await {
            if cut.members().len() == n {
                return cut;
            }
        }
        panic!("cfg_handle sender closed!");
    }
}