|
1 | 1 | package datadog.trace.core.propagation;
|
2 | 2 |
|
3 |
| -import datadog.trace.api.Config; |
4 |
| -import datadog.trace.api.TracePropagationStyle; |
5 | 3 | import datadog.trace.bootstrap.instrumentation.api.AgentPropagation;
|
6 | 4 | import datadog.trace.bootstrap.instrumentation.api.AgentSpan;
|
7 | 5 | import datadog.trace.bootstrap.instrumentation.api.AgentSpanContext;
|
8 |
| -import datadog.trace.core.DDSpanContext; |
9 | 6 | import datadog.trace.core.datastreams.DataStreamContextInjector;
|
10 | 7 | import java.util.LinkedHashMap;
|
11 |
| -import java.util.Map; |
12 | 8 |
|
13 | 9 | public class CorePropagation implements AgentPropagation {
|
14 |
| - private final HttpCodec.Injector injector; |
15 |
| - private final Map<TracePropagationStyle, HttpCodec.Injector> injectors; |
16 | 10 | private final DataStreamContextInjector dataStreamContextInjector;
|
17 | 11 | private final HttpCodec.Extractor extractor;
|
18 | 12 |
|
19 | 13 | /**
|
20 | 14 | * Constructor
|
21 | 15 | *
|
22 | 16 | * @param extractor The context extractor.
|
23 |
| - * @param defaultInjector The default injector when no {@link TracePropagationStyle} given. |
24 |
| - * @param injectors All the other injectors available for context injection. |
25 | 17 | * @param dataStreamContextInjector The DSM context injector, as a specific object until generic
|
26 | 18 | * context injection is available.
|
27 | 19 | */
|
28 | 20 | public CorePropagation(
|
29 |
| - HttpCodec.Extractor extractor, |
30 |
| - HttpCodec.Injector defaultInjector, |
31 |
| - Map<TracePropagationStyle, HttpCodec.Injector> injectors, |
32 |
| - DataStreamContextInjector dataStreamContextInjector) { |
| 21 | + HttpCodec.Extractor extractor, DataStreamContextInjector dataStreamContextInjector) { |
33 | 22 | this.extractor = extractor;
|
34 |
| - this.injector = defaultInjector; |
35 |
| - this.injectors = injectors; |
36 | 23 | this.dataStreamContextInjector = dataStreamContextInjector;
|
37 | 24 | }
|
38 | 25 |
|
39 |
| - @Override |
40 |
| - public <C> void inject(final AgentSpan span, final C carrier, final Setter<C> setter) { |
41 |
| - inject(span.context(), carrier, setter, null); |
42 |
| - } |
43 |
| - |
44 |
| - @Override |
45 |
| - public <C> void inject(AgentSpanContext context, C carrier, Setter<C> setter) { |
46 |
| - inject(context, carrier, setter, null); |
47 |
| - } |
48 |
| - |
49 |
| - @Override |
50 |
| - public <C> void inject(AgentSpan span, C carrier, Setter<C> setter, TracePropagationStyle style) { |
51 |
| - inject(span.context(), carrier, setter, style); |
52 |
| - } |
53 |
| - |
54 |
| - private <C> void inject( |
55 |
| - AgentSpanContext context, C carrier, Setter<C> setter, TracePropagationStyle style) { |
56 |
| - if (!(context instanceof DDSpanContext)) { |
57 |
| - return; |
58 |
| - } |
59 |
| - |
60 |
| - final DDSpanContext ddSpanContext = (DDSpanContext) context; |
61 |
| - ddSpanContext.getTraceCollector().setSamplingPriorityIfNecessary(); |
62 |
| - |
63 |
| - /** |
64 |
| - * If the experimental appsec standalone feature is enabled and appsec propagation is disabled |
65 |
| - * (no ASM events), stop propagation |
66 |
| - */ |
67 |
| - if (Config.get().isAppSecStandaloneEnabled() |
68 |
| - && !ddSpanContext.getPropagationTags().isAppsecPropagationEnabled()) { |
69 |
| - return; |
70 |
| - } |
71 |
| - |
72 |
| - if (null == style) { |
73 |
| - injector.inject(ddSpanContext, carrier, setter); |
74 |
| - } else { |
75 |
| - injectors.get(style).inject(ddSpanContext, carrier, setter); |
76 |
| - } |
77 |
| - } |
78 |
| - |
79 | 26 | @Override
|
80 | 27 | public <C> void injectPathwayContext(
|
81 | 28 | AgentSpan span, C carrier, Setter<C> setter, LinkedHashMap<String, String> sortedTags) {
|
|
0 commit comments