perf: 资产连接可以指定 AppletHost

apps/authentication/models/connection_token.py

@@ -175,7 +175,7 @@ class ConnectionToken(JMSOrgBaseModel):
         if not applet:
             return None
 
-        host_account = applet.select_host_account(self.user)
+        host_account = applet.select_host_account(self.user, self.asset)
         if not host_account:
             raise JMSException({'error': 'No host account available'})
 

apps/terminal/models/applet/applet.py

@@ -148,14 +148,18 @@ class Applet(JMSBaseModel):
         shutil.copytree(path, pkg_path)
         return instance, serializer
 
-    def select_host(self, user):
-        hosts = [
-            host for host in self.hosts.filter(is_active=True)
-            if host.load != 'offline'
-        ]
+    def select_host(self, user, asset):
+        hosts = self.hosts.filter(is_active=True)
+        hosts = [host for host in hosts if host.load != 'offline']
         if not hosts:
             return None
 
+        spec_label = asset.labels.filter(name__in=['AppletHost', '发布机']).first()
+        if spec_label:
+            host = [host for host in hosts if host.name == spec_label.value]
+            if host:
+                return host[0]
+
         prefer_key = 'applet_host_prefer_{}'.format(user.id)
         prefer_host_id = cache.get(prefer_key, None)
         pref_host = [host for host in hosts if host.id == prefer_host_id]
@@ -189,9 +193,9 @@ class Applet(JMSBaseModel):
             cache.set(prefer_host_account_key, account.id, timeout=None)
         return account
 
-    def select_host_account(self, user):
+    def select_host_account(self, user, asset):
         # 选择激活的发布机
-        host = self.select_host(user)
+        host = self.select_host(user, asset)
         if not host:
             return None
         host_concurrent = str(host.deploy_options.get('RDS_fSingleSessionPerUser', 0)) == '0'