/
connector_controller.go
84 lines (75 loc) · 2.21 KB
/
connector_controller.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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package controllers
import (
"log"
"time"
"github.com/att-comdev/jarvis-connector/services"
"github.com/att-comdev/jarvis-connector/types"
)
var (
Connector connectorController = &ConnectorControllerImpl{
pendingCheck: make(chan *types.PendingChecksInfo, 5),
pendingSubmit: make(chan *types.PendingSubmitInfo, 5),
}
)
type connectorController interface {
ServeCheck()
ServeSubmit()
PendingLoop()
}
type ConnectorControllerImpl struct {
pendingCheck chan *types.PendingChecksInfo
pendingSubmit chan *types.PendingSubmitInfo
}
// ServeCheck runs the serve loop, dispatching for checks that need it.
func (controller *ConnectorControllerImpl) ServeCheck() {
for p := range controller.pendingCheck {
// TODO: parallelism?.
if err := services.GerritChecker.ExecuteCheck(p); err != nil {
log.Printf("ExecuteCheck(%v): %v", p, err)
}
}
}
// ServeSubmit runs the serve loop, dispatching for submissions that need it.
func (controller *ConnectorControllerImpl) ServeSubmit() {
for p := range controller.pendingSubmit {
// TODO: parallelism?.
if err := services.GerritSubmitter.ExecuteSubmit(p); err != nil {
log.Printf("ExecuteSubmit(%v): %v", p, err)
}
}
}
// pendingLoop periodically contacts gerrit to find new checks and submissions to
// execute. It should be executed in a goroutine.
func (controller *ConnectorControllerImpl) PendingLoop() {
for {
// TODO: real rate limiting.
time.Sleep(10 * time.Second)
pendingChecks, err := services.GerritChecker.PendingChecksByScheme(checkerScheme)
if err == nil {
log.Printf("Received %d Pending Checks", len(pendingChecks))
for _, pc := range pendingChecks {
select {
case controller.pendingCheck <- pc:
default:
log.Println("too busy; dropping check.")
}
}
} else {
log.Printf("PendingChecksByScheme: %v", err)
}
// Handle Submissions
pendingSubmissions, err := services.GerritSubmitter.PendingSubmit()
if err == nil {
log.Printf("Received %d Pending Submissions", len(pendingSubmissions))
for _, ps := range pendingSubmissions {
select {
case controller.pendingSubmit <- ps:
default:
log.Println("too busy; dropping submission.")
}
}
} else {
log.Printf("PendingSubmit: %v", err)
}
}
}