grafana/pkg/services/live/pipeline/output_conditional.go

package pipeline

import (
	"context"

	"github.com/grafana/grafana-plugin-sdk-go/data"
)

type ConditionalOutput struct {
	Condition ConditionChecker
	Outputter Outputter
}

func NewConditionalOutput(condition ConditionChecker, outputter Outputter) *ConditionalOutput {
	return &ConditionalOutput{Condition: condition, Outputter: outputter}
}

const OutputTypeConditional = "conditional"

func (out *ConditionalOutput) Type() string {
	return OutputTypeConditional
}

func (out ConditionalOutput) Output(ctx context.Context, vars OutputVars, frame *data.Frame) ([]*ChannelFrame, error) {
	ok, err := out.Condition.CheckCondition(ctx, frame)
	if err != nil {
		return nil, err
	}
	if !ok {
		return nil, nil
	}
	return out.Outputter.Output(ctx, vars, frame)
}