github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: 优化 protocol 选择

pull/10032/head
ibuler 2023-03-22 15:26:23 +08:00
parent ba076f6612
commit 59d9572d07
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/ops/ansible/inventory.py
View File

@ -99,7 +99,8 @@ class JMSInventory:
if gateway: if gateway:
host.update(self.make_proxy_command(gateway)) host.update(self.make_proxy_command(gateway))
def asset_to_host(self, asset, account, automation, protocols, platform): @staticmethod
def get_primary_protocol(protocols):
primary_protocol = [p for p in protocols if p.primary] primary_protocol = [p for p in protocols if p.primary]
if len(primary_protocol) >= 1: if len(primary_protocol) >= 1:
primary = primary_protocol[0] primary = primary_protocol[0]
@ -114,6 +115,10 @@ class JMSInventory:
else: else:
protocol = 'null' protocol = 'null'
port = 0 port = 0
return protocol, port
def asset_to_host(self, asset, account, automation, protocols, platform):
protocol, port = self.get_primary_protocol(protocols)
host = { host = {
'name': '{}'.format(asset.name.replace(' ', '_')), 'name': '{}'.format(asset.name.replace(' ', '_')),