allow worker pool to be restarted

Author: ybirader

main.go

@@ -74,6 +74,7 @@ func NewFileProcessPool(numberOfWorkers int, executor func(f File)) (*FileProces
 }
 
 func (f *FileProcessPool) Start() {
+	f.reset()
 	f.wg.Add(f.numberOfWorkers)
 	for i := 0; i < f.numberOfWorkers; i++ {
 		go f.listen()
@@ -101,6 +102,10 @@ func (f *FileProcessPool) Enqueue(file File) {
 	f.tasks <- file
 }
 
+func (f *FileProcessPool) reset() {
+	f.tasks = make(chan File)
+}
+
 func (a *Archiver) newWalkDir(root string) error {
 	a.filesToWrite = make(chan File)
 	a.fileProcessPool.Start()

main_test.go

@@ -158,6 +158,30 @@ func TestFileProcessPool(t *testing.T) {
 		_, err := NewFileProcessPool(0, executor)
 		assert.Error(t, err)
 	})
+
+	t.Run("can be closed and restarted", func(t *testing.T) {
+		output := bytes.Buffer{}
+		executor := func(_ File) {
+			output.WriteString("hello ")
+		}
+
+		fileProcessPool, err := NewFileProcessPool(1, executor)
+		assert.NoError(t, err)
+		fileProcessPool.Start()
+
+		info := getFileInfo(t, helloTxtFileFixture)
+		fileProcessPool.Enqueue(File{Path: helloTxtFileFixture, Info: info})
+
+		fileProcessPool.Close()
+
+		fileProcessPool.Start()
+		info = getFileInfo(t, helloTxtFileFixture)
+		fileProcessPool.Enqueue(File{Path: helloTxtFileFixture, Info: info})
+
+		fileProcessPool.Close()
+
+		assert.Equal(t, "hello hello ", output.String())
+	})
 }
 
 func BenchmarkArchive(b *testing.B) {