1
0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2024-12-04 09:43:23 +02:00
opentelemetry-go/propagation/binary_propagator.go
2019-10-02 13:54:39 -07:00

76 lines
2.2 KiB
Go

// Copyright 2019, 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 propagation
import (
"encoding/binary"
"go.opentelemetry.io/api/core"
apipropagation "go.opentelemetry.io/api/propagation"
)
type binaryPropagator struct{}
var _ apipropagation.BinaryFormatPropagator = binaryPropagator{}
// BinaryPropagator creates a new propagator. The propagator implements
// ToBytes and FromBytes method to transform SpanContext to/from byte array.
func BinaryPropagator() binaryPropagator {
return binaryPropagator{}
}
// ToBytes implements ToBytes method of propagation.BinaryFormatPropagator.
// It serializes core.SpanContext into a byte array.
func (bp binaryPropagator) ToBytes(sc core.SpanContext) []byte {
if sc == core.EmptySpanContext() {
return nil
}
var b [29]byte
binary.BigEndian.PutUint64(b[2:10], sc.TraceID.High)
binary.BigEndian.PutUint64(b[10:18], sc.TraceID.Low)
b[18] = 1
binary.BigEndian.PutUint64(b[19:27], sc.SpanID)
b[27] = 2
b[28] = sc.TraceFlags
return b[:]
}
// FromBytes implements FromBytes method of propagation.BinaryFormatPropagator.
// It de-serializes bytes into core.SpanContext.
func (bp binaryPropagator) FromBytes(b []byte) (sc core.SpanContext) {
if len(b) == 0 {
return core.EmptySpanContext()
}
b = b[1:]
if len(b) >= 17 && b[0] == 0 {
sc.TraceID.High = binary.BigEndian.Uint64(b[1:9])
sc.TraceID.Low = binary.BigEndian.Uint64(b[9:17])
b = b[17:]
} else {
return core.EmptySpanContext()
}
if len(b) >= 9 && b[0] == 1 {
sc.SpanID = binary.BigEndian.Uint64(b[1:9])
b = b[9:]
}
if len(b) >= 2 && b[0] == 2 {
sc.TraceFlags = b[1]
}
if sc.IsValid() {
return sc
}
return core.EmptySpanContext()
}