mirror of
https://github.com/golang/go
synced 2024-11-19 04:44:41 -07:00
5b08f89bfc
this shuffles things so there a single exporter API rather than an observer It also removes most of the globals. per telemetry type. Change-Id: Iaa82abe2ded1fff9df8e067ed4a55bcbd9d9591f Reviewed-on: https://go-review.googlesource.com/c/tools/+/190405 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Emmanuel Odeke <emm.odeke@gmail.com>
72 lines
1.4 KiB
Go
72 lines
1.4 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 telemetry
|
|
|
|
import (
|
|
crand "crypto/rand"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"math/rand"
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
type TraceID [16]byte
|
|
type SpanID [8]byte
|
|
|
|
func (t TraceID) String() string {
|
|
return fmt.Sprintf("%02x", t[:])
|
|
}
|
|
|
|
func (s SpanID) String() string {
|
|
return fmt.Sprintf("%02x", s[:])
|
|
}
|
|
|
|
func (s SpanID) IsValid() bool {
|
|
return s != SpanID{}
|
|
}
|
|
|
|
var (
|
|
generationMu sync.Mutex
|
|
nextSpanID uint64
|
|
spanIDInc uint64
|
|
|
|
traceIDAdd [2]uint64
|
|
traceIDRand *rand.Rand
|
|
)
|
|
|
|
func initGenerator() {
|
|
var rngSeed int64
|
|
for _, p := range []interface{}{
|
|
&rngSeed, &traceIDAdd, &nextSpanID, &spanIDInc,
|
|
} {
|
|
binary.Read(crand.Reader, binary.LittleEndian, p)
|
|
}
|
|
traceIDRand = rand.New(rand.NewSource(rngSeed))
|
|
spanIDInc |= 1
|
|
}
|
|
|
|
func NewTraceID() TraceID {
|
|
generationMu.Lock()
|
|
defer generationMu.Unlock()
|
|
if traceIDRand == nil {
|
|
initGenerator()
|
|
}
|
|
var tid [16]byte
|
|
binary.LittleEndian.PutUint64(tid[0:8], traceIDRand.Uint64()+traceIDAdd[0])
|
|
binary.LittleEndian.PutUint64(tid[8:16], traceIDRand.Uint64()+traceIDAdd[1])
|
|
return tid
|
|
}
|
|
|
|
func NewSpanID() SpanID {
|
|
var id uint64
|
|
for id == 0 {
|
|
id = atomic.AddUint64(&nextSpanID, spanIDInc)
|
|
}
|
|
var sid [8]byte
|
|
binary.LittleEndian.PutUint64(sid[:], id)
|
|
return sid
|
|
}
|