forked from mirrors/gotosocial
78 lines
2.4 KiB
Go
78 lines
2.4 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 trace // import "go.opentelemetry.io/otel/sdk/trace"
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
crand "crypto/rand"
|
||
|
|
"encoding/binary"
|
||
|
|
"math/rand"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"go.opentelemetry.io/otel/trace"
|
||
|
|
)
|
||
|
|
|
||
|
|
// IDGenerator allows custom generators for TraceID and SpanID.
|
||
|
|
type IDGenerator interface {
|
||
|
|
// DO NOT CHANGE: any modification will not be backwards compatible and
|
||
|
|
// must never be done outside of a new major release.
|
||
|
|
|
||
|
|
// NewIDs returns a new trace and span ID.
|
||
|
|
NewIDs(ctx context.Context) (trace.TraceID, trace.SpanID)
|
||
|
|
// DO NOT CHANGE: any modification will not be backwards compatible and
|
||
|
|
// must never be done outside of a new major release.
|
||
|
|
|
||
|
|
// NewSpanID returns a ID for a new span in the trace with traceID.
|
||
|
|
NewSpanID(ctx context.Context, traceID trace.TraceID) trace.SpanID
|
||
|
|
// DO NOT CHANGE: any modification will not be backwards compatible and
|
||
|
|
// must never be done outside of a new major release.
|
||
|
|
}
|
||
|
|
|
||
|
|
type randomIDGenerator struct {
|
||
|
|
sync.Mutex
|
||
|
|
randSource *rand.Rand
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ IDGenerator = &randomIDGenerator{}
|
||
|
|
|
||
|
|
// NewSpanID returns a non-zero span ID from a randomly-chosen sequence.
|
||
|
|
func (gen *randomIDGenerator) NewSpanID(ctx context.Context, traceID trace.TraceID) trace.SpanID {
|
||
|
|
gen.Lock()
|
||
|
|
defer gen.Unlock()
|
||
|
|
sid := trace.SpanID{}
|
||
|
|
_, _ = gen.randSource.Read(sid[:])
|
||
|
|
return sid
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewIDs returns a non-zero trace ID and a non-zero span ID from a
|
||
|
|
// randomly-chosen sequence.
|
||
|
|
func (gen *randomIDGenerator) NewIDs(ctx context.Context) (trace.TraceID, trace.SpanID) {
|
||
|
|
gen.Lock()
|
||
|
|
defer gen.Unlock()
|
||
|
|
tid := trace.TraceID{}
|
||
|
|
_, _ = gen.randSource.Read(tid[:])
|
||
|
|
sid := trace.SpanID{}
|
||
|
|
_, _ = gen.randSource.Read(sid[:])
|
||
|
|
return tid, sid
|
||
|
|
}
|
||
|
|
|
||
|
|
func defaultIDGenerator() IDGenerator {
|
||
|
|
gen := &randomIDGenerator{}
|
||
|
|
var rngSeed int64
|
||
|
|
_ = binary.Read(crand.Reader, binary.LittleEndian, &rngSeed)
|
||
|
|
gen.randSource = rand.New(rand.NewSource(rngSeed))
|
||
|
|
return gen
|
||
|
|
}
|