Function countLinesInFilesParallel

vcs/git/git_stats.go:505–558

(paths []string)

}

func countLinesInFilesParallel(paths []string) map[string]int {
	if len(paths) == 0 {
		return make(map[string]int)
	}

	workerCount := runtime.GOMAXPROCS(0)
	if workerCount < 1 {
		workerCount = 1
	}
	if workerCount > len(paths) {
		workerCount = len(paths)
	}

	type lineCountResult struct {
		path  string
		lines int
	}

	jobs := make(chan string)
	results := make(chan lineCountResult, len(paths))
	var wg sync.WaitGroup

	for range workerCount {
		wg.Add(1)
		go func() {
			defer wg.Done()
			for path := range jobs {
				lines, err := countLinesInFile(path)
				if err != nil {
					continue
				}
				results <- lineCountResult{
					path:  path,
					lines: lines,
				}
			}
		}()
	}

	for _, path := range paths {
		jobs <- path
	}
	close(jobs)

	wg.Wait()
	close(results)

	lineCounts := make(map[string]int, len(paths))
	for result := range results {
		lineCounts[result.path] = result.lines
	}

	return lineCounts
}