github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2026-01-29 21:51:31 +00:00
Code Issues Projects Releases Wiki Activity

perf: 优化 gateway

Browse Source
This commit is contained in:
ibuler
2022-12-01 15:21:53 +08:00
parent d1461b33c5
commit 8162a1b17e
5 changed files with 64 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
apps/assets/api/domain.py
View File

@@ -35,7 +35,7 @@ class GatewayViewSet(OrgBulkModelViewSet):
serializer_class = serializers.GatewaySerializer
def get_queryset(self):
queryset = Host.get_gateway_queryset()
queryset = Domain.get_gateway_queryset()
return queryset
@@ -45,17 +45,17 @@ class GatewayTestConnectionApi(SingleObjectMixin, APIView):
}
def get_queryset(self):
queryset = Host.get_gateway_queryset()
queryset = Domain.get_gateway_queryset()
return queryset
def post(self, request, *args, **kwargs):
self.object = self.get_object()
local_port = self.request.data.get('port') or self.object.port
gateway = self.get_object()
local_port = self.request.data.get('port') or gateway.port
try:
local_port = int(local_port)
except ValueError:
raise ValidationError({'port': _('Number required')})
ok, e = self.object.test_connective(local_port=local_port)
ok, e = gateway.test_connective(local_port=local_port)
if ok:
return Response("ok")
else: