Sync non spring code to spring boot 3

.github/CODEOWNERS

@@ -44,7 +44,7 @@
 /sdk/batch/                                         @gingi @paterasMSFT @dpwatrous
 
 # PRLabel: %Communication
-/sdk/communication/                                 @JianpingChen @ankitarorabit @minnieliu @Azure/azure-sdk-communication-code-reviewers
+/sdk/communication/
 
 # PRLabel: %Communication - Calling Server
 /sdk/communication/sdk/communication/azure-communication-callingserver/ @minwoolee-msft
@@ -53,7 +53,7 @@
 /sdk/communication/sdk/communication/azure-communication-callautomation/ @juntuchen-msft @cochi2
 
 # PRLabel: %Communication - Chat
-/sdk/communication/sdk/communication/azure-communication-chat/ @JianpingChen @ankitarorabit @minnieliu @Azure/azure-sdk-communication-code-reviewers
+/sdk/communication/sdk/communication/azure-communication-chat/ @ankitarorabit @minnieliu @Azure/azure-sdk-communication-code-reviewers
 
 # PRLabel: %Communication - Identity
 /sdk/communication/azure-communication-identity/    @Azure/acs-identity-sdk @petrsvihlik @AikoBB @maximrytych-ms @ostoliarova-msft
@@ -62,7 +62,7 @@
 /sdk/communication/azure-communication-common/      @Azure/acs-identity-sdk @petrsvihlik @AikoBB @maximrytych-ms @ostoliarova-msft
 
 # PRLabel: %Communication - Network Traversal
-/sdk/communication/sdk/communication/azure-communication-networktraversal/ @JianpingChen @ankitarorabit @minnieliu @Azure/azure-sdk-communication-code-reviewers
+/sdk/communication/sdk/communication/azure-communication-networktraversal/ @ankitarorabit @minnieliu @Azure/azure-sdk-communication-code-reviewers
 
 # PRLabel: %Communication - Phone Numbers
 /sdk/communication/azure-communication-phonenumbers/ @miguhern @whisper6284 @lucasrsant @RoyHerrod @danielav7
@@ -86,11 +86,11 @@
 /sdk/core/azure-core-tracing-opentelemetry/         @samvaity @alzimmermsft @trask @lmolkova
 
 # PRLabel: %Cosmos
-/sdk/cosmos/                                        @kushagraThapar @FabianMeiswinkel @kirankumarkolli @xinlian12 @milismsft @aayush3011 @simorenoh
+/sdk/cosmos/                                        @kushagraThapar @FabianMeiswinkel @kirankumarkolli @xinlian12 @milismsft @aayush3011 @simorenoh @jeet1995
 # PRLabel: %azure-spring
-/sdk/cosmos/azure-spring-data-cosmos/               @kushagraThapar @FabianMeiswinkel @backwind1233 @chenrujun @hui1110 @netyyyy @saragluna @stliu @yiliuTo @xinlian12 @moarychan @aayush3011 @simorenoh @fangjian0423
+/sdk/cosmos/azure-spring-data-cosmos/               @kushagraThapar @FabianMeiswinkel @backwind1233 @chenrujun @hui1110 @netyyyy @saragluna @stliu @yiliuTo @xinlian12 @moarychan @aayush3011 @simorenoh @fangjian0423 @jeet1995
 # PRLabel: %azure-spring
-/sdk/cosmos/azure-spring-data-cosmos-test/          @kushagraThapar @FabianMeiswinkel @backwind1233 @chenrujun @hui1110 @netyyyy @saragluna @stliu @yiliuTo @xinlian12 @moarychan @aayush3011 @simorenoh @fangjian0423
+/sdk/cosmos/azure-spring-data-cosmos-test/          @kushagraThapar @FabianMeiswinkel @backwind1233 @chenrujun @hui1110 @netyyyy @saragluna @stliu @yiliuTo @xinlian12 @moarychan @aayush3011 @simorenoh @fangjian0423 @jeet1995
 
 # PRLabel: %Load Testing
 /sdk/loadtestservice/                               @Harshan01 @abranj1219

.vscode/cspell.json

@@ -165,7 +165,6 @@
     "sdk/parents/azure-client-sdk-parent/**",
     "sdk/parents/azure-sdk-parent/**",
     "sdk/parents/azure-data-sdk-parent/**",
-    "sdk/metricsadvisor/azure-ai-metricsadvisor/**",
     "sdk/personalizer/azure-ai-personalizer/**",
     "sdk/purview/azure-analytics-purview-administration/**",
     "sdk/quantum/azure-quantum-jobs/**",
