File: bench.rs

package info (click to toggle)
tiny 0.13.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 1,780 kB
  • sloc: makefile: 29
file content (58 lines) | stat: -rw-r--r-- 1,637 bytes parent folder | download
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
// A program that initializes the TUI and adds lines in the file (given as first argument) to it,
// with a draw() call after every line added.
//
// After adding all lines the program just quits.
//
// Useful for benchmarking TUI::draw().

use std::fs::File;
use std::io::{BufRead, BufReader};
use std::path::PathBuf;

use libtiny_common::MsgTarget;
use libtiny_tui::TUI;

fn main() {
    run_bench();

    let mut rusage: libc::rusage = unsafe { ::std::mem::zeroed() };
    match unsafe { libc::getrusage(libc::RUSAGE_SELF, &mut rusage as *mut _) } {
        0 => {
            println!("Max RSS (ru_maxrss): {} kb", rusage.ru_maxrss);
        }
        i => {
            println!("getrusage() returned {}", i);
        }
    }
}

fn run_bench() {
    let args = std::env::args().collect::<Vec<_>>();
    let file_path = &args[1];
    let file = File::open(file_path).unwrap();
    let file_buffered = BufReader::new(file);
    let lines = file_buffered.lines().map(Result::unwrap).collect();

    let runtime = tokio::runtime::Builder::new_current_thread()
        .enable_all()
        .build()
        .unwrap();

    let local = tokio::task::LocalSet::new();

    local.block_on(&runtime, async move {
        let (tui, _) = TUI::run(PathBuf::from("../tiny/config.yml"));
        tui.draw();
        bench_task(tui, lines).await;
    });
}

async fn bench_task(tui: TUI, lines: Vec<String>) {
    let msg_target = MsgTarget::Server { serv: "mentions" };
    let time = time::OffsetDateTime::now_utc();

    for line in &lines {
        tui.add_privmsg("server", line, time, &msg_target, false, false);
        tui.draw();
    }
}