resources, pgraph: Refactor Worker and simplify API

I'm still working on reducing the size of the monster patches that I
land, but I'm exercising the priviledge as the initial author. In any
case, this refactors worker into two, and cleans up the passing around
of the processChan. This puts common code into Init and Close.
This commit is contained in:
James Shubin
2017-02-10 09:37:12 -05:00
parent 2da21f90f4
commit fccf508dde
16 changed files with 276 additions and 274 deletions

View File

@@ -273,7 +273,7 @@ sending out erroneous `Event` messages to keep things alive until it finishes.
#### Example #### Example
```golang ```golang
// Watch is the listener and main loop for this resource. // Watch is the listener and main loop for this resource.
func (obj *FooRes) Watch(processChan chan *event.Event) error { func (obj *FooRes) Watch() error {
// setup the Foo resource // setup the Foo resource
var err error var err error
if err, obj.foo = OpenFoo(); err != nil { if err, obj.foo = OpenFoo(); err != nil {
@@ -282,7 +282,7 @@ func (obj *FooRes) Watch(processChan chan *event.Event) error {
defer obj.whatever.CloseFoo() // shutdown our defer obj.whatever.CloseFoo() // shutdown our
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -311,7 +311,7 @@ func (obj *FooRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event() // send the event!
} }
} }
} }

View File