@@ -177,7 +176,6 @@
     "sdk/schemaregistry/azure-data-schemaregistry-apacheavro/**",
     "sdk/servicebus/build/**",
     "sdk/spring/scripts/**",
-    "sdk/servicebus/azure-messaging-servicebus/**",
     "sdk/spring/spring-cloud-azure-actuator/**",
     "sdk/spring/spring-cloud-azure-actuator-autoconfigure/**",
     "sdk/spring/spring-cloud-azure-integration-tests/**",
@@ -258,6 +256,7 @@
   "words": [
     "adal",
     "amqp",
+    "Apim",
     "AUHours",
     "autoscale",
     "autodetection",
@@ -630,8 +629,20 @@
       "filename": "sdk/spring/CHANGELOG.md",
       "words": [
         "AADB",
-        "JPMS",
-        "nimbusds"
+        "JPMS"
+      ]
+    },
+    {
+      "filename": "sdk/spring/README.md",
+      "words": [
+        "Dcheckstyle",
+        "Dcodesnippet",
+        "Djacoco",
+        "Drevapi",
+        "Dskip",
+        "Dspotbugs",
+        "Pdev",
+        ""
       ]
     },
     {
@@ -672,6 +683,34 @@
         "RAGRS",
         "saoid"
       ]
+    },
+    {
+      "filename": "sdk/metricsadvisor/azure-ai-metricsadvisor/**",
+      "words": [
+        "APIV",
+        "bacf",
+        "alertme",
+        "adwiki",
+        "howto",
+        "bpfdfee",
+        "deduped",
+        "dedupe",
+        "POSTGRE",
+        "dgfbbbb",
+        "gdgfbbbb",
+        "dvhkl",
+        "yufrjo",
+        "kldn",
+        "dccf"
+      ]
+    },
+    {
+      "filename": "sdk/servicebus/azure-messaging-servicebus/**",
+      "words": [
+        "Conniey",
+        "qpid",
+        "unretriable"
+        ]
     }
   ],
   "allowCompoundWords": true

common/perf-test-core/src/main/java/com/azure/perf/test/core/ApiPerfTestBase.java

@@ -24,11 +24,12 @@
 import java.security.KeyManagementException;
 import java.security.NoSuchAlgorithmException;
 import java.security.SecureRandom;
