|
3 | 3 | // This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2017 Datadog, Inc. |
4 | 4 | // </copyright> |
5 | 5 |
|
| 6 | +using System; |
6 | 7 | using System.Collections.Generic; |
| 8 | +using System.Threading; |
7 | 9 | using Datadog.Trace.HttpOverStreams; |
8 | 10 | using Datadog.Trace.Util; |
9 | 11 |
|
10 | 12 | namespace Datadog.Trace.Telemetry |
11 | 13 | { |
12 | 14 | internal static class TelemetryHttpHeaderNames |
13 | 15 | { |
| 16 | + private static string _httpSerializedDefaultAgentHeaders; |
| 17 | + |
14 | 18 | /// <summary> |
15 | 19 | /// Gets the default agent headers in the format <c>Key: Value\r\n</c>. For use in HTTP headers. |
16 | | - /// Not a const because session headers depend on runtime values. |
| 20 | + /// Lazily initialized because session headers depend on runtime values. |
17 | 21 | /// </summary> |
18 | | - internal static string HttpSerializedDefaultAgentHeaders { get; } = BuildSerializedAgentHeaders(); |
| 22 | + internal static string HttpSerializedDefaultAgentHeaders => |
| 23 | + LazyInitializer.EnsureInitialized(ref _httpSerializedDefaultAgentHeaders, BuildSerializedAgentHeaders); |
19 | 24 |
|
20 | 25 | /// <summary> |
21 | 26 | /// Gets the default constant headers that should be added to any request to the agent |
22 | 27 | /// </summary> |
23 | 28 | internal static KeyValuePair<string, string>[] GetDefaultAgentHeaders() |
24 | 29 | { |
25 | | - var headers = new List<KeyValuePair<string, string>> |
26 | | - { |
27 | | - new(TelemetryConstants.ClientLibraryLanguageHeader, TracerConstants.Language), |
28 | | - new(TelemetryConstants.ClientLibraryVersionHeader, TracerConstants.AssemblyVersion), |
29 | | - new(HttpHeaderNames.TracingEnabled, "false"), // don't add automatic instrumentation to requests directed to the agent |
30 | | - new(TelemetryConstants.SessionIdHeader, RuntimeId.Get()), |
31 | | - }; |
| 30 | + var sessionId = RuntimeId.Get(); |
| 31 | + var rootSessionId = RuntimeId.GetRootSessionId(); |
| 32 | + var includeRoot = rootSessionId != sessionId; |
| 33 | + var headerCount = includeRoot ? 5 : 4; |
32 | 34 |
|
33 | | - AddRootSessionIdHeader(headers); |
| 35 | + var headers = new KeyValuePair<string, string>[headerCount]; |
| 36 | + headers[0] = new(TelemetryConstants.ClientLibraryLanguageHeader, TracerConstants.Language); |
| 37 | + headers[1] = new(TelemetryConstants.ClientLibraryVersionHeader, TracerConstants.AssemblyVersion); |
| 38 | + headers[2] = new(HttpHeaderNames.TracingEnabled, "false"); // don't add automatic instrumentation to requests directed to the agent |
| 39 | + headers[3] = new(TelemetryConstants.SessionIdHeader, sessionId); |
34 | 40 |
|
35 | | - return headers.ToArray(); |
| 41 | + if (includeRoot) |
| 42 | + { |
| 43 | + headers[4] = new(TelemetryConstants.RootSessionIdHeader, rootSessionId); |
| 44 | + } |
| 45 | + |
| 46 | + return headers; |
36 | 47 | } |
37 | 48 |
|
38 | 49 | /// <summary> |
39 | 50 | /// Gets the default constant headers that should be added to any request to the direct telemetry intake |
40 | 51 | /// </summary> |
41 | 52 | internal static KeyValuePair<string, string>[] GetDefaultIntakeHeaders(TelemetrySettings.AgentlessSettings settings) |
42 | 53 | { |
43 | | - var headers = new List<KeyValuePair<string, string>> |
44 | | - { |
45 | | - new(TelemetryConstants.ClientLibraryLanguageHeader, TracerConstants.Language), |
46 | | - new(TelemetryConstants.ClientLibraryVersionHeader, TracerConstants.AssemblyVersion), |
47 | | - new(HttpHeaderNames.TracingEnabled, "false"), |
48 | | - new(TelemetryConstants.ApiKeyHeader, settings.ApiKey), |
49 | | - new(TelemetryConstants.SessionIdHeader, RuntimeId.Get()), |
50 | | - }; |
| 54 | + var sessionId = RuntimeId.Get(); |
| 55 | + var rootSessionId = RuntimeId.GetRootSessionId(); |
| 56 | + var includeRoot = rootSessionId != sessionId; |
| 57 | + var baseCount = settings.Cloud is null ? 5 : 8; |
| 58 | + var headerCount = includeRoot ? baseCount + 1 : baseCount; |
| 59 | + |
| 60 | + var headers = new KeyValuePair<string, string>[headerCount]; |
| 61 | + headers[0] = new(TelemetryConstants.ClientLibraryLanguageHeader, TracerConstants.Language); |
| 62 | + headers[1] = new(TelemetryConstants.ClientLibraryVersionHeader, TracerConstants.AssemblyVersion); |
| 63 | + headers[2] = new(HttpHeaderNames.TracingEnabled, "false"); |
| 64 | + headers[3] = new(TelemetryConstants.ApiKeyHeader, settings.ApiKey); |
| 65 | + headers[4] = new(TelemetryConstants.SessionIdHeader, sessionId); |
51 | 66 |
|
| 67 | + var index = 5; |
52 | 68 | if (settings.Cloud is { } cloud) |
53 | 69 | { |
54 | | - headers.Add(new(TelemetryConstants.CloudProviderHeader, cloud.Provider)); |
55 | | - headers.Add(new(TelemetryConstants.CloudResourceTypeHeader, cloud.ResourceType)); |
56 | | - headers.Add(new(TelemetryConstants.CloudResourceIdentifierHeader, cloud.ResourceIdentifier)); |
| 70 | + headers[index++] = new(TelemetryConstants.CloudProviderHeader, cloud.Provider); |
| 71 | + headers[index++] = new(TelemetryConstants.CloudResourceTypeHeader, cloud.ResourceType); |
| 72 | + headers[index++] = new(TelemetryConstants.CloudResourceIdentifierHeader, cloud.ResourceIdentifier); |
57 | 73 | } |
58 | 74 |
|
59 | | - AddRootSessionIdHeader(headers); |
60 | | - |
61 | | - return headers.ToArray(); |
62 | | - } |
63 | | - |
64 | | - private static void AddRootSessionIdHeader(List<KeyValuePair<string, string>> headers) |
65 | | - { |
66 | | - var sessionId = RuntimeId.Get(); |
67 | | - var rootSessionId = RuntimeId.GetRootSessionId(); |
68 | | - if (rootSessionId != sessionId) |
| 75 | + if (includeRoot) |
69 | 76 | { |
70 | | - headers.Add(new(TelemetryConstants.RootSessionIdHeader, rootSessionId)); |
| 77 | + headers[index] = new(TelemetryConstants.RootSessionIdHeader, rootSessionId); |
71 | 78 | } |
| 79 | + |
| 80 | + return headers; |
72 | 81 | } |
73 | 82 |
|
74 | 83 | private static string BuildSerializedAgentHeaders() |
75 | 84 | { |
76 | | - var serialized = |
77 | | - $"{TelemetryConstants.ClientLibraryLanguageHeader}: {TracerConstants.Language}" + DatadogHttpValues.CrLf + |
78 | | - $"{TelemetryConstants.ClientLibraryVersionHeader}: {TracerConstants.AssemblyVersion}" + DatadogHttpValues.CrLf + |
79 | | - $"{HttpHeaderNames.TracingEnabled}: false" + DatadogHttpValues.CrLf + |
80 | | - $"{TelemetryConstants.SessionIdHeader}: {RuntimeId.Get()}" + DatadogHttpValues.CrLf; |
81 | | - |
82 | 85 | var sessionId = RuntimeId.Get(); |
83 | 86 | var rootSessionId = RuntimeId.GetRootSessionId(); |
| 87 | + |
| 88 | + var sb = StringBuilderCache.Acquire(); |
| 89 | + sb.Append(TelemetryConstants.ClientLibraryLanguageHeader).Append(": ").Append(TracerConstants.Language).Append(DatadogHttpValues.CrLf); |
| 90 | + sb.Append(TelemetryConstants.ClientLibraryVersionHeader).Append(": ").Append(TracerConstants.AssemblyVersion).Append(DatadogHttpValues.CrLf); |
| 91 | + sb.Append(HttpHeaderNames.TracingEnabled).Append(": false").Append(DatadogHttpValues.CrLf); |
| 92 | + sb.Append(TelemetryConstants.SessionIdHeader).Append(": ").Append(sessionId).Append(DatadogHttpValues.CrLf); |
| 93 | + |
84 | 94 | if (rootSessionId != sessionId) |
85 | 95 | { |
86 | | - serialized += $"{TelemetryConstants.RootSessionIdHeader}: {rootSessionId}" + DatadogHttpValues.CrLf; |
| 96 | + sb.Append(TelemetryConstants.RootSessionIdHeader).Append(": ").Append(rootSessionId).Append(DatadogHttpValues.CrLf); |
87 | 97 | } |
88 | 98 |
|
89 | | - return serialized; |
| 99 | + return StringBuilderCache.GetStringAndRelease(sb); |
90 | 100 | } |
91 | 101 | } |
92 | 102 | } |
0 commit comments