mirror of
https://github.com/open-telemetry/opentelemetry-go.git
synced 2024-12-12 10:04:29 +02:00
fefdf59a0b
* api/metric changes from jmacd:jmacd/batch_obs_2 * Add an SDK test * Use a single collector method * Two fixes * Comments; embed AsyncRunner * Comments * Comment fix * More comments * Renaming for clarity * Renaming for clarity (fix) * Lint
216 lines
4.9 KiB
Go
216 lines
4.9 KiB
Go
// Copyright The 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 metric
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"go.opentelemetry.io/otel/api/kv"
|
|
"go.opentelemetry.io/otel/api/metric"
|
|
apimetric "go.opentelemetry.io/otel/api/metric"
|
|
"go.opentelemetry.io/otel/api/metric/registry"
|
|
)
|
|
|
|
type (
|
|
Handle struct {
|
|
Instrument *Sync
|
|
Labels []kv.KeyValue
|
|
}
|
|
|
|
Batch struct {
|
|
// Measurement needs to be aligned for 64-bit atomic operations.
|
|
Measurements []Measurement
|
|
Ctx context.Context
|
|
Labels []kv.KeyValue
|
|
LibraryName string
|
|
}
|
|
|
|
MeterProvider struct {
|
|
lock sync.Mutex
|
|
impl *MeterImpl
|
|
unique metric.MeterImpl
|
|
registered map[string]apimetric.Meter
|
|
}
|
|
|
|
MeterImpl struct {
|
|
lock sync.Mutex
|
|
|
|
MeasurementBatches []Batch
|
|
|
|
asyncInstruments *AsyncInstrumentState
|
|
}
|
|
|
|
Measurement struct {
|
|
// Number needs to be aligned for 64-bit atomic operations.
|
|
Number apimetric.Number
|
|
Instrument apimetric.InstrumentImpl
|
|
}
|
|
|
|
Instrument struct {
|
|
meter *MeterImpl
|
|
descriptor apimetric.Descriptor
|
|
}
|
|
|
|
Async struct {
|
|
Instrument
|
|
|
|
runner apimetric.AsyncRunner
|
|
}
|
|
|
|
Sync struct {
|
|
Instrument
|
|
}
|
|
)
|
|
|
|
var (
|
|
_ apimetric.SyncImpl = &Sync{}
|
|
_ apimetric.BoundSyncImpl = &Handle{}
|
|
_ apimetric.MeterImpl = &MeterImpl{}
|
|
_ apimetric.AsyncImpl = &Async{}
|
|
)
|
|
|
|
func (i Instrument) Descriptor() apimetric.Descriptor {
|
|
return i.descriptor
|
|
}
|
|
|
|
func (a *Async) Implementation() interface{} {
|
|
return a
|
|
}
|
|
|
|
func (s *Sync) Implementation() interface{} {
|
|
return s
|
|
}
|
|
|
|
func (s *Sync) Bind(labels []kv.KeyValue) apimetric.BoundSyncImpl {
|
|
return &Handle{
|
|
Instrument: s,
|
|
Labels: labels,
|
|
}
|
|
}
|
|
|
|
func (s *Sync) RecordOne(ctx context.Context, number apimetric.Number, labels []kv.KeyValue) {
|
|
s.meter.doRecordSingle(ctx, labels, s, number)
|
|
}
|
|
|
|
func (h *Handle) RecordOne(ctx context.Context, number apimetric.Number) {
|
|
h.Instrument.meter.doRecordSingle(ctx, h.Labels, h.Instrument, number)
|
|
}
|
|
|
|
func (h *Handle) Unbind() {
|
|
}
|
|
|
|
func (m *MeterImpl) doRecordSingle(ctx context.Context, labels []kv.KeyValue, instrument apimetric.InstrumentImpl, number apimetric.Number) {
|
|
m.collect(ctx, labels, []Measurement{{
|
|
Instrument: instrument,
|
|
Number: number,
|
|
}})
|
|
}
|
|
|
|
func NewProvider() (*MeterImpl, apimetric.Provider) {
|
|
impl := &MeterImpl{
|
|
asyncInstruments: NewAsyncInstrumentState(nil),
|
|
}
|
|
p := &MeterProvider{
|
|
impl: impl,
|
|
unique: registry.NewUniqueInstrumentMeterImpl(impl),
|
|
registered: map[string]apimetric.Meter{},
|
|
}
|
|
return impl, p
|
|
}
|
|
|
|
func (p *MeterProvider) Meter(name string) apimetric.Meter {
|
|
p.lock.Lock()
|
|
defer p.lock.Unlock()
|
|
|
|
if lookup, ok := p.registered[name]; ok {
|
|
return lookup
|
|
}
|
|
m := apimetric.WrapMeterImpl(p.unique, name)
|
|
p.registered[name] = m
|
|
return m
|
|
}
|
|
|
|
func NewMeter() (*MeterImpl, apimetric.Meter) {
|
|
impl, p := NewProvider()
|
|
return impl, p.Meter("mock")
|
|
}
|
|
|
|
func (m *MeterImpl) NewSyncInstrument(descriptor metric.Descriptor) (apimetric.SyncImpl, error) {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
|
|
return &Sync{
|
|
Instrument{
|
|
descriptor: descriptor,
|
|
meter: m,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func (m *MeterImpl) NewAsyncInstrument(descriptor metric.Descriptor, runner metric.AsyncRunner) (apimetric.AsyncImpl, error) {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
|
|
a := &Async{
|
|
Instrument: Instrument{
|
|
descriptor: descriptor,
|
|
meter: m,
|
|
},
|
|
runner: runner,
|
|
}
|
|
m.asyncInstruments.Register(a, runner)
|
|
return a, nil
|
|
}
|
|
|
|
func (m *MeterImpl) RecordBatch(ctx context.Context, labels []kv.KeyValue, measurements ...apimetric.Measurement) {
|
|
mm := make([]Measurement, len(measurements))
|
|
for i := 0; i < len(measurements); i++ {
|
|
m := measurements[i]
|
|
mm[i] = Measurement{
|
|
Instrument: m.SyncImpl().Implementation().(*Sync),
|
|
Number: m.Number(),
|
|
}
|
|
}
|
|
m.collect(ctx, labels, mm)
|
|
}
|
|
|
|
func (m *MeterImpl) CollectAsync(labels []kv.KeyValue, obs ...metric.Observation) {
|
|
mm := make([]Measurement, len(obs))
|
|
for i := 0; i < len(obs); i++ {
|
|
o := obs[i]
|
|
mm[i] = Measurement{
|
|
Instrument: o.AsyncImpl(),
|
|
Number: o.Number(),
|
|
}
|
|
}
|
|
m.collect(context.Background(), labels, mm)
|
|
}
|
|
|
|
func (m *MeterImpl) collect(ctx context.Context, labels []kv.KeyValue, measurements []Measurement) {
|
|
m.lock.Lock()
|
|
defer m.lock.Unlock()
|
|
|
|
m.MeasurementBatches = append(m.MeasurementBatches, Batch{
|
|
Ctx: ctx,
|
|
Labels: labels,
|
|
Measurements: measurements,
|
|
})
|
|
}
|
|
|
|
func (m *MeterImpl) RunAsyncInstruments() {
|
|
m.asyncInstruments.Run(m)
|
|
}
|