manager.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

package main

import (
	"fmt"
	"sync"

	"github.com/Shugyousha/stasher/filter"
	"github.com/Shugyousha/stasher/input"
	"github.com/Shugyousha/stasher/output"
	"github.com/Shugyousha/stasher/work"
)

type Manager struct {
	Input  input.Input
	Filter filter.Filter
	Output output.Output
}

func (m *Manager) Run() {
	var wg sync.WaitGroup

	ic := m.Input.Start()
	for w := range ic {
		wg.Add(1)

		go func(w *work.Work) {
			nw := m.Filter.Filter(w)
			err := nw.Error()
			if err != nil {
				fmt.Printf("Got an error when processing Work: %q\n", err)
			}
			m.Output.Output(nw)

			wg.Done()
		}(w)
	}

	wg.Wait()
}