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
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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