mai 0.1.0

A higher-level mio interface.
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

use std::collections::BTreeMap;

use slab::Index;
use mio::Token;

use Protocol;
use ::token_bucket::TokenBucket;
use stream_session::StreamSession;

pub struct StreamManager<P: ?Sized> where P: Protocol {
  streams: BTreeMap<usize, StreamSession<P>>,
  token_bucket: TokenBucket,
}

impl <P: ?Sized> StreamManager <P> where P: Protocol {
  pub fn new() -> StreamManager<P> {
    StreamManager {
      streams: BTreeMap::new(),
      token_bucket: TokenBucket::new(),
    }
  }

  pub fn get_mut(&mut self, token: Token) -> Option<&mut StreamSession<P>> {
    self.streams.get_mut(&token.as_usize())
  }

  fn get_next_token(&mut self) -> Token {
    if let Some(token) = self.token_bucket.get() {
      return token;
    }
    Index::from_usize(self.streams.len())
  }

  pub fn insert(&mut self, stream_session: StreamSession<P>) -> Token {
    let next_token = self.get_next_token();
    self.streams.insert(next_token.as_usize(), stream_session);
    next_token
  }

  pub fn remove(&mut self, token: Token) -> Option<StreamSession<P>> {
    self.token_bucket.put(token);
    self.streams.remove(&token.as_usize())
  }
}