lib/ur: Fix panics in failure-reporting (fixes #7090) (#7091)

This commit is contained in:
Simon Frei 2020-11-03 12:29:33 +01:00 committed by GitHub
parent 942b8ebb27
commit a38b370c8d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -29,6 +29,8 @@ var (
minDelay = 10 * time.Second minDelay = 10 * time.Second
maxDelay = time.Minute maxDelay = time.Minute
sendTimeout = time.Minute sendTimeout = time.Minute
evChanClosed = "failure event channel closed"
invalidEventDataType = "failure event data is not a string"
) )
type FailureReport struct { type FailureReport struct {
@ -47,6 +49,7 @@ func NewFailureHandler(cfg config.Wrapper, evLogger events.Logger) FailureHandle
cfg: cfg, cfg: cfg,
evLogger: evLogger, evLogger: evLogger,
optsChan: make(chan config.OptionsConfiguration), optsChan: make(chan config.OptionsConfiguration),
buf: make(map[string]*failureStat),
} }
h.Service = util.AsServiceWithError(h.serve, h.String()) h.Service = util.AsServiceWithError(h.serve, h.String())
return h return h
@ -57,7 +60,6 @@ type failureHandler struct {
cfg config.Wrapper cfg config.Wrapper
evLogger events.Logger evLogger events.Logger
optsChan chan config.OptionsConfiguration optsChan chan config.OptionsConfiguration
evChan <-chan events.Event
buf map[string]*failureStat buf map[string]*failureStat
} }
@ -68,7 +70,10 @@ type failureStat struct {
func (h *failureHandler) serve(ctx context.Context) error { func (h *failureHandler) serve(ctx context.Context) error {
go func() { go func() {
h.optsChan <- h.cfg.Options() select {
case h.optsChan <- h.cfg.Options():
case <-ctx.Done():
}
}() }()
h.cfg.Subscribe(h) h.cfg.Subscribe(h)
defer h.cfg.Unsubscribe(h) defer h.cfg.Unsubscribe(h)
@ -76,6 +81,7 @@ func (h *failureHandler) serve(ctx context.Context) error {
var url string var url string
var err error var err error
var sub events.Subscription var sub events.Subscription
var evChan <-chan events.Event
timer := time.NewTimer(minDelay) timer := time.NewTimer(minDelay)
resetTimer := make(chan struct{}) resetTimer := make(chan struct{})
outer: outer:
@ -86,25 +92,29 @@ outer:
if opts.URAccepted > 0 { if opts.URAccepted > 0 {
if sub == nil { if sub == nil {
sub = h.evLogger.Subscribe(events.Failure) sub = h.evLogger.Subscribe(events.Failure)
h.evChan = sub.C() evChan = sub.C()
} }
} else if sub != nil { } else if sub != nil {
sub.Unsubscribe() sub.Unsubscribe()
sub = nil sub = nil
evChan = nil
} }
url = opts.CRURL + "/failure" url = opts.CRURL + "/failure"
case e := <-h.evChan: case e, ok := <-evChan:
descr := e.Data.(string) if !ok {
if stat, ok := h.buf[descr]; ok { // Just to be safe - shouldn't ever happen, as
stat.last = e.Time // evChan is set to nil when unsubscribing.
stat.count++ h.addReport(evChanClosed, time.Now())
} else { evChan = nil
h.buf[descr] = &failureStat{ continue
first: e.Time,
last: e.Time,
count: 1,
} }
descr, ok := e.Data.(string)
if !ok {
// Same here, shouldn't ever happen.
h.addReport(invalidEventDataType, time.Now())
continue
} }
h.addReport(descr, e.Time)
case <-timer.C: case <-timer.C:
reports := make([]FailureReport, 0, len(h.buf)) reports := make([]FailureReport, 0, len(h.buf))
now := time.Now() now := time.Now()
@ -141,6 +151,19 @@ outer:
return err return err
} }
func (h *failureHandler) addReport(descr string, evTime time.Time) {
if stat, ok := h.buf[descr]; ok {
stat.last = evTime
stat.count++
return
}
h.buf[descr] = &failureStat{
first: evTime,
last: evTime,
count: 1,
}
}
func (h *failureHandler) VerifyConfiguration(_, _ config.Configuration) error { func (h *failureHandler) VerifyConfiguration(_, _ config.Configuration) error {
return nil return nil
} }