feat(java-concurrent): Refactor smoke tests

Author: PerfectSlayer

dd-smoke-tests/concurrent/java-21/src/main/java/datadog/smoketest/concurrent/VirtualThreadExecuteCalculator.java

@@ -20,6 +20,11 @@ public long computeFibonacci(int n) throws ExecutionException, InterruptedExcept
     return task.result.get();
   }
 
+  @Override
+  public void close() {
+    this.executor.shutdown();
+  }
+
   public class FibonacciExecuteTask implements Runnable {
     private final long n;
     private final CompletableFuture<Long> result;
@@ -46,9 +51,4 @@ public void run() {
       }
     }
   }
-
-  @Override
-  public void close() {
-    this.executor.shutdown();
-  }
 }

dd-smoke-tests/concurrent/java-21/src/main/java/datadog/smoketest/concurrent/VirtualThreadInvokeAllCalculator.java

@@ -23,6 +23,11 @@ public long computeFibonacci(int n) throws ExecutionException, InterruptedExcept
     return this.executor.invokeAll(of(task)).getFirst().get();
   }
 
+  @Override
+  public void close() {
+    this.executor.shutdown();
+  }
+
   public class FibonacciSubmitTask implements Callable<Long> {
     private final long n;
 
@@ -41,9 +46,4 @@ public Long call() throws ExecutionException, InterruptedException {
       return futures.getFirst().get() + futures.getLast().get();
     }
   }
-
-  @Override
-  public void close() {
-    this.executor.shutdown();
-  }
 }

dd-smoke-tests/concurrent/java-21/src/main/java/datadog/smoketest/concurrent/VirtualThreadInvokeAnyCalculator.java

@@ -21,6 +21,11 @@ public long computeFibonacci(int n) throws ExecutionException, InterruptedExcept
     return this.executor.invokeAny(of(task));
   }
 
+  @Override
+  public void close() {
+    this.executor.shutdown();
+  }
+
   public class FibonacciSubmitTask implements Callable<Long> {
     private final long n;
 
@@ -38,9 +43,4 @@ public Long call() throws ExecutionException, InterruptedException {
       return executor.invokeAny(of(task1)) + executor.invokeAny(of(task2));
     }
   }
-
-  @Override
-  public void close() {
-    this.executor.shutdown();
-  }
 }

dd-smoke-tests/concurrent/java-21/src/main/java/datadog/smoketest/concurrent/VirtualThreadSubmitCallableCalculator.java

@@ -20,6 +20,11 @@ public long computeFibonacci(int n) throws ExecutionException, InterruptedExcept
     return this.executor.submit(task).get();
   }
 
+  @Override
+  public void close() {
+    this.executor.shutdown();
+  }
+
   public class FibonacciSubmitTask implements Callable<Long> {
     private final long n;
 
@@ -39,9 +44,4 @@ public Long call() throws ExecutionException, InterruptedException {
       return future1.get() + future2.get();
     }
   }
-
-  @Override
-  public void close() {
-    this.executor.shutdown();
-  }
 }

dd-smoke-tests/concurrent/java-21/src/main/java/datadog/smoketest/concurrent/VirtualThreadSubmitRunnableCalculator.java

@@ -20,6 +20,11 @@ public long computeFibonacci(int n) throws ExecutionException, InterruptedExcept
     return task.result.get();
   }
 
+  @Override
+  public void close() {
+    this.executor.shutdown();
+  }
+
   public class FibonacciSubmitTask implements Runnable {
     private final long n;
     private final CompletableFuture<Long> result;
@@ -46,9 +51,4 @@ public void run() {
       }
     }
   }
-
-  @Override
-  public void close() {
-    this.executor.shutdown();
-  }
 }