1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
// Copyright 2020 New Relic Corporation. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
package internal
import (
"encoding/hex"
"math/rand"
"sync"
)
// TraceIDGenerator creates identifiers for distributed tracing.
type TraceIDGenerator struct {
sync.Mutex
rnd *rand.Rand
}
// NewTraceIDGenerator creates a new trace identifier generator.
func NewTraceIDGenerator(seed int64) *TraceIDGenerator {
return &TraceIDGenerator{
rnd: rand.New(rand.NewSource(seed)),
}
}
// Float32 returns a random float32 from its random source.
func (tg *TraceIDGenerator) Float32() float32 {
tg.Lock()
defer tg.Unlock()
return tg.rnd.Float32()
}
const (
traceIDByteLen = 16
// TraceIDHexStringLen is the length of the trace ID when represented
// as a hex string.
TraceIDHexStringLen = 32
spanIDByteLen = 8
maxIDByteLen = 16
)
// GenerateTraceID creates a new trace identifier, which is a 32 character hex string.
func (tg *TraceIDGenerator) GenerateTraceID() string {
return tg.generateID(traceIDByteLen)
}
// GenerateSpanID creates a new span identifier, which is a 16 character hex string.
func (tg *TraceIDGenerator) GenerateSpanID() string {
return tg.generateID(spanIDByteLen)
}
func (tg *TraceIDGenerator) generateID(len int) string {
var bits [maxIDByteLen]byte
tg.Lock()
defer tg.Unlock()
tg.rnd.Read(bits[:len])
return hex.EncodeToString(bits[:len])
}
|
