Skip to content

Adding wildcard feature for DD_TRACE_HEADER_TAGS and enabling for Http Response headers #9067

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mhlidd merged 6 commits into from
Jul 3, 2025
Merged

Adding wildcard feature for DD_TRACE_HEADER_TAGS and enabling for Http Response headers #9067

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
2561072
Saving branch
mhlidd Jun 27, 2025
0f8e746
setting up structure and adding initial unit test for header tags
mhlidd Jun 30, 2025
c56858f
fixing and adding unit tests
mhlidd Jul 1, 2025
b6a465b
updating unit tests
mhlidd Jul 1, 2025
13d420f
responding PR comments
mhlidd Jul 2, 2025
61ad64f
Merge branch 'master' into mhlidd/update_trace_header_tags
mhlidd Jul 2, 2025
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
Next Next commit
Saving branch
  • Loading branch information
@mhlidd
mhlidd committed Jun 27, 2025
commit 256107213f447ab92b8828b4f2550403df704316
9 changes: 9 additions & 0 deletions .../src/main/java/datadog/trace/bootstrap/instrumentation/decorator/HttpServerDecorator.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -642,14 +642,23 @@ static ResponseHeaderTagClassifier create(AgentSpan span, Map<String, String> he

private final AgentSpan span;
private final Map<String, String> headerTags;
private final String wildcardHeaderPrefix;

public ResponseHeaderTagClassifier(AgentSpan span, Map<String, String> headerTags) {
this.span = span;
this.headerTags = headerTags;
if(headerTags.size() > 0)
System.out.println("responseHeaderTags Post-Processing\nsize: " + headerTags.size() + "; " + headerTags.keySet().toArray()[0] + " : " + headerTags.get(headerTags.keySet().toArray()[0]));
this.wildcardHeaderPrefix = this.headerTags.getOrDefault("*", null);
Comment thread
mhlidd marked this conversation as resolved.
Outdated
}

@Override
public boolean accept(String key, String value) {
System.out.println("key: " + key);
if (wildcardHeaderPrefix != null) {
System.out.println("wildcardKey: " + wildcardHeaderPrefix + key);
span.setTag(wildcardHeaderPrefix + key, value);
}
String mappedKey = headerTags.get(key.toLowerCase(Locale.ROOT));
if (mappedKey != null) {
span.setTag(mappedKey, value);
Expand Down
4 changes: 4 additions & 0 deletions internal-api/src/main/java/datadog/trace/api/Config.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -833,12 +833,16 @@ private Config(final ConfigProvider configProvider, final InstrumenterConfig ins
logIgnoredSettingWarning(RESPONSE_HEADER_TAGS, HEADER_TAGS, ".legacy.parsing.enabled");
}
} else {
System.out.println("requestHeaderTags: ");
requestHeaderTags =
configProvider.getMergedMapWithOptionalMappings(
"http.request.headers.", true, HEADER_TAGS, REQUEST_HEADER_TAGS);
System.out.println("responseHeaderTags: ");
responseHeaderTags =
configProvider.getMergedMapWithOptionalMappings(
"http.response.headers.", true, HEADER_TAGS, RESPONSE_HEADER_TAGS);
if(responseHeaderTags.size() > 0)
System.out.println("responseHeaderTags Post-Processing\nsize: " + responseHeaderTags.size() + "; " + responseHeaderTags.keySet().toArray()[0] + " : " + responseHeaderTags.get(responseHeaderTags.keySet().toArray()[0]));
}
requestHeaderTagsCommaAllowed =
configProvider.getBoolean(REQUEST_HEADER_TAGS_COMMA_ALLOWED, true);
Expand Down
6 changes: 6 additions & 0 deletions internal-api/src/main/java/datadog/trace/bootstrap/config/provider/ConfigConverter.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -312,14 +312,20 @@ private static void loadMapWithOptionalMapping(
String value;
if (delimiter == mapPos) {
value = trimmedHeader(str, delimiter + 1, end, false);
System.out.println("delimiter==mapPos value: " + value);
// tags must start with a letter
if (!value.isEmpty() && !Character.isLetter(value.charAt(0))) {
throw new BadFormatException(
"Illegal tag starting with non letter for key '" + key + "'");
}
} else {
if(key.charAt(0) == '*'){
map.put(key, defaultPrefix);
return;
Comment thread
mcculls marked this conversation as resolved.
}
if (Character.isLetter(key.charAt(0))) {
value = defaultPrefix + Strings.normalizedHeaderTag(key);
System.out.println("else value: " + value);
} else {
// tags must start with a letter
throw new BadFormatException(
Expand Down
1 change: 1 addition & 0 deletions internal-api/src/main/java/datadog/trace/bootstrap/config/provider/ConfigProvider.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -318,6 +318,7 @@ public Map<String, String> getMergedMapWithOptionalMappings(
for (String key : keys) {
for (int i = sources.length - 1; 0 <= i; i--) {
String value = sources[i].get(key);
// System.out.println("value: " + value);
Map<String, String> parsedMap =
ConfigConverter.parseMapWithOptionalMappings(value, key, defaultPrefix, lowercaseKeys);
if (!parsedMap.isEmpty()) {
Expand Down
3 changes: 3 additions & 0 deletions internal-api/src/test/groovy/datadog/trace/api/ConfigTest.groovy
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -165,6 +165,7 @@ class ConfigTest extends DDSpecification {
private static final DD_JMXFETCH_METRICS_CONFIGS_ENV = "DD_JMXFETCH_METRICS_CONFIGS"
private static final DD_TRACE_AGENT_PORT_ENV = "DD_TRACE_AGENT_PORT"
private static final DD_AGENT_PORT_LEGACY_ENV = "DD_AGENT_PORT"
private static final DD_TRACE_HEADER_TAGS = "DD_TRACE_HEADER_TAGS"
private static final DD_TRACE_REPORT_HOSTNAME = "DD_TRACE_REPORT_HOSTNAME"
private static final DD_RUNTIME_METRICS_ENABLED_ENV = "DD_RUNTIME_METRICS_ENABLED"
private static final DD_TRACE_LONG_RUNNING_ENABLED = "DD_TRACE_EXPERIMENTAL_LONG_RUNNING_ENABLED"
Expand Down Expand Up @@ -568,6 +569,7 @@ class ConfigTest extends DDSpecification {
environmentVariables.set(DD_TRACE_X_DATADOG_TAGS_MAX_LENGTH, "42")
environmentVariables.set(DD_TRACE_LONG_RUNNING_ENABLED, "true")
environmentVariables.set(DD_TRACE_LONG_RUNNING_FLUSH_INTERVAL, "81")
environmentVariables.set(DD_TRACE_HEADER_TAGS, "*")

when:
def config = new Config()
Expand All @@ -587,6 +589,7 @@ class ConfigTest extends DDSpecification {
config.xDatadogTagsMaxLength == 42
config.isLongRunningTraceEnabled()
config.getLongRunningTraceFlushInterval() == 81
config.responseHeaderTags == ["*":"http.response.headers."]
}

def "sys props override env vars"() {
Expand Down