fix header control

thundra-io · Sep 19, 2023 · 24fcd09 · 24fcd09
1 parent 20e0c37
commit 24fcd09
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/src/main/java/com/catchpoint/tracing/demo/todo/config/ChaosConfiguration.java b/src/main/java/com/catchpoint/tracing/demo/todo/config/ChaosConfiguration.java
@@ -11,10 +11,27 @@ public class ChaosConfiguration {
 
     private static final short RANDOM_EXCEPTION_RATIO = 5;
 
-    @Value("${chaos.enabled:false}")
+
+    @Value("${chaos.enabled:true}")
+    private boolean enabled;
+
     private boolean isRandomException;
 
     private final Random random = new Random();
+
+    public void setEnabled(boolean enabled) {
+        isRandomException = enabled;
+    }
+
+    public void restore() {
+        isRandomException = enabled;
+    }
+
+    public ChaosConfiguration() {
+        restore();
+    }
+
+
     private static class ServerDownException extends Exception {
         ServerDownException() {
             super("Server is down!");

diff --git a/src/main/java/com/catchpoint/tracing/demo/todo/controller/TodoController.java b/src/main/java/com/catchpoint/tracing/demo/todo/controller/TodoController.java
@@ -60,16 +60,16 @@ public TodoController(TodoService service, ChaosConfiguration chaosConfiguration
     public void handleMap(Map<String, String> map) {
         if (map.containsKey("x-chaos-inject-error") && map.get("x-chaos-inject-error").equals("true")) {
             throw new RuntimeException("This error is thrown for testing purpose.");
-        }
-
-        if (map.containsKey("x-chaos-inject-latency") && map.get("x-chaos-inject-latency").equals("true")) {
+        } else if (map.containsKey("x-chaos-inject-latency") && map.get("x-chaos-inject-latency").equals("true")) {
+            chaosConfiguration.setEnabled(false);
             try {
                 Thread.sleep(TEN_SECONDS_AS_MILLIS);
             } catch (InterruptedException e) {
                 e.printStackTrace();
             }
+        } else {
+            chaosConfiguration.restore();
         }
-
     }
 
     public void checkUserAccess(Map<String, String> headers) {