mirror of
https://github.com/golang/go
synced 2024-09-30 16:18:35 -06:00
a90b7300be
Instead of tagging events with their type, instead we infer the type from the label pattern. The standard event creators all have a matching test that returns true if the the labels pattern matches the ones that would be built by the creator. Spans and logs already have a unique label pattern, other event types required a special label marker. This makes the system much more extensible, and also cleans up some the API. Change-Id: I1fbc9ec07aa84ead6c12bbd5ca65b13b605bfa4a Reviewed-on: https://go-review.googlesource.com/c/tools/+/229242 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Emmanuel Odeke <emm.odeke@gmail.com>
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
// Copyright 2019 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package metric aggregates events into metrics that can be exported.
|
|
package metric
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
|
|
"golang.org/x/tools/internal/event"
|
|
"golang.org/x/tools/internal/event/core"
|
|
"golang.org/x/tools/internal/event/keys"
|
|
"golang.org/x/tools/internal/event/label"
|
|
)
|
|
|
|
var Entries = keys.New("metric_entries", "The set of metrics calculated for an event")
|
|
|
|
type Config struct {
|
|
subscribers map[interface{}][]subscriber
|
|
}
|
|
|
|
type subscriber func(time.Time, label.Map, label.Label) Data
|
|
|
|
func (e *Config) subscribe(key label.Key, s subscriber) {
|
|
if e.subscribers == nil {
|
|
e.subscribers = make(map[interface{}][]subscriber)
|
|
}
|
|
e.subscribers[key] = append(e.subscribers[key], s)
|
|
}
|
|
|
|
func (e *Config) Exporter(output event.Exporter) event.Exporter {
|
|
var mu sync.Mutex
|
|
return func(ctx context.Context, ev core.Event, lm label.Map) context.Context {
|
|
if !event.IsMetric(ev) {
|
|
return output(ctx, ev, lm)
|
|
}
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
var metrics []Data
|
|
for index := 0; ev.Valid(index); index++ {
|
|
l := ev.Label(index)
|
|
if !l.Valid() {
|
|
continue
|
|
}
|
|
id := l.Key()
|
|
if list := e.subscribers[id]; len(list) > 0 {
|
|
for _, s := range list {
|
|
metrics = append(metrics, s(ev.At(), lm, l))
|
|
}
|
|
}
|
|
}
|
|
lm = label.MergeMaps(label.NewMap(Entries.Of(metrics)), lm)
|
|
return output(ctx, ev, lm)
|
|
}
|
|
}
|