This repository was archived by the owner on Oct 3, 2023. It is now read-only.
File tree 1 file changed +11
-2
lines changed
packages/opencensus-propagation-stackdriver/src
1 file changed +11
-2
lines changed Original file line number Diff line number Diff line change @@ -42,13 +42,22 @@ const TRACE_TRUE = 0x1;
42
42
43
43
/** Propagates span context through Stackdriver Format propagation. */
44
44
export class StackdriverFormat implements Propagation {
45
+ headerName : string ;
46
+
47
+ constructor ( config ?: { headerName ?: string } ) {
48
+ if ( config && config . headerName ) {
49
+ this . headerName = config . headerName ;
50
+ } else {
51
+ this . headerName = TRACE_CONTEXT_HEADER_NAME ;
52
+ }
53
+ }
45
54
/**
46
55
* Gets the span context from a request headers. If there is no span context
47
56
* in the headers, null is returned.
48
57
* @param getter
49
58
*/
50
59
extract ( getter : HeaderGetter ) : SpanContext | null {
51
- const traceContextHeader = getter . getHeader ( TRACE_CONTEXT_HEADER_NAME ) ;
60
+ const traceContextHeader = getter . getHeader ( this . headerName ) ;
52
61
if ( typeof traceContextHeader !== 'string' ) {
53
62
return null ;
54
63
}
@@ -83,7 +92,7 @@ export class StackdriverFormat implements Propagation {
83
92
header += `;o=${ spanContext . options } ` ;
84
93
}
85
94
86
- setter . setHeader ( TRACE_CONTEXT_HEADER_NAME , header ) ;
95
+ setter . setHeader ( this . headerName , header ) ;
87
96
}
88
97
89
98
/** Generate SpanContexts */
You can’t perform that action at this time.
0 commit comments