blob: 0f330e3a08a0ea378091d4326b5a4daa34eaceb7 [file] [log] [blame]
Joey Armstronga6af1522023-01-17 16:06:16 -05001// Copyright The OpenTelemetry Authors
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package trace
16
17import (
18 "context"
19)
20
21type traceContextKeyType int
22
23const (
24 currentSpanKey traceContextKeyType = iota
25 remoteContextKey
26)
27
28// ContextWithSpan creates a new context with a current span set to
29// the passed span.
30func ContextWithSpan(ctx context.Context, span Span) context.Context {
31 return context.WithValue(ctx, currentSpanKey, span)
32}
33
34// SpanFromContext returns the current span stored in the context.
35func SpanFromContext(ctx context.Context) Span {
36 if span, has := ctx.Value(currentSpanKey).(Span); has {
37 return span
38 }
39 return noopSpan{}
40}
41
42// ContextWithRemoteSpanContext creates a new context with a remote
43// span context set to the passed span context.
44func ContextWithRemoteSpanContext(ctx context.Context, sc SpanContext) context.Context {
45 return context.WithValue(ctx, remoteContextKey, sc)
46}
47
48// RemoteSpanContextFromContext returns the remote span context stored
49// in the context.
50func RemoteSpanContextFromContext(ctx context.Context) SpanContext {
51 if sc, ok := ctx.Value(remoteContextKey).(SpanContext); ok {
52 return sc
53 }
54 return EmptySpanContext()
55}