Skip to content

Commit 4af7297

Browse files
Separate V1/V2 custom events
1 parent 37a4462 commit 4af7297

File tree

4 files changed

+20
-6
lines changed

4 files changed

+20
-6
lines changed

dd-java-agent/appsec/src/test/groovy/com/datadog/appsec/user/AppSecEventTrackerSpecification.groovy

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ class AppSecEventTrackerSpecification extends DDSpecification {
143143
1 * traceSegment.setTagTop('_dd.p.ts', ProductTraceSource.ASM)
144144
0 * _
145145
146-
assertAppSecSdkEvent(CUSTOM, V2)
146+
assertAppSecSdkEvent(CUSTOM, V1)
147147
}
148148
149149
def 'test track login success event V2 (SDK)'() {
@@ -198,6 +198,8 @@ class AppSecEventTrackerSpecification extends DDSpecification {
198198
1 * traceSegment.setTagTop('asm.keep', true)
199199
1 * traceSegment.setTagTop('_dd.p.ts', ProductTraceSource.ASM)
200200
0 * _
201+
202+
assertAppSecSdkEvent(CUSTOM, V2)
201203
}
202204
203205
def 'test track user (SDK)'() {

dd-trace-api/src/main/java/datadog/appsec/api/login/EventTrackerService.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,13 @@ public void trackUserLoginFailure(
1515
final String login, final boolean exists, final Map<String, String> metadata) {}
1616

1717
@Override
18-
public void trackCustomEvent(final String eventName, final Map<String, String> metadata) {}
18+
public void trackCustomEventV2(
19+
final String eventName, final Map<String, String> metadata) {}
1920
};
2021

2122
void trackUserLoginSuccess(String login, String userId, Map<String, String> metadata);
2223

2324
void trackUserLoginFailure(String login, boolean exists, Map<String, String> metadata);
2425

25-
void trackCustomEvent(String eventName, Map<String, String> metadata);
26+
void trackCustomEventV2(String eventName, Map<String, String> metadata);
2627
}

dd-trace-api/src/main/java/datadog/appsec/api/login/EventTrackerV2.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,6 @@ public static void trackUserLoginFailure(
4747
* @param metadata custom metadata data represented as key/value map
4848
*/
4949
public static void trackCustomEvent(String eventName, Map<String, String> metadata) {
50-
SERVICE.trackCustomEvent(eventName, metadata);
50+
SERVICE.trackCustomEventV2(eventName, metadata);
5151
}
5252
}

internal-api/src/main/java/datadog/trace/api/appsec/AppSecEventTracker.java

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,18 @@ public final void trackLoginFailureEvent(
9292
}
9393
}
9494

95+
@SuppressWarnings("deprecation")
96+
@Override
97+
public final void trackCustomEvent(String eventName, Map<String, String> metadata) {
98+
if (eventName == null || eventName.isEmpty()) {
99+
throw new IllegalArgumentException("eventName is null or empty");
100+
}
101+
WafMetricCollector.get().appSecSdkEvent(CUSTOM, V1);
102+
if (handleLoginEvent(V1, eventName, SDK, null, null, null, metadata)) {
103+
throw new BlockingException("Blocked request (for custom event)");
104+
}
105+
}
106+
95107
@Override
96108
public void trackUserLoginSuccess(
97109
final String login, final String userId, final Map<String, String> metadata) {
@@ -116,9 +128,8 @@ public void trackUserLoginFailure(
116128
}
117129
}
118130

119-
@SuppressWarnings("deprecation")
120131
@Override
121-
public final void trackCustomEvent(String eventName, Map<String, String> metadata) {
132+
public final void trackCustomEventV2(final String eventName, final Map<String, String> metadata) {
122133
if (eventName == null || eventName.isEmpty()) {
123134
throw new IllegalArgumentException("eventName is null or empty");
124135
}

0 commit comments

Comments
 (0)