union.go

package kapacitor

import (
	"github.com/influxdb/kapacitor/pipeline"
)

type UnionNode struct {
	node
	u *pipeline.UnionNode
}

// Create a new  UnionNode which combines all parent data streams into a single stream.
// No transformation of any kind is performed.
func newUnionNode(et *ExecutingTask, n *pipeline.UnionNode) (*UnionNode, error) {
	un := &UnionNode{
		u:    n,
		node: node{Node: n, et: et},
	}
	un.node.runF = un.runUnion
	return un, nil
}

func (u *UnionNode) runUnion() error {
	errors := make(chan error, len(u.ins))
	for _, in := range u.ins {
		go func(e *Edge) {
			switch u.Wants() {
			case pipeline.StreamEdge:
				for p := e.NextPoint(); p != nil; p = e.NextPoint() {
					for _, out := range u.outs {
						err := out.CollectPoint(p)
						if err != nil {
							errors <- err
							return
						}
					}
				}
			case pipeline.BatchEdge:
				for b := e.NextBatch(); b != nil; b = e.NextBatch() {
					for _, out := range u.outs {
						err := out.CollectBatch(b)
						if err != nil {
							errors <- err
							return
						}
					}
				}
			}
			errors <- nil
		}(in)
	}

	for range u.ins {
		err := <-errors
		if err != nil {
			return err
		}
	}
	return nil
}