From f183e711088ee723246aa30fe169059bdd91b726 Mon Sep 17 00:00:00 2001
From: Hideo Hattori <hhatto.jp@gmail.com>
Date: Sat, 27 May 2023 11:53:03 +0900
Subject: [PATCH 1/2] add jobs option, execute N goroutine only with -w option

---
 main.go | 87 ++++++++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 71 insertions(+), 16 deletions(-)

diff --git a/main.go b/main.go
index a20446c..35a7dd7 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"bytes"
+	"context"
 	"errors"
 	"fmt"
 	"io/ioutil"
@@ -9,6 +10,7 @@ import (
 	"path/filepath"
 	"runtime/pprof"
 	"strings"
+	"sync"
 
 	log "github.com/sirupsen/logrus"
 	prefixed "github.com/x-cray/logrus-prefixed-formatter"
@@ -70,6 +72,9 @@ var (
 	writeOutput = kingpin.Flag(
 		"write-output",
 		"Write output to source instead of stdout").Short('w').Default("false").Bool()
+	jobs = kingpin.Flag(
+		"jobs",
+		"execute N jobs with -w option").Short('j').Default("1").Int()
 
 	// Args
 	paths = kingpin.Arg(
@@ -112,6 +117,36 @@ func main() {
 	}
 }
 
+type WorkerProcessRequest struct {
+	path string
+}
+
+func startWorker(ctx context.Context, shortener *Shortener, wg *sync.WaitGroup, num int) (requestch chan *WorkerProcessRequest) {
+	requestch = make(chan *WorkerProcessRequest)
+
+	for i := 0; i < num; i++ {
+		go func() {
+			for {
+				select {
+				case req := <-requestch:
+					contents, result, err := processFile(shortener, req.path)
+					if err != nil {
+						log.Fatal(err)
+					}
+					err = handleOutput(req.path, contents, result)
+					if err != nil {
+						log.Fatal(err)
+					}
+					wg.Done()
+				case <-ctx.Done():
+					return
+				}
+			}
+		}()
+	}
+
+	return
+}
 func run() error {
 	config := ShortenerConfig{
 		MaxLen:           *maxLen,
@@ -125,6 +160,11 @@ func run() error {
 		ChainSplitDots:   *chainSplitDots,
 	}
 	shortener := NewShortener(config)
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	wg := &sync.WaitGroup{}
+	worker := startWorker(ctx, shortener, wg, *jobs)
 
 	if len(*paths) == 0 {
 		// Read input from stdin
@@ -166,14 +206,21 @@ func run() error {
 						}
 
 						if !subInfo.IsDir() && strings.HasSuffix(subPath, ".go") {
-							// Shorten file and generate output
-							contents, result, err := processFile(shortener, subPath)
-							if err != nil {
-								return err
-							}
-							err = handleOutput(subPath, contents, result)
-							if err != nil {
-								return err
+							if *writeOutput && *jobs >= 2 {
+								wg.Add(1)
+								worker <- &WorkerProcessRequest{
+									path: subPath,
+								}
+							} else {
+								// Shorten file and generate output
+								contents, result, err := processFile(shortener, subPath)
+								if err != nil {
+									return err
+								}
+								err = handleOutput(subPath, contents, result)
+								if err != nil {
+									return err
+								}
 							}
 						}
 
@@ -184,18 +231,26 @@ func run() error {
 					return err
 				}
 			default:
-				// Path is a file
-				contents, result, err := processFile(shortener, path)
-				if err != nil {
-					return err
-				}
-				err = handleOutput(path, contents, result)
-				if err != nil {
-					return err
+				if *writeOutput && *jobs >= 2 {
+					wg.Add(1)
+					worker <- &WorkerProcessRequest{
+						path: path,
+					}
+				} else {
+					// Path is a file
+					contents, result, err := processFile(shortener, path)
+					if err != nil {
+						return err
+					}
+					err = handleOutput(path, contents, result)
+					if err != nil {
+						return err
+					}
 				}
 			}
 		}
 	}
+	wg.Wait()
 
 	return nil
 }

From f2dcbd70d12f45f38866c2d5b9b8f8e992c151c3 Mon Sep 17 00:00:00 2001
From: Hideo Hattori <hhatto.jp@gmail.com>
Date: Thu, 1 Jun 2023 18:37:28 +0900
Subject: [PATCH 2/2] jobs option with listFiles

---
 main.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 35a7dd7..b5fb1c2 100644
--- a/main.go
+++ b/main.go
@@ -206,7 +206,7 @@ func run() error {
 						}
 
 						if !subInfo.IsDir() && strings.HasSuffix(subPath, ".go") {
-							if *writeOutput && *jobs >= 2 {
+							if (*writeOutput || *listFiles) && *jobs >= 2 {
 								wg.Add(1)
 								worker <- &WorkerProcessRequest{
 									path: subPath,
@@ -231,7 +231,7 @@ func run() error {
 					return err
 				}
 			default:
-				if *writeOutput && *jobs >= 2 {
+				if (*writeOutput || *listFiles) && *jobs >= 2 {
 					wg.Add(1)
 					worker <- &WorkerProcessRequest{
 						path: path,