95 lines
1.9 KiB
Go
95 lines
1.9 KiB
Go
package core
|
|
|
|
import (
|
|
"log"
|
|
"fmt"
|
|
)
|
|
|
|
type Engine struct {
|
|
Input chan Result
|
|
handlers []*GenericHandler
|
|
checks []*GenericCheck
|
|
killswitch chan bool
|
|
killresponse chan bool
|
|
}
|
|
|
|
func NewEngine() *Engine {
|
|
engine := &Engine{
|
|
make(chan Result),
|
|
make([]*GenericHandler, 0),
|
|
make([]*GenericCheck, 0),
|
|
make(chan bool, 0),
|
|
make(chan bool, 0),
|
|
}
|
|
|
|
go engine.startProcessor()
|
|
|
|
return engine
|
|
}
|
|
|
|
func (this *Engine) AddCheck(check *GenericCheck) {
|
|
if check == nil {
|
|
return
|
|
}
|
|
|
|
this.checks = append(this.checks, check)
|
|
log.Print(fmt.Sprintf("%v", this.checks))
|
|
}
|
|
|
|
func (this *Engine) AddHandler(handler *GenericHandler) {
|
|
if handler == nil {
|
|
return
|
|
}
|
|
|
|
this.handlers = append(this.handlers, handler)
|
|
log.Print(fmt.Sprintf("%v", this.handlers))
|
|
}
|
|
|
|
func (this *Engine) Stop() bool {
|
|
this.killswitch<- true
|
|
return <-this.killresponse
|
|
}
|
|
|
|
func (this *Engine) startProcessor() {
|
|
statuses := make(map[string]Status)
|
|
for true {
|
|
select {
|
|
case result := <-this.Input:
|
|
//No transition if we don't exist
|
|
if result.Status == statuses[result.RecordValue] {
|
|
continue
|
|
}
|
|
|
|
//Create a record with to, from
|
|
change := Transition{
|
|
result.Status,
|
|
statuses[result.RecordValue],
|
|
result.RecordValue,
|
|
}
|
|
|
|
//Send the record to everyone who cares
|
|
for _, relay := range this.handlers {
|
|
relay.Channel<- change
|
|
}
|
|
|
|
//And set our new status
|
|
statuses[result.RecordValue] = result.Status
|
|
case <-this.killswitch:
|
|
this.killresponse<- this.stop()
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (this *Engine) stop() bool {
|
|
exitStatus := true
|
|
for _, handler := range this.handlers {
|
|
exitStatus = exitStatus && handler.Stop()
|
|
}
|
|
for _, check := range this.checks {
|
|
exitStatus = exitStatus && check.Stop()
|
|
}
|
|
return exitStatus
|
|
}
|
|
|