environment.go

package phonelab

// The environment maintains state about what we know how to create/run
type ParserGen func() Parser

type ProcessorGen interface {
	GenerateProcessor(info *PipelineSourceInstance, kwargs map[string]interface{}) Processor
}

type ProcessorGenFunc func(info *PipelineSourceInstance, kwargs map[string]interface{}) Processor

type ProcessorGenWrapper struct {
	Gen ProcessorGenFunc
}

func (g *ProcessorGenWrapper) GenerateProcessor(info *PipelineSourceInstance,
	kwargs map[string]interface{}) Processor {

	return g.Gen(info, kwargs)
}

type DataCollectorGen func(kwargs map[string]interface{}) DataCollector

type Environment struct {
	// Parsers we know about
	Parsers        map[string]ParserGen
	Processors     map[string]ProcessorGen
	DataCollectors map[string]DataCollectorGen
	Filters        map[string]StringFilter
}

func NewEnvironment() *Environment {
	env := &Environment{
		Parsers:        make(map[string]ParserGen),
		Processors:     make(map[string]ProcessorGen),
		DataCollectors: make(map[string]DataCollectorGen),
		Filters:        make(map[string]StringFilter),
	}

	env.RegisterKnownParsers()

	return env
}

// Add a generator for all of the parsers we know about.
func (env *Environment) RegisterKnownParsers() {
	env.RegisterParserGenerator(TAG_PRINTK,
		func() Parser {
			pk := NewPrintkParser()
			pk.ErrOnUnknownTag = false
			return pk
		})

	env.RegisterParserGenerator(TAG_TRACE,
		func() Parser {
			tparser := NewKernelTraceParser()
			tparser.ErrOnUnknownTag = false
			return tparser
		})

	env.RegisterParserGenerator(TAG_PL_POWER_BATTERY,
		NewPLPowerBatteryParser)

	env.RegisterParserGenerator(TAG_QOE_LIFECYCLE,
		NewQoEActivityLifecycleParser)
}

// Add a parser generator for a given log tag.
// By default, all known parsers are registered when the environment is created.
// Client code can register any custom parsers
func (env *Environment) RegisterParserGenerator(tag string, gen ParserGen) {
	env.Parsers[tag] = gen
}