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


use futures::future::Future;
use hyper::server::Request;
use chashmap::CHashMap;
//use ::Shared;
use ::plugins::{Plugin, PluginData};

#[derive(Default)]
pub struct PageVisits {
    counts: CHashMap<String, u32>
}

impl PageVisits {
    pub fn new() -> PageVisits
    {
        PageVisits {
            counts: CHashMap::new()
        }
    }

    pub fn get(&self, url: &str) -> Option<u32>
    {
        let url = url.to_owned();
        self.counts.get(&url).map(|guard| *guard)
    }
}

impl<S: Send + Sync + 'static, E: 'static> Plugin<S,E> for PageVisits {
    fn before_handler(&self, request: &mut Request) {
        let url: String = request.uri().as_ref().to_owned();
        if let Some(mut count) = self.counts.get_mut(&url) {
            *count += 1;
        } else {
            self.counts.insert(url, 1);
        }
    }

    fn after_handler(&self,
                     future: Box<Future<Item = PluginData<S>, Error = E>>)
                     -> Box<Future<Item = PluginData<S>, Error = E>>
    {
        // We do nothing here
        future
    }
}