/
chained_handlers.go
73 lines (60 loc) · 2.07 KB
/
chained_handlers.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
package eventhandler
import (
"context"
"fmt"
"time"
"github.com/bacalhau-project/bacalhau/pkg/model"
"github.com/bacalhau-project/bacalhau/pkg/system"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)
// An event handler implementation that chains multiple event handlers, and accepts a context provider
// to setup up the context once for all handlers.
type ChainedJobEventHandler struct {
eventHandlers []JobEventHandler
contextProvider ContextProvider
}
func NewChainedJobEventHandler(contextProvider ContextProvider) *ChainedJobEventHandler {
return &ChainedJobEventHandler{contextProvider: contextProvider}
}
func (r *ChainedJobEventHandler) AddHandlers(handlers ...JobEventHandler) {
r.eventHandlers = append(r.eventHandlers, handlers...)
}
func (r *ChainedJobEventHandler) HandleJobEvent(ctx context.Context, event model.JobEvent) (err error) {
startTime := time.Now()
defer logEvent(ctx, event, startTime)(&err)
if r.eventHandlers == nil {
return fmt.Errorf("no event handlers registered")
}
jobCtx := r.contextProvider.GetContext(ctx, event.JobID)
// All handlers are called, unless one of them returns an error.
for _, handler := range r.eventHandlers {
if err = handler.HandleJobEvent(jobCtx, event); err != nil { //nolint:gocritic
return err
}
}
return nil
}
func logEvent(ctx context.Context, event model.JobEvent, startTime time.Time) func(*error) {
return func(handlerError *error) {
var logMsg *zerolog.Event
// TODO: #829 Is checking environment every event the most efficient way
// to do this? Could we just shunt logs to different places?
switch system.GetEnvironment() {
case system.EnvironmentDev, system.EnvironmentTest:
logMsg = log.Ctx(ctx).Trace()
default:
logMsg = log.Ctx(ctx).Info()
}
logMsg = logMsg.
Str("EventName", event.EventName.String()).
Str("JobID", event.JobID).
Str("NodeID", event.SourceNodeID).
Str("Status", event.Status).
Dur("HandleDuration", time.Since(startTime))
if *handlerError != nil {
logMsg = logMsg.AnErr("HandlerError", *handlerError)
}
logMsg.Msg("Handled event")
}
}