summaryrefslogtreecommitdiff
path: root/types.go
blob: abd2658c65202e8614d2b0fc6ce46a2901c47eeb (plain)
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
40
41
42
43
44
45
46
47
48
package main

import (
	"fmt"
	"sync"

	"github.com/Shugyousha/stasher/work"
)

type Input interface {
	Start() chan *work.Work
}

type Filter interface {
	Filter(*work.Work) *work.Work
}

type Output interface {
	Output(*work.Work)
}

type Manager struct {
	Input  Input
	Filter Filter
	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()
}