extended load endpoint

Author: oleksandrh

src/main/java/org/springframework/samples/petclinic/clinicactivity/ClinicActivityController.java

@@ -194,17 +194,32 @@ public ResponseEntity<String> createLockContentionLoad(@RequestParam(name = "thr
 	 * Focuses on disk I/O bottlenecks that can be improved by faster storage or read replicas
 	 */
 	@PostMapping("/io-intensive-load")
-	public ResponseEntity<String> createIOIntensiveLoad(@RequestParam(name = "duration", defaultValue = "5") int durationMinutes) {
-		logger.warn("Received request to create I/O INTENSIVE LOAD for {} minutes - This will MAX OUT disk I/O operations!", durationMinutes);
+	public ResponseEntity<String> createIOIntensiveLoad(@RequestParam(name = "duration", defaultValue = "5") int durationMinutes,
+														@RequestParam(name = "threads", defaultValue = "6") int numThreads,
+														@RequestParam(name = "limit", defaultValue = "400000") int limit) {
+		logger.warn("Received request to create I/O INTENSIVE LOAD for {} minutes with {} threads and {} limit - This will MAX OUT disk I/O operations!", 
+					durationMinutes, numThreads, limit);
 		if (durationMinutes <= 0) {
 			return ResponseEntity.badRequest().body("Duration must be a positive integer.");
 		}
 		if (durationMinutes > 60) {
 			return ResponseEntity.badRequest().body("Duration too high for I/O intensive load - maximum 60 minutes to prevent storage overload.");
 		}
+		if (numThreads <= 0) {
+			return ResponseEntity.badRequest().body("Number of threads must be a positive integer.");
+		}
+		if (numThreads > 20) {
+			return ResponseEntity.badRequest().body("Too many threads for I/O intensive load - maximum 20 to prevent system crash.");
+		}
+		if (limit <= 0) {
+			return ResponseEntity.badRequest().body("Limit must be a positive integer.");
+		}
+		if (limit > 1000000) {
+			return ResponseEntity.badRequest().body("Limit too high for I/O intensive load - maximum 1,000,000 to prevent excessive resource usage.");
+		}
 		try {
-			dataService.createIOIntensiveLoad(durationMinutes);
-			return ResponseEntity.ok("Successfully completed I/O INTENSIVE LOAD for " + durationMinutes + " minutes - Disk I/O was maxed out!");
+			dataService.createIOIntensiveLoad(durationMinutes, numThreads, limit);
+			return ResponseEntity.ok("Successfully completed I/O INTENSIVE LOAD for " + durationMinutes + " minutes with " + numThreads + " threads and " + limit + " limit - Disk I/O was maxed out!");
 		} catch (Exception e) {
 			logger.error("Error during I/O intensive load", e);
 			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Error during I/O intensive load: " + e.getMessage());

src/main/java/org/springframework/samples/petclinic/clinicactivity/ClinicActivityDataService.java

@@ -411,25 +411,24 @@ else if (operationCount % 5 == 3) {
      * Uses simple queries with large data transfers to keep I/O busy while minimizing CPU/Memory usage
      * Focuses on disk I/O bottlenecks that can be improved by faster storage or read replicas
      */
-    public void createIOIntensiveLoad(int durationMinutes) {
-        logger.warn("Starting I/O INTENSIVE load test for {} minutes - This will MAX OUT disk I/O operations!", durationMinutes);
+    public void createIOIntensiveLoad(int durationMinutes, int numThreads, int limit) {
+        logger.warn("Starting I/O INTENSIVE load test for {} minutes with {} threads and {} limit - This will MAX OUT disk I/O operations!", 
+                   durationMinutes, numThreads, limit);
         long startTime = System.currentTimeMillis();
         long endTime = startTime + (durationMinutes * 60 * 1000L);
 
         try {
             AtomicInteger globalOperationCount = new AtomicInteger(0);
             List<Thread> threads = new ArrayList<>();
 
-            // Use fewer threads than CPU intensive to minimize CPU usage (focus on I/O)
-            int numThreads = 6;
-            logger.info("Creating {} I/O intensive threads (fewer than CPU load to focus on disk I/O)...", numThreads);
+            logger.info("Creating {} I/O intensive threads with {} record limit per query...", numThreads, limit);
 
             // Create I/O intensive threads
             for (int t = 0; t < numThreads; t++) {
                 final int threadId = t;
                 Thread ioThread = new Thread(() -> {
                     try {
-                        executeIOIntensiveThread(threadId, endTime, globalOperationCount);
+                        executeIOIntensiveThread(threadId, endTime, globalOperationCount, limit);
                     } catch (Exception e) {
                         logger.error("Error in I/O intensive thread {}", threadId, e);
                     }
@@ -456,21 +455,21 @@ public void createIOIntensiveLoad(int durationMinutes) {
             }
 
             long actualEndTime = System.currentTimeMillis();
-            logger.warn("Completed I/O INTENSIVE load test in {} ms. Total operations: {}",
-                (actualEndTime - startTime), globalOperationCount.get());
+            logger.warn("Completed I/O INTENSIVE load test in {} ms with {} threads and {} limit. Total operations: {}",
+                (actualEndTime - startTime), numThreads, limit, globalOperationCount.get());
 
         } catch (Exception e) {
             logger.error("Error during I/O intensive load test", e);
             throw new RuntimeException("Error during I/O intensive load test: " + e.getMessage(), e);
         }
     }
 
-    private void executeIOIntensiveThread(int threadId, long endTime, AtomicInteger globalOperationCount) {
+    private void executeIOIntensiveThread(int threadId, long endTime, AtomicInteger globalOperationCount, int limit) {
         Random random = new Random();
         Faker faker = new Faker(new Locale("en-US"));
         int localOperationCount = 0;
 
-        logger.info("I/O Thread {} starting I/O intensive operations...", threadId);
+        logger.info("I/O Thread {} starting I/O intensive operations with {} record limit...", threadId, limit);
 
         while (System.currentTimeMillis() < endTime) {
 			try {
@@ -481,7 +480,7 @@ private void executeIOIntensiveThread(int threadId, long endTime, AtomicInteger
                             "FROM clinic_activity_logs " +
                             "WHERE LENGTH(payload) > 100 " +
 							"ORDER BY random()" +
-                            "LIMIT 350000");
+                            "LIMIT " + limit);
 
 
                 localOperationCount++;