Add concurrency.

This commit is contained in:
Greg 2020-04-02 14:06:11 -04:00
parent 341054857e
commit 8b70410d58
1 changed files with 32 additions and 20 deletions

18
main.go
View File

@ -6,23 +6,31 @@ import (
"fmt" "fmt"
"io" "io"
"os" "os"
"sync"
"github.com/zeebo/blake3" "github.com/zeebo/blake3"
) )
var ( var (
help = flag.Bool("h", false, "Prints help information") help = flag.Bool("h", false, "Prints help information")
length = flag.Int("l", 32, "The number of output bytes, prior to hex encoding (default 32)") length = flag.Int("l", 32, "The number of output bytes, prior to hex encoding")
par = flag.Int("p", 4, "The maximum concurrency")
) )
func main() { func main() {
flag.Parse() flag.Parse()
if *help || len(flag.Args()) == 0 { if *help || len(flag.Args()) == 0 || *par < 1 {
flag.Usage() flag.Usage()
os.Exit(0) os.Exit(0)
} }
buf := [65536]byte{} q := make(chan struct{}, *par)
wg := &sync.WaitGroup{}
for _, name := range flag.Args() { for _, name := range flag.Args() {
q <- struct{}{}
wg.Add(1)
go func(name string) {
buf := [65536]byte{}
file, err := os.Open(name) file, err := os.Open(name)
if err != nil { if err != nil {
panic(err) panic(err)
@ -42,5 +50,9 @@ func main() {
fmt.Print(hex.EncodeToString(hash)) fmt.Print(hex.EncodeToString(hash))
fmt.Print(" ") fmt.Print(" ")
fmt.Println(name) fmt.Println(name)
<- q
wg.Done()
}(name)
} }
wg.Wait()
} }