]> Sergey Matveev's repositories - btrtrc.git/commitdiff
internal/cmd/bloom-false-positives: Switch to tagflag
authorMatt Joiner <anacrolix@gmail.com>
Mon, 9 Nov 2015 12:05:52 +0000 (23:05 +1100)
committerMatt Joiner <anacrolix@gmail.com>
Mon, 9 Nov 2015 12:05:52 +0000 (23:05 +1100)
Was looking at using github.com/tylertreat/BoomFilters instead of github.com/willf/bloom

internal/cmd/bloom-false-positives/main.go

index 0b94480a39bdfc59df025dccb735bf34bd3b3940..456e2106680a35704aca72afc4548b26db676d61 100644 (file)
@@ -2,18 +2,20 @@ package main
 
 import (
        "bufio"
-       "flag"
        "fmt"
        "os"
 
+       "github.com/anacrolix/tagflag"
        "github.com/willf/bloom"
 )
 
 func main() {
-       m := flag.Uint("m", 0, "")
-       k := flag.Uint("k", 0, "")
-       flag.Parse()
-       filter := bloom.New(*m, *k)
+       var args struct {
+               M uint `help:"num bits"`
+               K uint `help:"num hashing functions"`
+       }
+       tagflag.Parse(&args, tagflag.Description("adds lines from stdin to a bloom filter with the given configuration, and gives collision stats at EOF"))
+       filter := bloom.New(args.M, args.K)
        scanner := bufio.NewScanner(os.Stdin)
        n := 0
        collisions := 0