opentelemetry-go/experimental/streaming/sdk/internal/test_observer.go

// Copyright 2019, OpenTelemetry Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package internal // import "go.opentelemetry.io/experimental/streaming/sdk/internal"

import "go.opentelemetry.io/experimental/streaming/exporter"

type eventsMap map[exporter.EventType][]exporter.Event

type TestObserver struct {
	events eventsMap
}

var _ exporter.Observer = &TestObserver{}

func NewTestObserver() *TestObserver {
	return &TestObserver{}
}

func (o *TestObserver) Observe(e exporter.Event) {
	if o.events == nil {
		o.events = make(eventsMap)
	}
	o.events[e.Type] = append(o.events[e.Type], e)
}

func (o *TestObserver) Events(eType exporter.EventType) []exporter.Event {
	if o.events == nil {
		return nil
	}
	return o.events[eType]
}