Add java endpoints to Test_RemoteConfigurationExtraServices (#2911)

Author: jandro996

manifests/java.yml

@@ -1162,7 +1162,9 @@ tests/:
       TestTracerFlareV1: v1.25.0
   remote_config/:
     test_remote_configuration.py:
-      Test_RemoteConfigurationExtraServices: missing_feature
+      Test_RemoteConfigurationExtraServices:
+          '*': v1.31.0
+          spring-boot-3-native: missing_feature (GraalVM. Tracing support only)
       Test_RemoteConfigurationUpdateSequenceASMDD:
         '*': v1.4.0
         spring-boot-3-native: missing_feature (GraalVM. Tracing support only)

utils/build/docker/java/akka-http/src/main/scala/com/datadoghq/akka_http/AppSecRoutes.scala

@@ -32,6 +32,14 @@ object AppSecRoutes {
           complete("Hello world!")
         }
       }
+    } ~
+    path("createextraservice") {
+      get {
+        parameter("serviceName") { serviceName =>
+          setRootSpanTag("service", serviceName)
+          complete("OK")
+        }
+      }
     } ~
       path("headers") {
         get {

utils/build/docker/java/jersey-grizzly2/src/main/java/com/datadoghq/jersey/MyResource.java

@@ -269,6 +269,13 @@ public Response setCookie(@QueryParam("name") String name, @QueryParam("value")
         return Response.ok().header("Set-Cookie", name + "=" + value).build();
     }
 
+    @GET
+    @Path("/createextraservice")
+    public String createextraservice(@QueryParam("serviceName") String serviceName) {
+        setRootSpanTag("service", serviceName);
+        return "ok";
+    }
+
     public static final class DistantCallResponse {
         public String url;
         public int status_code;

utils/build/docker/java/play/app/controllers/AppSecController.scala

@@ -162,6 +162,11 @@ class AppSecController @Inject()(cc: MessagesControllerComponents, ws: WSClient,
     Ok(Json.toJson(headers))
   }
 
+  def createextraservice(serviceName: String) = Action { request =>
+    setRootSpanTag("service", serviceName)
+    Results.Ok("ok")
+  }
+
   def setCookie(name: Option[String], value: Option[String]) = Action { request =>
     val cookieName = name.getOrElse("defaultName")
     val cookieValue = value.getOrElse("defaultValue")

utils/build/docker/java/play/conf/routes

@@ -16,3 +16,4 @@ GET  /rasp/sqli                controllers.RaspController.sqli
 POST /rasp/sqli                controllers.RaspController.sqli
 GET  /requestdownstream        controllers.AppSecController.requestdownstream
 GET  /returnheaders            controllers.AppSecController.returnheaders
+GET  /createextraservice        controllers.AppSecController.createextraservice(serviceName: String)

utils/build/docker/java/ratpack/src/main/java/com/datadoghq/ratpack/Main.java

@@ -217,6 +217,12 @@ public static void main(String[] args) throws Exception {
 
                                 ctx.getResponse().send("application/json", json);
                             })
+                            .get("createextraservice", ctx -> {
+                                MultiValueMap<String, String> qp = ctx.getRequest().getQueryParams();
+                                String serviceName = qp.get("serviceName");
+                                setRootSpanTag("service", serviceName);
+                                ctx.getResponse().send("ok");
+                            })
                             .get("set_cookie", ctx -> {
                                 final String name = ctx.getRequest().getQueryParams().get("name");
                                 final String value = ctx.getRequest().getQueryParams().get("value");

utils/build/docker/java/resteasy-netty3/src/main/java/com/datadoghq/resteasy/MyResource.java

@@ -268,6 +268,13 @@ public Response setCookie(@QueryParam("name") String name, @QueryParam("value")
         return Response.ok().header("Set-Cookie", name + "=" + value).build();
     }
 
+    @GET
+    @Path("/createextraservice")
+    public String createextraservice(@QueryParam("serviceName") String serviceName) {
+        setRootSpanTag("service", serviceName);
+        return "ok";
+    }
+
     public static final class DistantCallResponse {
         public String url;
         public int status_code;

utils/build/docker/java/spring-boot/src/main/java/com/datadoghq/system_tests/springboot/App.java

@@ -86,6 +86,9 @@
 import java.util.Map.Entry;
 import java.util.Set;
 
+import io.opentracing.Span;
+import io.opentracing.util.GlobalTracer;
+
 import static com.mongodb.client.model.Filters.eq;
 import static io.opentelemetry.api.trace.SpanKind.INTERNAL;
 import static io.opentelemetry.api.trace.StatusCode.ERROR;
@@ -736,6 +739,15 @@ DistantCallResponse make_distant_call(@RequestParam String url) throws Exception
         return result;
     }
 
+    @RequestMapping("/createextraservice")
+    public String createextraservice(@RequestParam String serviceName) {
+        final Span span = GlobalTracer.get().activeSpan();
+        if (span != null) {
+            span.setTag("service", serviceName);
+        }
+        return "OK";
+    }
+
     public static final class DistantCallResponse {
         public String url;
         public int status_code;

utils/build/docker/java/vertx3/src/main/java/com/datadoghq/vertx3/Main.java

@@ -201,6 +201,12 @@ public void onResponse(Call call, Response response) throws IOException {
                     String value = ctx.request().getParam("value");
                     ctx.response().putHeader("Set-Cookie", name + "=" + value).end("ok");
                 });
+        router.get("/createextraservice")
+                .handler(ctx -> {
+                    String serviceName = ctx.request().getParam("serviceName");
+                    setRootSpanTag("service", serviceName);
+                    ctx.response().end("ok");
+                });
 
         iastRouteProviders().forEach(provider -> provider.accept(router));
         raspRouteProviders().forEach(provider -> provider.accept(router));

utils/build/docker/java/vertx4/src/main/java/com/datadoghq/vertx4/Main.java

@@ -200,6 +200,12 @@ public void onResponse(Call call, Response response) throws IOException {
                     String value = ctx.request().getParam("value");
                     ctx.response().putHeader("Set-Cookie", name + "=" + value).end("ok");
                 });
+        router.get("/createextraservice")
+                .handler(ctx -> {
+                    String serviceName = ctx.request().getParam("serviceName");
+                    setRootSpanTag("service", serviceName);
+                    ctx.response().end("ok");
+                });
 
         iastRouteProviders().forEach(provider -> provider.accept(router));
         raspRouteProviders().forEach(provider -> provider.accept(router));