feat(maintenance): add maintenance support for ingress

Author: duanhongyi

rootfs/api/models/app.py

@@ -944,15 +944,18 @@ def maintenance_mode(self, mode):
         """
         Turn application maintenance mode on/off
         """
-        service = self._fetch_service_config(self.id)
-        old_service = service.copy()  # in case anything fails for rollback
-
-        try:
-            service['metadata']['annotations']['router.drycc.cc/maintenance'] = str(mode).lower()
-            self._scheduler.svc.update(self.id, self.id, data=service)
-        except KubeException as e:
-            self._scheduler.svc.update(self.id, self.id, data=old_service)
-            raise ServiceUnavailable(str(e)) from e
+        if mode:
+            namespace = ingress = self.id
+            try:
+                data = self._scheduler.ingress.get(namespace, ingress).json()
+                for rule in data["spec"]["rules"]:
+                    for path in rule["http"]["paths"]:
+                        path["backend"]["serviceName"] = "drycc-maintenance-proxy"
+                self._scheduler.ingress.patch(ingress, namespace, data)
+            except KubeException:
+                self.log("creating Ingress {}".format(namespace), level=logging.INFO)
+        else:
+            self.refresh_ingress_and_tls()
 
     def routable(self, routable):
         """

rootfs/scheduler/resources/ingress.py

@@ -135,6 +135,15 @@ def put(self, ingress, ingress_class, namespace, version, **kwargs):
         data = self.manifest(self.api_version, ingress, ingress_class, namespace, **kwargs)
         response = self.http_put(url, json=data)
 
+        if self.unhealthy(response.status_code):
+            raise KubeHTTPException(response, "put Ingress {}".format(namespace))
+
+        return response
+
+    def patch(self, ingress, namespace, data):
+        url = self.api("/namespaces/{}/ingresses/{}", namespace, ingress)
+        response = self.http_put(url, json=data)
+
         if self.unhealthy(response.status_code):
             raise KubeHTTPException(response, "patch Ingress {}".format(namespace))