summary_event_flow.go

package evaluation

import (
	"errors"
	"strings"

	"github.com/rai-project/evaluation/eventflow"
	"github.com/rai-project/evaluation/writer"
	db "upper.io/db.v3"
)

//easyjson:json
type SummaryEventFlow struct {
	SummaryBase `json:",inline"`
	EventFlow   eventflow.Events `json:"event_flow,omitempty"`
}

type SummaryEventFlows []SummaryEventFlow

func (SummaryEventFlow) Header(opts ...writer.Option) []string {
	extra := eventflow.Event{}.Header()
	return append(SummaryBase{}.Header(opts...), extra...)
}

func (s SummaryEventFlow) Row(opts ...writer.Option) []string {
	events := []string{}
	for _, e := range s.EventFlow {
		events = append(events, strings.Join(e.Row(opts...), "\t"))
	}
	extra := []string{
		strings.Join(events, ";"),
	}
	return append(s.SummaryBase.Row(opts...), extra...)
}

func (SummaryEventFlows) Header(opts ...writer.Option) []string {
	return SummaryEventFlow{}.Header(opts...)
}

func (s SummaryEventFlows) Rows(opts ...writer.Option) [][]string {
	rows := [][]string{}
	for _, e := range s {
		rows = append(rows, e.Row(opts...))
	}
	return rows
}

func spansToEventFlow(spans Spans) eventflow.Events {
	return eventflow.SpansToEvenFlow(spans)
}

func (p Performance) EventFlowSummary(e Evaluation) (*SummaryEventFlow, error) {
	perfSpans, err := p.Spans()
	if err != nil {
		return nil, err
	}
	flow := spansToEventFlow(perfSpans)
	return &SummaryEventFlow{
		SummaryBase: e.summaryBase(),
		EventFlow:   flow,
	}, nil
}

func (e Evaluation) EventFlowSummary(perfCol *PerformanceCollection) (*SummaryEventFlow, error) {
	perfs, err := perfCol.Find(db.Cond{"_id": e.PerformanceID})
	if err != nil {
		return nil, err
	}
	if len(perfs) != 1 {
		return nil, errors.New("expecting on performance output")
	}
	perf := perfs[0]
	return perf.EventFlowSummary(e)
}

func (es Evaluations) EventFlowSummary(perfCol *PerformanceCollection) (SummaryEventFlows, error) {
	res := []SummaryEventFlow{}
	for _, e := range es {
		s, err := e.EventFlowSummary(perfCol)
		if err != nil {
			log.WithError(err).Error("failed to get layer information summary")
			continue
		}
		if s == nil {
			continue
		}
		res = append(res, *s)
	}
	return res, nil
}