Skip to content
This repository has been archived by the owner on Oct 3, 2023. It is now read-only.

StackDriver Propagator: Make the google header configurable #818

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

StackDriver Propagator: Make the google header configurable #818

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
13 changes: 11 additions & 2 deletions packages/opencensus-propagation-stackdriver/src/stackdriver-format.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,13 +42,22 @@ const TRACE_TRUE = 0x1;

/** Propagates span context through Stackdriver Format propagation. */
export class StackdriverFormat implements Propagation {
headerName: string;

constructor(config?: { headerName?: string }) {
if (config && config.headerName) {
this.headerName = config.headerName;
} else {
this.headerName = TRACE_CONTEXT_HEADER_NAME;
}
}
/**
* Gets the span context from a request headers. If there is no span context
* in the headers, null is returned.
* @param getter
*/
extract(getter: HeaderGetter): SpanContext | null {
const traceContextHeader = getter.getHeader(TRACE_CONTEXT_HEADER_NAME);
const traceContextHeader = getter.getHeader(this.headerName);
if (typeof traceContextHeader !== 'string') {
return null;
}
Expand Down Expand Up @@ -83,7 +92,7 @@ export class StackdriverFormat implements Propagation {
header += `;o=${spanContext.options}`;
}

setter.setHeader(TRACE_CONTEXT_HEADER_NAME, header);
setter.setHeader(this.headerName, header);
}

/** Generate SpanContexts */
Expand Down