feature: 订阅更新接口支持 steps 更新 (closed TencentBlueKing#1002)

apps/backend/subscription/serializers.py

@@ -118,6 +118,7 @@ class ScopeSerializer(serializers.Serializer):
 
     class StepSerializer(serializers.Serializer):
         id = serializers.CharField()
+        type = serializers.CharField(required=False)
         params = serializers.DictField()
         config = serializers.DictField(required=False)
 

apps/backend/subscription/views.py

@@ -15,7 +15,7 @@
 from collections import defaultdict
 from copy import deepcopy
 from functools import cmp_to_key, reduce
-from typing import Dict
+from typing import Dict, List, Set
 
 import ujson as json
 from django.core.cache import caches
@@ -349,17 +349,44 @@ def update_subscription(self, request):
             subscription.bk_biz_scope = params.get("bk_biz_scope")
             subscription.save()
 
-            steps_group_by_id = {step["id"]: step for step in params["steps"]}
+            step_ids: Set[str] = set()
+            step_id__obj_map: Dict[str, models.SubscriptionStep] = {
+                step_obj.step_id: step_obj for step_obj in subscription.steps
+            }
+            step_objs_to_be_created: List[models.SubscriptionStep] = []
+            step_objs_to_be_updated: List[models.SubscriptionStep] = []
+
+            for index, step_info in enumerate(params["steps"]):
+                if step_info["id"] in step_id__obj_map:
+                    # 存在则更新
+                    step_obj: models.SubscriptionStep = step_id__obj_map[step_info["id"]]
+                    step_obj.params = step_info["params"]
+                    if "config" in step_info:
+                        step_obj.config = step_info["config"]
+                    step_obj.index = index
+                    step_objs_to_be_updated.append(step_obj)
+                else:
+                    # 新增
+                    step_objs_to_be_created.append(
+                        models.SubscriptionStep(
+                            subscription_id=subscription.id,
+                            index=index,
+                            step_id=step_info["id"],
+                            type=step_info["type"],
+                            config=step_info["config"],
+                            params=step_info["params"],
+                        )
+                    )
+                step_ids.add(step_info["id"])
 
-            for step in subscription.steps:
-                step.params = steps_group_by_id[step.step_id]["params"]
-                if "config" in steps_group_by_id[step.step_id]:
-                    step.config = steps_group_by_id[step.step_id]["config"]
-                step.save()
+            # 删除更新后不存在的 step
+            models.SubscriptionStep.objects.filter(
+                subscription_id=subscription.id, step_id__in=set(step_id__obj_map.keys()) - step_ids
+            ).delete()
+            models.SubscriptionStep.objects.bulk_update(step_objs_to_be_updated, fields=["config", "params", "index"])
+            models.SubscriptionStep.objects.bulk_create(step_objs_to_be_created)
 
-            result = {
-                "subscription_id": subscription.id,
-            }
+        result = {"subscription_id": subscription.id}
 
         if run_immediately:
             if subscription.is_running():

apps/node_man/migrations/0065_alter_subscriptionstep_unique_together.py

@@ -0,0 +1,17 @@
+# Generated by Django 3.2.4 on 2022-08-19 08:52
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("node_man", "0064_auto_20220721_1557"),
+    ]
+
+    operations = [
+        migrations.AlterUniqueTogether(
+            name="subscriptionstep",
+            unique_together={("subscription_id", "step_id")},
+        ),
+    ]

apps/node_man/models.py

@@ -1777,7 +1777,7 @@ class Meta:
         verbose_name = _("订阅步骤")
         verbose_name_plural = _("订阅步骤")
         ordering = ["index"]
-        unique_together = (("subscription_id", "index"), ("subscription_id", "step_id"))
+        unique_together = (("subscription_id", "step_id"),)
 
     def __str__(self):
         return (
@@ -1848,10 +1848,11 @@ class CategoryType(object):
     @property
     def steps(self):
         if not getattr(self, "_steps", None):
-            self._steps = SubscriptionStep.objects.filter(subscription_id=self.id)
+            self._steps = list(SubscriptionStep.objects.filter(subscription_id=self.id))
             for step in self._steps:
                 # 设置 subscription 属性，减少查询次数
                 step.subscription = self
+        self._steps = sorted(self._steps, key=lambda x: x.index)
         return self._steps
 
     @steps.setter

dev_log/2.2.22/crayon_202208191706.yaml

@@ -0,0 +1,3 @@
+---
+feature:
+  - "订阅更新接口支持 steps 更新 (closed #1002)"