-import java.util.Arrays;
+import java.util.Collections;
 import java.util.concurrent.CompletableFuture;
 
 /**
  * The Base Performance Test class for API based Perf Tests.
+ *
  * @param <TOptions> the performance test options to use while running the test.
  */
 public abstract class ApiPerfTestBase<TOptions extends PerfStressOptions> extends PerfTestBase<TOptions> {
@@ -46,6 +47,7 @@ public abstract class ApiPerfTestBase<TOptions extends PerfStressOptions> extend
 
     /**
      * Creates an instance of the Http Based Performance test.
+     *
      * @param options the performance test options to use while running the test.
      * @throws IllegalStateException if an errors is encountered with building ssl context.
      */
@@ -58,7 +60,7 @@ public ApiPerfTestBase(TOptions options) {
             recordPlaybackHttpClient = createRecordPlaybackClient(options);
             testProxy = options.getTestProxies().get(parallelIndex % options.getTestProxies().size());
             testProxyPolicy = new TestProxyPolicy(testProxy);
-            policies = Arrays.asList(testProxyPolicy);
+            policies = Collections.singletonList(testProxyPolicy);
         } else {
             recordPlaybackHttpClient = null;
             testProxy = null;
@@ -79,7 +81,7 @@ private static HttpClient createHttpClient(PerfStressOptions options) {
 
                         reactor.netty.http.client.HttpClient nettyHttpClient =
                             reactor.netty.http.client.HttpClient.create()
-                            .secure(sslContextSpec -> sslContextSpec.sslContext(sslContext));
+                                .secure(sslContextSpec -> sslContextSpec.sslContext(sslContext));
 
                         return new NettyAsyncHttpClientBuilder(nettyHttpClient).build();
                     } catch (SSLException e) {
@@ -127,21 +129,20 @@ private static reactor.netty.http.client.HttpClient createRecordPlaybackClient(P
     }
 
     /**
-     * Attempts to configure a ClientBuilder using reflection.  If a ClientBuilder does not follow the standard convention,
-     * it can be configured manually using the "httpClient" and "policies" fields.
+     * Attempts to configure a ClientBuilder using reflection.  If a ClientBuilder does not follow the standard
+     * convention, it can be configured manually using the "httpClient" and "policies" fields.
+     *
      * @param clientBuilder The client builder.
      * @throws IllegalStateException If reflective access to get httpClient or addPolicy methods fail.
      */
     protected void configureClientBuilder(HttpTrait<?> clientBuilder) {
-        if (httpClient != null || policies != null) {
-            if (httpClient != null) {
-                clientBuilder.httpClient(httpClient);
-            }
+        if (httpClient != null) {
+            clientBuilder.httpClient(httpClient);
+        }
 
-            if (policies != null) {
-                for (HttpPipelinePolicy policy : policies) {
-                    clientBuilder.addPolicy(policy);
-                }
+        if (policies != null) {
+            for (HttpPipelinePolicy policy : policies) {
+                clientBuilder.addPolicy(policy);
             }
         }
     }
@@ -163,35 +164,45 @@ public Mono<Void> runAllAsync(long endNanoTime) {
         lastCompletionNanoTime = 0;
         long startNanoTime = System.nanoTime();
 
-        return Flux.just(1)
-            .repeat()
-            .flatMap(i -> runTestAsync(), 1)
-            .doOnNext(v -> {
-                completedOperations += v;
+        return Flux.generate(sink -> {
+                if (System.nanoTime() < endNanoTime) {
+                    sink.next(1);
+                } else {
+                    sink.complete();
+                }
+            })
+            .flatMap(ignored -> {
+                if (System.nanoTime() < endNanoTime) {
+                    return runTestAsync();
+                } else {
+                    return Mono.just(0);
+                }
+            }, 1)
+            .doOnNext(result -> {
+                completedOperations += result;
                 lastCompletionNanoTime = System.nanoTime() - startNanoTime;
             })
-            .takeWhile(i -> System.nanoTime() < endNanoTime)
             .then();
     }
 
     /**
-     * Indicates how many operations were completed in a single run of the test.
-     * Good to be used for batch operations.
+     * Indicates how many operations were completed in a single run of the test. Good to be used for batch operations.
      *
      * @return the number of successful operations completed.
      */
     abstract int runTest();
 
     /**
-     * Indicates how many operations were completed in a single run of the async test.
-     * Good to be used for batch operations.
+     * Indicates how many operations were completed in a single run of the async test. Good to be used for batch
+     * operations.
      *
      * @return the number of successful operations completed.
      */
     abstract Mono<Integer> runTestAsync();
 
     /**
      * Stops playback tests.
+     *
      * @return An empty {@link Mono}.
      */
     public Mono<Void> stopPlaybackAsync() {
@@ -245,6 +256,7 @@ private Mono<Void> startPlaybackAsync() {
 
     /**
      * Records responses and starts tests in playback mode.
+     *
      * @return
      */
     @Override
@@ -253,16 +265,16 @@ Mono<Void> postSetupAsync() {
 
             // Make one call to Run() before starting recording, to avoid capturing one-time setup like authorization requests.
             return runSyncOrAsync()
-            .then(startRecordingAsync())
-            .then(Mono.defer(() -> {
+                .then(startRecordingAsync())
+                .then(Mono.defer(() -> {
                     testProxyPolicy.setRecordingId(recordingId);
                     testProxyPolicy.setMode("record");
                     return Mono.empty();
                 }))
-            .then(runSyncOrAsync())
-            .then(stopRecordingAsync())
-            .then(startPlaybackAsync())
-            .then(Mono.defer(() -> {
+                .then(runSyncOrAsync())
+                .then(stopRecordingAsync())
+                .then(startPlaybackAsync())
+                .then(Mono.defer(() -> {
                     testProxyPolicy.setRecordingId(recordingId);
                     testProxyPolicy.setMode("playback");
                     return Mono.empty();

common/perf-test-core/src/main/java/com/azure/perf/test/core/PerfStressProgram.java

@@ -18,26 +18,38 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+import java.util.concurrent.Callable;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ForkJoinPool;
+import java.util.concurrent.TimeUnit;
 import java.util.function.Supplier;
 import java.util.stream.IntStream;
-import java.util.stream.Stream;
 
 /**
  * Represents the main program class which reflectively runs and manages the performance tests.
  */
 public class PerfStressProgram {
     private static final int NANOSECONDS_PER_SECOND = 1_000_000_000;
 
-    private static int getCompletedOperations(PerfTestBase<?>[] tests) {
-        return Stream.of(tests).mapToInt(perfStressTest -> Long.valueOf(perfStressTest.getCompletedOperations()).intValue()).sum();
+    private static long getCompletedOperations(PerfTestBase<?>[] tests) {
+        long completedOperations = 0;
+        for (PerfTestBase<?> test : tests) {
+            completedOperations += test.getCompletedOperations();
+        }
+
+        return completedOperations;
     }
 
     private static double getOperationsPerSecond(PerfTestBase<?>[] tests) {
-        return IntStream.range(0, tests.length)
-            .mapToDouble(i -> tests[i].getCompletedOperations() / (((double) tests[i].lastCompletionNanoTime) / NANOSECONDS_PER_SECOND))
-            .sum();
+        double operationsPerSecond = 0.0D;
+        for (PerfTestBase<?> test : tests) {
+            double temp = test.getCompletedOperations() / (((double) test.lastCompletionNanoTime) / NANOSECONDS_PER_SECOND);
+            if (!Double.isNaN(temp)) {
+                operationsPerSecond += temp;
+            }
+        }
+
+        return operationsPerSecond;
     }
 
     /**
@@ -187,7 +199,7 @@ public static void run(Class<?> testClass, PerfStressOptions options) {
                     if (!options.isNoCleanup()) {
                         cleanupStatus = printStatus("=== Cleanup ===", () -> ".", false, false);
 
-                        Flux.just(tests).flatMap(t -> t.cleanupAsync()).blockLast();
+                        Flux.just(tests).flatMap(PerfTestBase::cleanupAsync).blockLast();
                     }
                 }
             }
@@ -222,11 +234,11 @@ public static void runTests(PerfTestBase<?>[] tests, boolean sync, int parallel,
 
         long endNanoTime = System.nanoTime() + ((long) durationSeconds * 1000000000);
 
-        int[] lastCompleted = new int[]{0};
+        long[] lastCompleted = new long[]{0};
         Disposable progressStatus = printStatus(
             "=== " + title + " ===" + System.lineSeparator() + "Current\t\tTotal\t\tAverage", () -> {
-                int totalCompleted = getCompletedOperations(tests);
-                int currentCompleted = totalCompleted - lastCompleted[0];
+                long totalCompleted = getCompletedOperations(tests);
+                long currentCompleted = totalCompleted - lastCompleted[0];
                 double averageCompleted = getOperationsPerSecond(tests);
 
                 lastCompleted[0] = totalCompleted;
@@ -236,10 +248,16 @@ public static void runTests(PerfTestBase<?>[] tests, boolean sync, int parallel,
         try {
             if (sync) {
                 ForkJoinPool forkJoinPool = new ForkJoinPool(parallel);
-                forkJoinPool.submit(() -> {
-                    IntStream.range(0, parallel).parallel().forEach(i -> tests[i].runAll(endNanoTime));
-                }).get();
+                List<Callable<Integer>> operations = new ArrayList<>(parallel);
+                for (PerfTestBase<?> test : tests) {
+                    operations.add(() -> {
+                        test.runAll(endNanoTime);
+                        return 1;
+                    });
+                }
 
+                forkJoinPool.invokeAll(operations, (durationSeconds * 1000L) + 100L, TimeUnit.MILLISECONDS);
+                forkJoinPool.shutdown();
             } else {
                 // Exceptions like OutOfMemoryError are handled differently by the default Reactor schedulers. Instead of terminating the
                 // Flux, the Flux will hang and the exception is only sent to the thread's uncaughtExceptionHandler and the Reactor
@@ -251,13 +269,13 @@ public static void runTests(PerfTestBase<?>[] tests, boolean sync, int parallel,
                 });
 
                 Flux.range(0, parallel)
-                    .parallel()
-                    .runOn(Schedulers.boundedElastic())
-                    .flatMap(i -> tests[i].runAllAsync(endNanoTime))
+                    .parallel(parallel)
+                    .runOn(Schedulers.parallel())
+                    .flatMap(i -> tests[i].runAllAsync(endNanoTime), false, Math.min(parallel, 1000 / parallel), 1)
                     .then()
                     .block();
             }
-        } catch (InterruptedException | ExecutionException e) {
+        } catch (InterruptedException e) {
             System.err.println("Error occurred when submitting jobs to ForkJoinPool. " + System.lineSeparator() + e);
             e.printStackTrace(System.err);
             throw new RuntimeException(e);
@@ -270,7 +288,7 @@ public static void runTests(PerfTestBase<?>[] tests, boolean sync, int parallel,
 
         System.out.println("=== Results ===");
 
-        int totalOperations = getCompletedOperations(tests);
+        long totalOperations = getCompletedOperations(tests);
         if (totalOperations == 0) {
             throw new IllegalStateException("Zero operations has been completed");
         }