func (w *WhereNode) runWhere(snapshot []byte) error {
switch w.Wants() {
case pipeline.StreamEdge:
for p, ok := w.ins[0].NextPoint(); ok; p, ok = w.ins[0].NextPoint() {
w.timer.Start()
expr := w.expressions[p.Group]
if expr == nil {
expr = tick.NewStatefulExpr(w.w.Expression)
w.expressions[p.Group] = expr
}
if pass, err := EvalPredicate(expr, p.Time, p.Fields, p.Tags); pass {
w.timer.Pause()
for _, child := range w.outs {
err := child.CollectPoint(p)
if err != nil {
return err
}
}
w.timer.Resume()
} else if err != nil {
w.logger.Println("E! error while evaluating expression:", err)
}
w.timer.Stop()
}
case pipeline.BatchEdge:
for b, ok := w.ins[0].NextBatch(); ok; b, ok = w.ins[0].NextBatch() {
w.timer.Start()
expr := w.expressions[b.Group]
if expr == nil {
expr = tick.NewStatefulExpr(w.w.Expression)
w.expressions[b.Group] = expr
}
for i := 0; i < len(b.Points); {
p := b.Points[i]
if pass, err := EvalPredicate(expr, p.Time, p.Fields, p.Tags); !pass {
if err != nil {
w.logger.Println("E! error while evaluating WHERE expression:", err)
}
b.Points = append(b.Points[:i], b.Points[i+1:]...)
} else {
i++
}
}
w.timer.Stop()
if len(b.Points) > 0 {
for _, child := range w.outs {
err := child.CollectBatch(b)
if err != nil {
return err
}
}
}
}
}
return nil
}
func (w *WhereNode) runWhere() error {
switch w.Wants() {
case pipeline.StreamEdge:
for p, ok := w.ins[0].NextPoint(); ok; p, ok = w.ins[0].NextPoint() {
expr := w.expressions[p.Group]
if expr == nil {
expr = tick.NewStatefulExpr(w.w.Expression)
w.expressions[p.Group] = expr
}
if pass, err := EvalPredicate(expr, p.Fields, p.Tags); pass {
for _, child := range w.outs {
err := child.CollectPoint(p)
if err != nil {
return err
}
}
} else if err != nil {
w.logger.Println("E! error while evaluating expression:", err)
}
}
case pipeline.BatchEdge:
for b, ok := w.ins[0].NextBatch(); ok; b, ok = w.ins[0].NextBatch() {
expr := w.expressions[b.Group]
if expr == nil {
expr = tick.NewStatefulExpr(w.w.Expression)
w.expressions[b.Group] = expr
}
for i, p := range b.Points {
if pass, err := EvalPredicate(expr, p.Fields, p.Tags); !pass {
if err != nil {
w.logger.Println("E! error while evaluating WHERE expression:", err)
}
b.Points = append(b.Points[:i], b.Points[i+1:]...)
}
}
for _, child := range w.outs {
err := child.CollectBatch(b)
if err != nil {
return err
}
}
}
}
return nil
}
// Create a new EvalNode which applies a transformation func to each point in a stream and returns a single point.
func newEvalNode(et *ExecutingTask, n *pipeline.EvalNode, l *log.Logger) (*EvalNode, error) {
if len(n.AsList) != len(n.Expressions) {
return nil, errors.New("must provide one name per expression via the 'As' property")
}
en := &EvalNode{
node: node{Node: n, et: et, logger: l},
e: n,
}
// Create stateful expressions
en.expressions = make([]*tick.StatefulExpr, len(n.Expressions))
for i, expr := range n.Expressions {
en.expressions[i] = tick.NewStatefulExpr(expr)
}
en.node.runF = en.runEval
return en, nil
}
// Create a new StreamNode which filters data from a source.
func newStreamNode(et *ExecutingTask, n *pipeline.StreamNode, l *log.Logger) (*StreamNode, error) {
sn := &StreamNode{
node: node{Node: n, et: et, logger: l},
s: n,
db: n.Database,
rp: n.RetentionPolicy,
name: n.Measurement,
}
sn.node.runF = sn.runStream
sn.allDimensions, sn.dimensions = determineDimensions(n.Dimensions)
if n.Expression != nil {
sn.expression = tick.NewStatefulExpr(n.Expression)
}
return sn, nil
}
// Create a new AlertNode which caches the most recent item and exposes it over the HTTP API.
func newAlertNode(et *ExecutingTask, n *pipeline.AlertNode) (an *AlertNode, err error) {
an = &AlertNode{
node: node{Node: n, et: et},
a: n,
}
an.node.runF = an.runAlert
// Parse templates
tmpl, err := template.New("id").Parse(n.Id)
if err != nil {
return nil, err
}
an.idTmpl = tmpl
tmpl, err = template.New("message").Parse(n.Message)
if err != nil {
return nil, err
}
an.messageTmpl = tmpl
// Construct alert handlers
an.handlers = make([]AlertHandler, 0)
for _, post := range n.PostHandlers {
post := post
an.handlers = append(an.handlers, func(ad *AlertData) { an.handlePost(post, ad) })
}
for _, email := range n.EmailHandlers {
email := email
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleEmail(email, ad) })
}
if len(n.EmailHandlers) == 0 && (et.tm.SMTPService != nil && et.tm.SMTPService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleEmail(&pipeline.EmailHandler{}, ad) })
}
// If email has been configured globally only send state changes.
if et.tm.SMTPService != nil && et.tm.SMTPService.Global() {
n.IsStateChangesOnly = true
}
for _, exec := range n.ExecHandlers {
exec := exec
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleExec(exec, ad) })
}
for _, log := range n.LogHandlers {
log := log
if !path.IsAbs(log.FilePath) {
return nil, fmt.Errorf("alert log path must be absolute: %s is not absolute", log.FilePath)
}
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleLog(log, ad) })
}
for _, vo := range n.VictorOpsHandlers {
vo := vo
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleVictorOps(vo, ad) })
}
if len(n.VictorOpsHandlers) == 0 && (et.tm.VictorOpsService != nil && et.tm.VictorOpsService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleVictorOps(&pipeline.VictorOpsHandler{}, ad) })
}
for _, pd := range n.PagerDutyHandlers {
pd := pd
an.handlers = append(an.handlers, func(ad *AlertData) { an.handlePagerDuty(pd, ad) })
}
if len(n.PagerDutyHandlers) == 0 && (et.tm.PagerDutyService != nil && et.tm.PagerDutyService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handlePagerDuty(&pipeline.PagerDutyHandler{}, ad) })
}
for _, slack := range n.SlackHandlers {
slack := slack
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleSlack(slack, ad) })
}
if len(n.SlackHandlers) == 0 && (et.tm.SlackService != nil && et.tm.SlackService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleSlack(&pipeline.SlackHandler{}, ad) })
}
// If slack has been configured globally only send state changes.
if et.tm.SlackService != nil && et.tm.SlackService.Global() {
n.IsStateChangesOnly = true
}
for _, hipchat := range n.HipChatHandlers {
hipchat := hipchat
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleHipChat(hipchat, ad) })
}
if len(n.HipChatHandlers) == 0 && (et.tm.HipChatService != nil && et.tm.HipChatService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleHipChat(&pipeline.HipChatHandler{}, ad) })
}
// If HipChat has been configured globally only send state changes.
if et.tm.HipChatService != nil && et.tm.HipChatService.Global() {
n.IsStateChangesOnly = true
}
for _, og := range n.OpsGenieHandlers {
og := og
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleOpsGenie(og, ad) })
}
if len(n.OpsGenieHandlers) == 0 && (et.tm.OpsGenieService != nil && et.tm.OpsGenieService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleOpsGenie(&pipeline.OpsGenieHandler{}, ad) })
}
// Parse level expressions
an.levels = make([]*tick.StatefulExpr, CritAlert+1)
if n.Info != nil {
an.levels[InfoAlert] = tick.NewStatefulExpr(n.Info)
}
if n.Warn != nil {
an.levels[WarnAlert] = tick.NewStatefulExpr(n.Warn)
}
if n.Crit != nil {
an.levels[CritAlert] = tick.NewStatefulExpr(n.Crit)
}
// Setup states
if n.History < 2 {
n.History = 2
}
an.states = make(map[models.GroupID]*alertState)
// Configure flapping
if n.UseFlapping {
if n.FlapLow > 1 || n.FlapHigh > 1 {
return nil, errors.New("alert flap thresholds are percentages and should be between 0 and 1")
}
}
return
}
// Create a new AlertNode which caches the most recent item and exposes it over the HTTP API.
func newAlertNode(et *ExecutingTask, n *pipeline.AlertNode, l *log.Logger) (an *AlertNode, err error) {
an = &AlertNode{
node: node{Node: n, et: et, logger: l},
a: n,
}
an.node.runF = an.runAlert
// Parse templates
an.idTmpl, err = text.New("id").Parse(n.Id)
if err != nil {
return nil, err
}
an.messageTmpl, err = text.New("message").Parse(n.Message)
if err != nil {
return nil, err
}
an.detailsTmpl, err = html.New("details").Funcs(html.FuncMap{
"json": func(v interface{}) html.JS {
a, _ := json.Marshal(v)
return html.JS(a)
},
}).Parse(n.Details)
if err != nil {
return nil, err
}
// Construct alert handlers
an.handlers = make([]AlertHandler, 0)
for _, post := range n.PostHandlers {
post := post
an.handlers = append(an.handlers, func(ad *AlertData) { an.handlePost(post, ad) })
}
for _, email := range n.EmailHandlers {
email := email
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleEmail(email, ad) })
}
if len(n.EmailHandlers) == 0 && (et.tm.SMTPService != nil && et.tm.SMTPService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleEmail(&pipeline.EmailHandler{}, ad) })
}
// If email has been configured globally only send state changes.
if et.tm.SMTPService != nil && et.tm.SMTPService.Global() {
n.IsStateChangesOnly = true
}
for _, exec := range n.ExecHandlers {
exec := exec
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleExec(exec, ad) })
}
for _, log := range n.LogHandlers {
log := log
if !path.IsAbs(log.FilePath) {
return nil, fmt.Errorf("alert log path must be absolute: %s is not absolute", log.FilePath)
}
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleLog(log, ad) })
}
for _, vo := range n.VictorOpsHandlers {
vo := vo
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleVictorOps(vo, ad) })
}
if len(n.VictorOpsHandlers) == 0 && (et.tm.VictorOpsService != nil && et.tm.VictorOpsService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleVictorOps(&pipeline.VictorOpsHandler{}, ad) })
}
for _, pd := range n.PagerDutyHandlers {
pd := pd
an.handlers = append(an.handlers, func(ad *AlertData) { an.handlePagerDuty(pd, ad) })
}
if len(n.PagerDutyHandlers) == 0 && (et.tm.PagerDutyService != nil && et.tm.PagerDutyService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handlePagerDuty(&pipeline.PagerDutyHandler{}, ad) })
}
for _, sensu := range n.SensuHandlers {
sensu := sensu
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleSensu(sensu, ad) })
}
for _, slack := range n.SlackHandlers {
slack := slack
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleSlack(slack, ad) })
}
if len(n.SlackHandlers) == 0 && (et.tm.SlackService != nil && et.tm.SlackService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleSlack(&pipeline.SlackHandler{}, ad) })
}
// If slack has been configured globally only send state changes.
if et.tm.SlackService != nil && et.tm.SlackService.Global() {
n.IsStateChangesOnly = true
}
for _, hipchat := range n.HipChatHandlers {
hipchat := hipchat
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleHipChat(hipchat, ad) })
}
if len(n.HipChatHandlers) == 0 && (et.tm.HipChatService != nil && et.tm.HipChatService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleHipChat(&pipeline.HipChatHandler{}, ad) })
}
// If HipChat has been configured globally only send state changes.
if et.tm.HipChatService != nil && et.tm.HipChatService.Global() {
n.IsStateChangesOnly = true
}
for _, alerta := range n.AlertaHandlers {
// Validate alerta templates
rtmpl, err := text.New("resource").Parse(alerta.Resource)
if err != nil {
return nil, err
}
gtmpl, err := text.New("group").Parse(alerta.Group)
if err != nil {
return nil, err
}
vtmpl, err := text.New("value").Parse(alerta.Value)
if err != nil {
return nil, err
}
ai := alertaHandler{
AlertaHandler: alerta,
resourceTmpl: rtmpl,
groupTmpl: gtmpl,
valueTmpl: vtmpl,
}
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleAlerta(ai, ad) })
}
for _, og := range n.OpsGenieHandlers {
og := og
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleOpsGenie(og, ad) })
}
if len(n.OpsGenieHandlers) == 0 && (et.tm.OpsGenieService != nil && et.tm.OpsGenieService.Global()) {
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleOpsGenie(&pipeline.OpsGenieHandler{}, ad) })
}
for _, talk := range n.TalkHandlers {
talk := talk
an.handlers = append(an.handlers, func(ad *AlertData) { an.handleTalk(talk, ad) })
}
// Parse level expressions
an.levels = make([]*tick.StatefulExpr, CritAlert+1)
if n.Info != nil {
an.levels[InfoAlert] = tick.NewStatefulExpr(n.Info)
}
if n.Warn != nil {
an.levels[WarnAlert] = tick.NewStatefulExpr(n.Warn)
}
if n.Crit != nil {
an.levels[CritAlert] = tick.NewStatefulExpr(n.Crit)
}
// Setup states
if n.History < 2 {
n.History = 2
}
an.states = make(map[models.GroupID]*alertState)
// Configure flapping
if n.UseFlapping {
if n.FlapLow > 1 || n.FlapHigh > 1 {
return nil, errors.New("alert flap thresholds are percentages and should be between 0 and 1")
}
}
return
}