mheffner / rust-sketches-ddsketch Goto Github PK

use rand::prelude::*;
use rand::rngs::StdRng;
use sketches_ddsketch::{Config, DDSketch};

fn main() {
    let config = Config::new(0.0001, 32_768, 1.0e-9);
    let mut sketches = vec![
        DDSketch::new(config),
        DDSketch::new(config),
        DDSketch::new(config)];


    let mut rng: StdRng = rand::SeedableRng::seed_from_u64(42);
    let mut i = 1;
    loop {
        let samples = rng.gen_range(0..20000);
        sketches[0] = DDSketch::new(config);
        for _ in 0..samples {
            let rv: i64 = rng.gen_range(0..8000000000);
            sketches[0].add(rv as f64);
        }

        let mut merged = DDSketch::new(config);
        for sketch in &sketches {
            merged.merge(sketch).unwrap();
        }
        sketches.rotate_right(1);

        if i % 1000 == 0 {
            print!(".");
        }
        i += 1;
    }
}