@@ -235,7 +235,7 @@ func (g *Graph) Process(v *Vertex) error {
// would simplify the resources by removing boilerplate // would simplify the resources by removing boilerplate
if v.Meta().Poll > 0 { if v.Meta().Poll > 0 {
if !checkOK { // something changed, restart timer if !checkOK { // something changed, restart timer
cuid := v.Res.ConvergerUID() // get the converger uid used to report status cuid, _, _ := v.Res.ConvergerUIDs() // get the converger uid used to report status
cuid.ResetTimer() // activity! cuid.ResetTimer() // activity!
if g.Flags.Debug { if g.Flags.Debug {
log.Printf("%s[%s]: Converger: ResetTimer", obj.Kind(), obj.GetName()) log.Printf("%s[%s]: Converger: ResetTimer", obj.Kind(), obj.GetName())
@@ -306,41 +306,13 @@ func (obj *SentinelErr) Error() string {
return obj.err.Error() return obj.err.Error()
} }
// Worker is the common run frontend of the vertex. It handles all of the retry // innerWorker is the CheckApply runner that reads from processChan.
// and retry delay common code, and ultimately returns the final status of this func (g *Graph) innerWorker(v *Vertex) {
// vertex execution.
func (g *Graph) Worker(v *Vertex) error {
// listen for chan events from Watch() and run
// the Process() function when they're received
// this avoids us having to pass the data into
// the Watch() function about which graph it is
// running on, which isolates things nicely...
obj := v.Res obj := v.Res
// run the init (should match 1-1 with Close function if this succeeds)
if err := obj.Init(); err != nil {
return errwrap.Wrapf(err, "could not Init() resource")
}
lock := &sync.Mutex{} // lock around processChan closing and sending
finished := false // did we close processChan ?
processChan := make(chan *event.Event)
// if the CheckApply run takes longer than the converged
// timeout, we could inappropriately converge mid-apply!
// avoid this by blocking convergence with a fake report
// we also add a similar blocker around the worker loop!
wcuid := obj.Converger().Register() // get an extra cuid for the worker!
defer wcuid.Unregister()
wcuid.SetConverged(true) // starts off false, and waits for loop timeout
pcuid := obj.Converger().Register() // get an extra cuid for the process
defer pcuid.Unregister()
pcuid.SetConverged(true) // starts off true, because it's not running...
go func() {
running := false running := false
done := make(chan struct{}) done := make(chan struct{})
playback := false // do we need to run another one? playback := false // do we need to run another one?
_, wcuid, pcuid := obj.ConvergerUIDs() // get extra cuids (worker, process)
waiting := false waiting := false
var timer = time.NewTimer(time.Duration(math.MaxInt64)) // longest duration var timer = time.NewTimer(time.Duration(math.MaxInt64)) // longest duration
@@ -353,10 +325,10 @@ func (g *Graph) Worker(v *Vertex) error {
var limiter = rate.NewLimiter(v.Meta().Limit, v.Meta().Burst) var limiter = rate.NewLimiter(v.Meta().Limit, v.Meta().Burst)
limited := false limited := false
Loop: Loop:
for { for {
select { select {
case ev, ok := <-processChan: // must use like this case ev, ok := <-obj.ProcessChan(): // must use like this
if !ok { // processChan closed, let's exit if !ok { // processChan closed, let's exit
break Loop // no event, so no ack! break Loop // no event, so no ack!
} }
@@ -463,15 +435,9 @@ func (g *Graph) Worker(v *Vertex) error {
playback = false playback = false
// this lock avoids us sending to // this lock avoids us sending to
// channel after we've closed it! // channel after we've closed it!
lock.Lock()
go func() {
if !finished {
// TODO: can this experience indefinite postponement ? // TODO: can this experience indefinite postponement ?
// see: https://github.com/golang/go/issues/11506 // see: https://github.com/golang/go/issues/11506
obj.Event(processChan) // replay a new event go obj.Event() // replay a new event
}
lock.Unlock()
}()
} }
running = false running = false
pcuid.SetConverged(true) // "unblock" Process pcuid.SetConverged(true) // "unblock" Process
@@ -481,7 +447,42 @@ func (g *Graph) Worker(v *Vertex) error {
continue continue
} }
} }
}() }
// Worker is the common run frontend of the vertex. It handles all of the retry
// and retry delay common code, and ultimately returns the final status of this
// vertex execution.
func (g *Graph) Worker(v *Vertex) error {
// listen for chan events from Watch() and run
// the Process() function when they're received
// this avoids us having to pass the data into
// the Watch() function about which graph it is
// running on, which isolates things nicely...
obj := v.Res
if g.Flags.Debug {
log.Printf("%s[%s]: Worker: Running", v.Kind(), v.GetName())
defer log.Printf("%s[%s]: Worker: Stopped", v.Kind(), v.GetName())
}
// run the init (should match 1-1 with Close function)
if err := obj.Init(); err != nil {
// always exit the worker function by finishing with Close()
if e := obj.Close(); e != nil {
err = multierr.Append(err, e) // list of errors
}
return errwrap.Wrapf(err, "could not Init() resource")
}
// if the CheckApply run takes longer than the converged
// timeout, we could inappropriately converge mid-apply!
// avoid this by blocking convergence with a fake report
// we also add a similar blocker around the worker loop!
_, wcuid, pcuid := obj.ConvergerUIDs() // get extra cuids (worker, process)
// XXX: put these in Init() ?
wcuid.SetConverged(true) // starts off false, and waits for loop timeout
pcuid.SetConverged(true) // starts off true, because it's not running...
go g.innerWorker(v)
var err error // propagate the error up (this is a permanent BAD error!) var err error // propagate the error up (this is a permanent BAD error!)
// the watch delay runs inside of the Watch resource loop, so that it // the watch delay runs inside of the Watch resource loop, so that it
// can still process signals and exit if needed. It shouldn't run any // can still process signals and exit if needed. It shouldn't run any
@@ -545,24 +546,22 @@ func (g *Graph) Worker(v *Vertex) error {
// NOTE: we can avoid the send if running Watch guarantees // NOTE: we can avoid the send if running Watch guarantees
// one CheckApply event on startup! // one CheckApply event on startup!
//if pendingSendEvent { // TODO: should this become a list in the future? //if pendingSendEvent { // TODO: should this become a list in the future?
// if exit, err := obj.DoSend(processChan, ""); exit || err != nil { // if err := obj.Event() err != nil {
// return err // we exit or bubble up a NACK... // return err // we exit or bubble up a NACK...
// } // }
//} //}
} }
// TODO: reset the watch retry count after some amount of success // TODO: reset the watch retry count after some amount of success
v.Res.RegisterConverger()
var e error var e error
if v.Res.Meta().Poll > 0 { // poll instead of watching :( if v.Res.Meta().Poll > 0 { // poll instead of watching :(
cuid := v.Res.ConvergerUID() // get the converger uid used to report status cuid, _, _ := v.Res.ConvergerUIDs() // get the converger uid used to report status
cuid.StartTimer() cuid.StartTimer()
e = v.Res.Poll(processChan) e = v.Res.Poll()
cuid.StopTimer() // clean up nicely cuid.StopTimer() // clean up nicely
} else { } else {
e = v.Res.Watch(processChan) // run the watch normally e = v.Res.Watch() // run the watch normally
} }
v.Res.UnregisterConverger()
if e == nil { // exit signal if e == nil { // exit signal
err = nil // clean exit err = nil // clean exit
break break
@@ -586,10 +585,6 @@ func (g *Graph) Worker(v *Vertex) error {
// by getting the Watch resource to send one event once it's up! // by getting the Watch resource to send one event once it's up!
//v.SendEvent(eventPoke, false, false) //v.SendEvent(eventPoke, false, false)
} }
lock.Lock() // lock to avoid a send when closed!
finished = true
close(processChan)
lock.Unlock()
// close resource and return possible errors if any // close resource and return possible errors if any
if e := obj.Close(); err == nil { if e := obj.Close(); err == nil {
@@ -635,6 +630,7 @@ func (g *Graph) Start(first bool) { // start or continue
// TODO: if a sufficient number of workers error, // TODO: if a sufficient number of workers error,
// should something be done? Should these restart // should something be done? Should these restart
// after perma-failure if we have a graph change? // after perma-failure if we have a graph change?
log.Printf("%s[%s]: Started", vv.Kind(), vv.GetName())
if err := g.Worker(vv); err != nil { // contains the Watch and CheckApply loops if err := g.Worker(vv); err != nil { // contains the Watch and CheckApply loops
log.Printf("%s[%s]: Exited with failure: %v", vv.Kind(), vv.GetName(), err) log.Printf("%s[%s]: Exited with failure: %v", vv.Kind(), vv.GetName(), err)
return return

View File

@@ -25,7 +25,6 @@ import (
"os" "os"
"strings" "strings"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/recwatch" "github.com/purpleidea/mgmt/recwatch"
errwrap "github.com/pkg/errors" errwrap "github.com/pkg/errors"
@@ -101,7 +100,7 @@ func (obj *AugeasRes) Init() error {
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
// Taken from the File resource. // Taken from the File resource.
// FIXME: DRY - This is taken from the file resource // FIXME: DRY - This is taken from the file resource
func (obj *AugeasRes) Watch(processChan chan *event.Event) error { func (obj *AugeasRes) Watch() error {
var err error var err error
obj.recWatcher, err = recwatch.NewRecWatcher(obj.File, false) obj.recWatcher, err = recwatch.NewRecWatcher(obj.File, false)
if err != nil { if err != nil {
@@ -110,7 +109,7 @@ func (obj *AugeasRes) Watch(processChan chan *event.Event) error {
defer obj.recWatcher.Close() defer obj.recWatcher.Close()
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -146,7 +145,7 @@ func (obj *AugeasRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -26,7 +26,6 @@ import (
"os/exec" "os/exec"
"strings" "strings"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/util" "github.com/purpleidea/mgmt/util"
errwrap "github.com/pkg/errors" errwrap "github.com/pkg/errors"
@@ -98,7 +97,7 @@ func (obj *ExecRes) BufioChanScanner(scanner *bufio.Scanner) (chan string, chan
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *ExecRes) Watch(processChan chan *event.Event) error { func (obj *ExecRes) Watch() error {
var send = false // send event? var send = false // send event?
var exit *error var exit *error
bufioch, errch := make(chan string), make(chan error) bufioch, errch := make(chan string), make(chan error)
@@ -141,7 +140,7 @@ func (obj *ExecRes) Watch(processChan chan *event.Event) error {
} }
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -174,7 +173,7 @@ func (obj *ExecRes) Watch(processChan chan *event.Event) error {
send = false send = false
// it is okay to invalidate the clean state on poke too // it is okay to invalidate the clean state on poke too
obj.StateOK(false) // something made state dirty obj.StateOK(false) // something made state dirty
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -34,7 +34,6 @@ import (
"strings" "strings"
"syscall" "syscall"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/recwatch" "github.com/purpleidea/mgmt/recwatch"
"github.com/purpleidea/mgmt/util" "github.com/purpleidea/mgmt/util"
@@ -181,7 +180,7 @@ func (obj *FileRes) GetPath() string {
// If the Watch returns an error, it means that something has gone wrong, and it // If the Watch returns an error, it means that something has gone wrong, and it
// must be restarted. On a clean exit it returns nil. // must be restarted. On a clean exit it returns nil.
// FIXME: Also watch the source directory when using obj.Source !!! // FIXME: Also watch the source directory when using obj.Source !!!
func (obj *FileRes) Watch(processChan chan *event.Event) error { func (obj *FileRes) Watch() error {
var err error var err error
obj.recWatcher, err = recwatch.NewRecWatcher(obj.path, obj.Recurse) obj.recWatcher, err = recwatch.NewRecWatcher(obj.path, obj.Recurse)
if err != nil { if err != nil {
@@ -190,7 +189,7 @@ func (obj *FileRes) Watch(processChan chan *event.Event) error {
defer obj.recWatcher.Close() defer obj.recWatcher.Close()
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -226,7 +225,7 @@ func (obj *FileRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -23,7 +23,6 @@ import (
"fmt" "fmt"
"log" "log"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/util" "github.com/purpleidea/mgmt/util"
"github.com/godbus/dbus" "github.com/godbus/dbus"
@@ -102,7 +101,7 @@ func (obj *HostnameRes) Init() error {
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *HostnameRes) Watch(processChan chan *event.Event) error { func (obj *HostnameRes) Watch() error {
// if we share the bus with others, we will get each others messages!! // if we share the bus with others, we will get each others messages!!
bus, err := util.SystemBusPrivateUsable() // don't share the bus connection! bus, err := util.SystemBusPrivateUsable() // don't share the bus connection!
if err != nil { if err != nil {
@@ -120,7 +119,7 @@ func (obj *HostnameRes) Watch(processChan chan *event.Event) error {
bus.Signal(signals) bus.Signal(signals)
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -144,7 +143,7 @@ func (obj *HostnameRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -24,8 +24,6 @@ import (
"regexp" "regexp"
"strings" "strings"
"github.com/purpleidea/mgmt/event"
"github.com/coreos/go-systemd/journal" "github.com/coreos/go-systemd/journal"
) )
@@ -122,9 +120,9 @@ func (obj *MsgRes) journalPriority() journal.Priority {
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *MsgRes) Watch(processChan chan *event.Event) error { func (obj *MsgRes) Watch() error {
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -142,7 +140,7 @@ func (obj *MsgRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -21,8 +21,6 @@ import (
"encoding/gob" "encoding/gob"
"fmt" "fmt"
"log" "log"
"github.com/purpleidea/mgmt/event"
) )
func init() { func init() {
@@ -56,9 +54,9 @@ func (obj *NoopRes) Init() error {
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *NoopRes) Watch(processChan chan *event.Event) error { func (obj *NoopRes) Watch() error {
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -76,7 +74,7 @@ func (obj *NoopRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -23,7 +23,6 @@ import (
"fmt" "fmt"
"log" "log"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/util" "github.com/purpleidea/mgmt/util"
systemdUtil "github.com/coreos/go-systemd/util" systemdUtil "github.com/coreos/go-systemd/util"
@@ -98,7 +97,7 @@ func (obj *NspawnRes) Init() error {
} }
// Watch for state changes and sends a message to the bus if there is a change // Watch for state changes and sends a message to the bus if there is a change
func (obj *NspawnRes) Watch(processChan chan *event.Event) error { func (obj *NspawnRes) Watch() error {
// this resource depends on systemd ensure that it's running // this resource depends on systemd ensure that it's running
if !systemdUtil.IsRunningSystemd() { if !systemdUtil.IsRunningSystemd() {
return fmt.Errorf("Systemd is not running.") return fmt.Errorf("Systemd is not running.")
@@ -122,7 +121,7 @@ func (obj *NspawnRes) Watch(processChan chan *event.Event) error {
bus.Signal(buschan) bus.Signal(buschan)
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -155,7 +154,7 @@ func (obj *NspawnRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -28,7 +28,6 @@ import (
"path" "path"
"strings" "strings"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/recwatch" "github.com/purpleidea/mgmt/recwatch"
errwrap "github.com/pkg/errors" errwrap "github.com/pkg/errors"
@@ -165,7 +164,7 @@ Loop:
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *PasswordRes) Watch(processChan chan *event.Event) error { func (obj *PasswordRes) Watch() error {
var err error var err error
obj.recWatcher, err = recwatch.NewRecWatcher(obj.path, false) obj.recWatcher, err = recwatch.NewRecWatcher(obj.path, false)
if err != nil { if err != nil {
@@ -174,7 +173,7 @@ func (obj *PasswordRes) Watch(processChan chan *event.Event) error {
defer obj.recWatcher.Close() defer obj.recWatcher.Close()
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -203,7 +202,7 @@ func (obj *PasswordRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -24,7 +24,6 @@ import (
"path" "path"
"strings" "strings"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/resources/packagekit" "github.com/purpleidea/mgmt/resources/packagekit"
"github.com/purpleidea/mgmt/util" "github.com/purpleidea/mgmt/util"
@@ -104,7 +103,7 @@ func (obj *PkgRes) Init() error {
// It uses the PackageKit UpdatesChanged signal to watch for changes. // It uses the PackageKit UpdatesChanged signal to watch for changes.
// TODO: https://github.com/hughsie/PackageKit/issues/109 // TODO: https://github.com/hughsie/PackageKit/issues/109
// TODO: https://github.com/hughsie/PackageKit/issues/110 // TODO: https://github.com/hughsie/PackageKit/issues/110
func (obj *PkgRes) Watch(processChan chan *event.Event) error { func (obj *PkgRes) Watch() error {
bus := packagekit.NewBus() bus := packagekit.NewBus()
if bus == nil { if bus == nil {
return fmt.Errorf("Can't connect to PackageKit bus.") return fmt.Errorf("Can't connect to PackageKit bus.")
@@ -117,7 +116,7 @@ func (obj *PkgRes) Watch(processChan chan *event.Event) error {
} }
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -156,7 +155,7 @@ func (obj *PkgRes) Watch(processChan chan *event.Event) error {
// do all our event sending all together to avoid duplicate msgs // do all our event sending all together to avoid duplicate msgs
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -141,12 +141,10 @@ type Base interface {
AssociateData(*Data) AssociateData(*Data)
IsWorking() bool IsWorking() bool
Converger() converger.Converger Converger() converger.Converger
RegisterConverger() ConvergerUIDs() (converger.ConvergerUID, converger.ConvergerUID, converger.ConvergerUID)
UnregisterConverger()
ConvergerUID() converger.ConvergerUID
GetState() ResState GetState() ResState
SetState(ResState) SetState(ResState)
Event(chan *event.Event) error Event() error
SendEvent(event.EventName, error) error SendEvent(event.EventName, error) error
ReadEvent(*event.Event) (*error, bool) ReadEvent(*event.Event) (*error, bool)
Refresh() bool // is there a pending refresh to run? Refresh() bool // is there a pending refresh to run?
@@ -161,10 +159,11 @@ type Base interface {
GetGroup() []Res // return everyone grouped inside me GetGroup() []Res // return everyone grouped inside me
SetGroup([]Res) SetGroup([]Res)
VarDir(string) (string, error) VarDir(string) (string, error)
Running(chan *event.Event) error // notify the engine that Watch started Running() error // notify the engine that Watch started
Started() <-chan struct{} // returns when the resource has started Started() <-chan struct{} // returns when the resource has started
Starter(bool) Starter(bool)
Poll(chan *event.Event) error // poll alternative to watching :( Poll() error // poll alternative to watching :(
ProcessChan() chan *event.Event
Prometheus() *prometheus.Prometheus Prometheus() *prometheus.Prometheus
} }
@@ -176,7 +175,7 @@ type Res interface {
Init() error Init() error
Close() error Close() error
UIDs() []ResUID // most resources only return one UIDs() []ResUID // most resources only return one
Watch(chan *event.Event) error // send on channel to signal process() events Watch() error // send on channel to signal process() events
CheckApply(apply bool) (checkOK bool, err error) CheckApply(apply bool) (checkOK bool, err error)
AutoEdges() AutoEdge AutoEdges() AutoEdge
Compare(Res) bool Compare(Res) bool
@@ -195,6 +194,8 @@ type BaseRes struct {
events chan *event.Event events chan *event.Event
converger converger.Converger // converged tracking converger converger.Converger // converged tracking
cuid converger.ConvergerUID cuid converger.ConvergerUID
wcuid converger.ConvergerUID
pcuid converger.ConvergerUID
prometheus *prometheus.Prometheus prometheus *prometheus.Prometheus
prefix string // base prefix for this resource prefix string // base prefix for this resource
debug bool debug bool
@@ -206,6 +207,9 @@ type BaseRes struct {
isStateOK bool // whether the state is okay based on events or not isStateOK bool // whether the state is okay based on events or not
isGrouped bool // am i contained within a group? isGrouped bool // am i contained within a group?
grouped []Res // list of any grouped resources grouped []Res // list of any grouped resources
processLock *sync.Mutex
processDone bool
processChan chan *event.Event
refresh bool // does this resource have a refresh to run? refresh bool // does this resource have a refresh to run?
//refreshState StatefulBool // TODO: future stateful bool //refreshState StatefulBool // TODO: future stateful bool
} }
@@ -291,7 +295,13 @@ func (obj *BaseRes) Init() error {
if obj.kind == "" { if obj.kind == "" {
return fmt.Errorf("Resource did not set kind!") return fmt.Errorf("Resource did not set kind!")
} }
obj.cuid = obj.converger.Register()
obj.wcuid = obj.converger.Register() // get a cuid for the worker!
obj.pcuid = obj.converger.Register() // get a cuid for the process
obj.mutex = &sync.Mutex{} obj.mutex = &sync.Mutex{}
obj.working = true // Worker method should now be running...
obj.events = make(chan *event.Event) // unbuffered chan to avoid stale events obj.events = make(chan *event.Event) // unbuffered chan to avoid stale events
obj.started = make(chan struct{}) // closes when started obj.started = make(chan struct{}) // closes when started
@@ -303,6 +313,10 @@ func (obj *BaseRes) Init() error {
obj.Meta().Limit = rate.Inf obj.Meta().Limit = rate.Inf
} }
obj.processLock = &sync.Mutex{} // lock around processChan closing and sending
obj.processDone = false // did we close processChan ?
obj.processChan = make(chan *event.Event)
//dir, err := obj.VarDir("") //dir, err := obj.VarDir("")
//if err != nil { //if err != nil {
// return errwrap.Wrapf(err, "VarDir failed in Init()") // return errwrap.Wrapf(err, "VarDir failed in Init()")
@@ -310,7 +324,6 @@ func (obj *BaseRes) Init() error {
// TODO: this StatefulBool implementation could be eventually swappable // TODO: this StatefulBool implementation could be eventually swappable
//obj.refreshState = &DiskBool{Path: path.Join(dir, refreshPathToken)} //obj.refreshState = &DiskBool{Path: path.Join(dir, refreshPathToken)}
obj.working = true // Worker method should now be running...
return nil return nil
} }
@@ -319,10 +332,21 @@ func (obj *BaseRes) Close() error {
if obj.debug { if obj.debug {
log.Printf("%s[%s]: Close()", obj.Kind(), obj.GetName()) log.Printf("%s[%s]: Close()", obj.Kind(), obj.GetName())
} }
obj.processLock.Lock() // lock to avoid a send when closed!
obj.processDone = true
close(obj.processChan)
obj.processLock.Unlock()
obj.mutex.Lock() obj.mutex.Lock()
obj.working = false // Worker method should now be closing... obj.working = false // Worker method should now be closing...
close(obj.events) // this is where we properly close this channel! close(obj.events) // this is where we properly close this channel!
obj.mutex.Unlock() obj.mutex.Unlock()
obj.pcuid.Unregister()
obj.wcuid.Unregister()
obj.cuid.Unregister()
return nil return nil
} }
@@ -375,22 +399,11 @@ func (obj *BaseRes) Converger() converger.Converger {
return obj.converger return obj.converger
} }
// RegisterConverger sets up the cuid for the resource. This is a helper // ConvergerUIDs returns the ConvergerUIDs for the resource. This is called by
// function for the engine, and shouldn't be called by the resources directly. // the various methods that need one of these ConvergerUIDs. They are registered
func (obj *BaseRes) RegisterConverger() { // by the Init method and unregistered on the resource Close.
obj.cuid = obj.converger.Register() func (obj *BaseRes) ConvergerUIDs() (cuid converger.ConvergerUID, wcuid converger.ConvergerUID, pcuid converger.ConvergerUID) {
} return obj.cuid, obj.wcuid, obj.pcuid
// UnregisterConverger tears down the cuid for the resource. This is a helper
// function for the engine, and shouldn't be called by the resources directly.
func (obj *BaseRes) UnregisterConverger() {
obj.cuid.Unregister()
}
// ConvergerUID returns the ConvergerUID for the resource. This should be called
// by the Watch method of the resource to set the converged state.
func (obj *BaseRes) ConvergerUID() converger.ConvergerUID {
return obj.cuid
} }
// GetState returns the state of the resource. // GetState returns the state of the resource.
@@ -416,6 +429,11 @@ func (obj *BaseRes) StateOK(b bool) {
obj.isStateOK = b obj.isStateOK = b
} }
// ProcessChan returns the chan that resources send events to. Internal API!
func (obj *BaseRes) ProcessChan() chan *event.Event {
return obj.processChan
}
// GroupCmp compares two resources and decides if they're suitable for grouping // GroupCmp compares two resources and decides if they're suitable for grouping
// You'll probably want to override this method when implementing a resource... // You'll probably want to override this method when implementing a resource...
func (obj *BaseRes) GroupCmp(res Res) bool { func (obj *BaseRes) GroupCmp(res Res) bool {
@@ -528,15 +546,15 @@ func (obj *BaseRes) Started() <-chan struct{} { return obj.started }
func (obj *BaseRes) Starter(b bool) { obj.starter = b } func (obj *BaseRes) Starter(b bool) { obj.starter = b }
// Poll is the watch replacement for when we want to poll, which outputs events. // Poll is the watch replacement for when we want to poll, which outputs events.
func (obj *BaseRes) Poll(processChan chan *event.Event) error { func (obj *BaseRes) Poll() error {
cuid := obj.ConvergerUID() // get the converger uid used to report status cuid, _, _ := obj.ConvergerUIDs() // get the converger uid used to report status
// create a time.Ticker for the given interval // create a time.Ticker for the given interval
ticker := time.NewTicker(time.Duration(obj.Meta().Poll) * time.Second) ticker := time.NewTicker(time.Duration(obj.Meta().Poll) * time.Second)
defer ticker.Stop() defer ticker.Stop()
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
cuid.SetConverged(false) // quickly stop any converge due to Running() cuid.SetConverged(false) // quickly stop any converge due to Running()
@@ -559,7 +577,7 @@ func (obj *BaseRes) Poll(processChan chan *event.Event) error {
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -29,9 +29,13 @@ import (
) )
// Event sends off an event, but doesn't block the incoming event queue. // Event sends off an event, but doesn't block the incoming event queue.
func (obj *BaseRes) Event(processChan chan *event.Event) error { func (obj *BaseRes) Event() error {
resp := event.NewResp() resp := event.NewResp()
processChan <- &event.Event{Name: event.EventNil, Resp: resp} // trigger process obj.processLock.Lock()
if !obj.processDone {
obj.processChan <- &event.Event{Name: event.EventNil, Resp: resp} // trigger process
}
obj.processLock.Unlock()
return resp.Wait() return resp.Wait()
} }
@@ -97,14 +101,14 @@ func (obj *BaseRes) ReadEvent(ev *event.Event) (exit *error, send bool) {
// Running is called by the Watch method of the resource once it has started up. // Running is called by the Watch method of the resource once it has started up.
// This signals to the engine to kick off the initial CheckApply resource check. // This signals to the engine to kick off the initial CheckApply resource check.
func (obj *BaseRes) Running(processChan chan *event.Event) error { func (obj *BaseRes) Running() error {
// TODO: If a non-polling resource wants to use the converger, then it // TODO: If a non-polling resource wants to use the converger, then it
// should probably tell Running (via an arg) to not do this. Currently // should probably tell Running (via an arg) to not do this. Currently
// it is a very unlikey race that could cause an early converge if the // it is a very unlikey race that could cause an early converge if the
// converge timeout is very short ( ~ 1s) and the Watch method doesn't // converge timeout is very short ( ~ 1s) and the Watch method doesn't
// immediately SetConverged(false) to stop possible early termination. // immediately SetConverged(false) to stop possible early termination.
if obj.Meta().Poll == 0 { // if not polling, unblock this... if obj.Meta().Poll == 0 { // if not polling, unblock this...
cuid := obj.ConvergerUID() cuid, _, _ := obj.ConvergerUIDs()
cuid.SetConverged(true) // a reasonable initial assumption cuid.SetConverged(true) // a reasonable initial assumption
} }
@@ -116,7 +120,7 @@ func (obj *BaseRes) Running(processChan chan *event.Event) error {
var err error var err error
if obj.starter { // vertices of indegree == 0 should send initial pokes if obj.starter { // vertices of indegree == 0 should send initial pokes
err = obj.Event(processChan) // trigger a CheckApply err = obj.Event() // trigger a CheckApply
} }
return err // bubble up any possible error (or nil) return err // bubble up any possible error (or nil)
} }

View File

@@ -24,7 +24,6 @@ import (
"fmt" "fmt"
"log" "log"
"github.com/purpleidea/mgmt/event"
"github.com/purpleidea/mgmt/util" "github.com/purpleidea/mgmt/util"
systemd "github.com/coreos/go-systemd/dbus" // change namespace systemd "github.com/coreos/go-systemd/dbus" // change namespace
@@ -71,7 +70,7 @@ func (obj *SvcRes) Init() error {
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *SvcRes) Watch(processChan chan *event.Event) error { func (obj *SvcRes) Watch() error {
// obj.Name: svc name // obj.Name: svc name
if !systemdUtil.IsRunningSystemd() { if !systemdUtil.IsRunningSystemd() {
return fmt.Errorf("Systemd is not running.") return fmt.Errorf("Systemd is not running.")
@@ -96,7 +95,7 @@ func (obj *SvcRes) Watch(processChan chan *event.Event) error {
bus.Signal(buschan) bus.Signal(buschan)
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -197,7 +196,7 @@ func (obj *SvcRes) Watch(processChan chan *event.Event) error {
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -22,8 +22,6 @@ import (
"fmt" "fmt"
"log" "log"
"time" "time"
"github.com/purpleidea/mgmt/event"
) )
func init() { func init() {
@@ -70,13 +68,13 @@ func (obj *TimerRes) newTicker() *time.Ticker {
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *TimerRes) Watch(processChan chan *event.Event) error { func (obj *TimerRes) Watch() error {
// create a time.Ticker for the given interval // create a time.Ticker for the given interval
obj.ticker = obj.newTicker() obj.ticker = obj.newTicker()
defer obj.ticker.Stop() defer obj.ticker.Stop()
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -96,7 +94,7 @@ func (obj *TimerRes) Watch(processChan chan *event.Event) error {
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }

View File

@@ -29,8 +29,6 @@ import (
"strings" "strings"
"time" "time"
"github.com/purpleidea/mgmt/event"
multierr "github.com/hashicorp/go-multierror" multierr "github.com/hashicorp/go-multierror"
"github.com/libvirt/libvirt-go" "github.com/libvirt/libvirt-go"
libvirtxml "github.com/libvirt/libvirt-go-xml" libvirtxml "github.com/libvirt/libvirt-go-xml"
@@ -250,7 +248,7 @@ func (obj *VirtRes) connect() (conn *libvirt.Connect, err error) {
} }
// Watch is the primary listener for this resource and it outputs events. // Watch is the primary listener for this resource and it outputs events.
func (obj *VirtRes) Watch(processChan chan *event.Event) error { func (obj *VirtRes) Watch() error {
domChan := make(chan libvirt.DomainEventType) // TODO: do we need to buffer this? domChan := make(chan libvirt.DomainEventType) // TODO: do we need to buffer this?
gaChan := make(chan *libvirt.DomainEventAgentLifecycle) gaChan := make(chan *libvirt.DomainEventAgentLifecycle)
errorChan := make(chan error) errorChan := make(chan error)
@@ -308,7 +306,7 @@ func (obj *VirtRes) Watch(processChan chan *event.Event) error {
defer obj.conn.DomainEventDeregister(gaCallbackID) defer obj.conn.DomainEventDeregister(gaCallbackID)
// notify engine that we're running // notify engine that we're running
if err := obj.Running(processChan); err != nil { if err := obj.Running(); err != nil {
return err // bubble up a NACK... return err // bubble up a NACK...
} }
@@ -400,7 +398,7 @@ func (obj *VirtRes) Watch(processChan chan *event.Event) error {
if send { if send {
send = false send = false
obj.Event(processChan) obj.Event()
} }
} }
} }