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
80

extern crate config;
extern crate chrono;
extern crate riker;
#[macro_use]
extern crate runtime_fmt;

use std::marker::PhantomData;

use config::Config;

use riker::actors::*;
use riker::system::LoggerProps;

pub struct SimpleLogger<Msg: Message> {
    msg: PhantomData<Msg>,
    cfg: LoggerConfig,
}

impl<Msg> Actor for SimpleLogger<Msg>
    where Msg: Message
{
    type Msg = Msg;

    fn receive(&mut self, _: &Context<Self::Msg>, _: Self::Msg, _: Option<ActorRef<Self::Msg>>) {
    }

    fn system_receive(&mut self, _: &Context<Self::Msg>, msg: SystemMsg<Self::Msg>, _: Option<ActorRef<Self::Msg>>) {
        if let SystemMsg::Log(entry) = msg {
            let now = chrono::Utc::now();

            rt_println!(self.cfg.log_fmt,
                        date = now.format(&self.cfg.date_fmt),
                        time = now.format(&self.cfg.time_fmt),
                        level = entry.level,
                        module = entry.module.unwrap_or_default(),
                        body = entry.body
                        ).unwrap();
        }
    }
}

impl<Msg> LoggerProps for SimpleLogger<Msg>
    where Msg: Message
{
    type Msg = Msg;

    fn props(config: &Config) -> BoxActorProd<Self::Msg> {
        Props::new_args(Box::new(SimpleLogger::actor), LoggerConfig::from(config))
    }
}

impl<Msg> SimpleLogger<Msg>
    where Msg: Message
{
    fn actor(cfg: LoggerConfig) -> BoxActor<Msg> {
        let actor = SimpleLogger {
            msg: PhantomData,
            cfg
        };
        Box::new(actor)
    }
}

#[derive(Clone)]
struct LoggerConfig {
    time_fmt: String,
    date_fmt: String,
    log_fmt: String,
}

impl<'a> From<&'a Config> for LoggerConfig {
    fn from(config: &Config) -> Self {
        LoggerConfig {
            time_fmt: config.get_str("log.time_format").unwrap().to_string(),
            date_fmt: config.get_str("log.date_format").unwrap().to_string(),
            log_fmt: config.get_str("log.log_format").unwrap().to_string()
        }
    }
}