go-examples/shadirconcurrent/main.go

package main

import "fmt"
import "crypto/sha256"
import "io"
import "log"
import "os"
import "path/filepath"
import "sync"

func sha256File(path string) (string, error) {
	f, err := os.Open(path)
	if err != nil {
		return "", err
	}
	defer f.Close()
	h := sha256.New()
	io.Copy(h, f)
	return fmt.Sprintf("SHA256 (%s) =  %x", path, h.Sum(nil)), nil
}

func main() {
	if len(os.Args) < 2 {
		log.Fatalln("Usage shadir file")
	}

	var wg sync.WaitGroup
	chan_hash, chan_path := make(chan string), make(chan string)

	for i := 0; i < 100; i++ {
		wg.Add(1)
		go func() {
			for path := range chan_path {
				if h, err := sha256File(path); err != nil {
					log.Print("Error "+path, err)
				} else {
					chan_hash <- h
				}
			}
			wg.Done()
		}()
	}
	go func() {
		wg.Wait()
		close(chan_hash)
	}()

	go func() {
		filepath.Walk(os.Args[1], func(path string, info os.FileInfo, err error) error {
			if err == nil && !info.IsDir() {
				chan_path <- path
			}
			return err
		})
		close(chan_path)
	}()

	for h := range chan_hash {
		fmt.Println(h)
	}
}