heavykeeper-rs

Top-K Heavykeeper algorithm for Top-K elephant flows

This is based on the paper HeavyKeeper: An Accurate Algorithm for Finding Top-k Elephant Flows by Junzhi Gong, Tong Yang, Haowei Zhang, and Hao Li, Peking University; Steve Uhlig, Queen Mary, University of London; Shigang Chen, University of Florida; Lorna Uden, Staffordshire University; Xiaoming Li, Peking University

Example

See examples/basic.rs for a complete example, or examples/ip_files.rs for an example of counting top-k IP flows in a file.

Basic usage:

use heavykeeper::TopK;

// create a new TopK with k=10, width=1000, depth=4, decay=0.9
let mut topk: TopK<String> = TopK::new(10, 1000, 4, 0.9);

// add some items
topk.add("example item", 5);
topk.add("another item", 1);

// check the counts
for node in topk.list() {
    println!("{} {}", node.item, node.count);
}

Other Implementations

Name	Language	Github Repo
SegmentIO	Go	https://github.com/segmentio/topk
Aegis	Go	https://github.com/go-kratos/aegis/blob/main/topk/heavykeeper.go
Tomasz Kolaj	Go	https://github.com/migotom/heavykeeper
HeavyKeeper Paper	C++	https://github.com/papergitkeeper/heavy-keeper-project
Jigsaw-Sketch	C++	https://github.com/duyang92/jigsaw-sketch-paper/tree/main/CPU/HeavyKeeper
Redis Bloom Heavykeeper	C	https://github.com/RedisBloom/RedisBloom/blob/master/src/topk.c
Count-Min-Sketch	Rust	https://github.com/alecmocatta/streaming_algorithms
Sliding Window HeavyKeeper	Go	https://github.com/keilerkonzept/topk

Running

Word Count Example

A word count program that demonstrates the HeavyKeeper algorithm can be found at examples/word_count.rs.

Usage

cargo build --example word_count --release
target/release/examples/word_count -k 10 -w 8192 -d 2 -y 0.95 -f data/war_and_peace.txt

Running the basic example

cargo run --example basic --release

Running the IPv4 example

cargo run --example ip_files --release

Run the benchmarks

cargo bench

Benchmark the sample word count app

hyperfine 'target/release/examples/word_count -k 10 -w 8192 -d 2 -y 0.95 -f data/war_and_peace.txt'

Test Data

For information about test data format and how to obtain or generate test data, please see data/README.md.

License

This project is dual licensed under the Apache/MIT license.

0.6.2	0	0	0	0	0
0.6.1	0	0	0	0	0
0.6.0	0	0	0	0	0
0.5.1	0	0	0	0	0
0.5.0	0	0	0	0	0
0.4.0	0	0	0	0	0
0.3.1	0	0	0	0	0
0.3.0	0	0	0	0	0
0.2.7	0	0	0	0	0
0.2.6	0	0	0	0	0
0.2.5	0	0	0	0	0
0.2.4	0	0	0	0	0
0.2.3	0	0	0	0	0
0.2.2	0	0	0	0	0
0.2.1	0	0	0	0	0

Componentpedia / Listings / heavykeeper

heavykeeper