meutel/go-examples
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

SHA dir refactoring: routines runner

Browse Source
This commit is contained in:
Meutel 2017-07-08 17:32:21 +02:00
parent 5a237532fa
commit 6b2182b5d4
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
shadirconcurrent/main.go
View File

@ -19,6 +19,16 @@ func sha256File(path string) (string, error) {
return fmt.Sprintf("SHA256 (%s) = %x", path, h.Sum(nil)), nil return fmt.Sprintf("SHA256 (%s) = %x", path, h.Sum(nil)), nil
} }
func runParallel(n uint, what func()) {
var wg sync.WaitGroup
for i := 0; i < int(n); i++ {
wg.Add(1)
what()
wg.Done()
}
wg.Wait()
}
func main() { func main() {
if len(os.Args) < 2 { if len(os.Args) < 2 {
log.Fatalln("Usage shadir file") log.Fatalln("Usage shadir file")
@ -26,10 +36,8 @@ func main() {
chan_hash, chan_path := make(chan string), make(chan string) chan_hash, chan_path := make(chan string), make(chan string)
go func() { // 100 routines hashing go func() {
var wg sync.WaitGroup runParallel(100, func() { // 100 routines hashing
for i := 0; i < 100; i++ {
wg.Add(1)
for path := range chan_path { for path := range chan_path {
if h, err := sha256File(path); err != nil { if h, err := sha256File(path); err != nil {
log.Print("Error "+path, err) log.Print("Error "+path, err)
@ -37,9 +45,7 @@ func main() {
chan_hash <- h chan_hash <- h
} }
} }
wg.Done() })
}
wg.Wait()
close(chan_hash) close(chan_hash)
}() }()