chore(controller): transfer app ownership in async

Author: jianxiaoguo

rootfs/api/tasks.py

@@ -6,7 +6,7 @@
 from django.core import signals
 from celery import shared_task
 
-from api import manager
+from api import manager, models
 from api.exceptions import ServiceUnavailable
 from api.models.resource import Resource
 logger = logging.getLogger(__name__)
@@ -107,3 +107,24 @@ def mount_app(app, user, volume):
         raise e
     else:
         signals.request_finished.send(sender=task_id)
+
+
+@shared_task(
+    retry_kwargs={'max_retries': None}
+)
+def downstream_model_owner(app, old_owner, new_owner):
+    task_id = uuid.uuid4().hex
+    signals.request_started.send(sender=task_id)
+    try:
+        for downstream_model in [
+            models.appsettings.AppSettings, models.build.Build, models.certificate.Certificate,
+            models.config.Config, models.domain.Domain, models.key.Key, models.release.Release,
+            models.resource.Resource, models.tls.TLS, models.service.Service,
+            models.volume.Volume, models.gateway.Gateway, models.gateway.Route]:
+            downstream_model.objects.filter(owner=old_owner, app=app).update(owner=new_owner)
+        app.owner = new_owner
+        app.save()
+    except Exception as e:
+        signals.got_request_exception.send(sender=task_id)
+    else:
+        signals.request_finished.send(sender=task_id)

rootfs/api/views.py

@@ -19,7 +19,7 @@
 from rest_framework.viewsets import GenericViewSet
 
 from api import monitor, models, permissions, serializers, viewsets, authentication
-from api.tasks import scale_app, restart_app, mount_app
+from api.tasks import scale_app, restart_app, mount_app, downstream_model_owner
 from api.exceptions import AlreadyExists, ServiceUnavailable, DryccException, \
     UnprocessableEntity
 
@@ -281,14 +281,9 @@ def update(self, request, **kwargs):
             if self.request.user != app.owner and not self.request.user.is_superuser:
                 raise PermissionDenied()
             new_owner = get_object_or_404(User, username=request.data['owner'])
-            app.owner = new_owner
             # ensure all downstream objects that are owned by this user and are part of this app
             # is also updated
-            for downstream_model in [
-                    models.appsettings.AppSettings, models.build.Build, models.config.Config,
-                    models.domain.Domain, models.release.Release, models.tls.TLS]:
-                downstream_model.objects.filter(owner=old_owner, app=app).update(owner=new_owner)
-        app.save()
+            downstream_model_owner.delay(app, old_owner, new_owner)
         return Response(status=status.HTTP_200_OK)