Merge pull request microsoft#64 from xlab-uiuc/noop-astro

Add noop problem for astronomy shop

Author: HacksonClark

aiopslab/orchestrator/problems/no_op/no_op.py

@@ -7,6 +7,7 @@
 from aiopslab.service.kubectl import KubeCtl
 from aiopslab.service.apps.hotelres import HotelReservation
 from aiopslab.service.apps.socialnet import SocialNetwork
+from aiopslab.service.apps.astronomy_shop import AstronomyShop
 from aiopslab.generators.workload.wrk import Wrk
 from aiopslab.generators.fault.inject_noop import NoopFaultInjector
 from aiopslab.session import SessionItem
@@ -17,20 +18,25 @@
 
 class NoOpBaseTask:
     def __init__(self, app_name: str = "hotel"):
-        if app_name == "hotel":
+        self.app_name = app_name
+
+        if self.app_name == "hotel":
             self.app = HotelReservation()
             self.payload_script = (
                 TARGET_MICROSERVICES
                 / "hotelReservation/wrk2/scripts/hotel-reservation/mixed-workload_type_1.lua"
             )
             self.faulty_service = "PLACEHOLDER"
-        elif app_name == "social":
+        elif self.app_name == "social":
             self.app = SocialNetwork()
             self.payload_script = (
                 TARGET_MICROSERVICES
                 / "socialNetwork/wrk2/scripts/social-network/compose-post.lua"
             )
             self.faulty_service = "PLACEHOLDER"
+        elif self.app_name == "astronomy_shop":
+            self.app = AstronomyShop()
+            self.faulty_service = "PLACEHOLDER"
         else:
             raise ValueError(f"Unsupported app_name: {app_name}")
 
@@ -39,14 +45,18 @@ def __init__(self, app_name: str = "hotel"):
         self.injector = NoopFaultInjector(namespace=self.namespace)
 
     def start_workload(self):
-        print("== Start Workload ==")
-        frontend_url = get_frontend_url(self.app)
+        if self.app_name != 'astronomy_shop':
+            print("== Start Workload ==")
+            frontend_url = get_frontend_url(self.app)
 
-        wrk = Wrk(rate=10, dist="exp", connections=2, duration=10, threads=2)
-        wrk.start_workload(
-            payload_script=self.payload_script,
-            url=f"{frontend_url}",
-        )
+            wrk = Wrk(rate=10, dist="exp", connections=2, duration=10, threads=2)
+            wrk.start_workload(
+                payload_script=self.payload_script,
+                url=f"{frontend_url}",
+            )
+        else:
+            # Skip workload since astronomy shop has its own workload generator
+            print("== Workload Skipped ==")
 
     def inject_fault(self):
         print("== Fault Injection ==")

aiopslab/orchestrator/problems/registry.py

@@ -159,6 +159,7 @@ def __init__(self):
                 app_name="hotel"
             ),
             "noop_detection_social_network-1": lambda: NoOpDetection(app_name="social"),
+            "noop_detection_astronomy_shop-1": lambda: NoOpDetection(app_name="astronomy_shop"),
             # NOTE: This should be getting fixed by the great powers of @jinghao-jia
             # Kernel fault -> https://github.com/xlab-uiuc/agent-ops/pull/10#issuecomment-2468992285
             # There's a bug in chaos mesh regarding this fault, wait for resolution and retest kernel fault

aiopslab/service/apps/astronomy_shop.py

@@ -18,7 +18,7 @@ def __init__(self):
     def load_app_json(self):
         super().load_app_json()
         metadata = self.get_app_json()
-        self.frontend_service = "astronomy-shop-frontendproxy"
+        self.frontend_service = "frontend-proxy"
         self.frontend_port = 8080
 
     def deploy(self):