1
0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-05-06 15:16:32 +00:00

fix: select_for_update error for out join sql

This commit is contained in:
Bai 2024-06-13 15:49:28 +08:00
parent 2a33337963
commit e40c66c7ed

View File

@ -189,18 +189,15 @@ class SessionViewSet(RecordViewLogMixin, OrgBulkModelViewSet):
return Response({'msg': 'ok', 'id': activity_log.id})
def get_queryset(self):
queryset = super().get_queryset() \
.prefetch_related('terminal') \
.annotate(terminal_display=F('terminal__name'))
return queryset
def filter_queryset(self, queryset):
queryset = super().filter_queryset(queryset)
# 解决guacamole更新session时并发导致幽灵会话的问题暂不处理
if self.request.method in ('PATCH',):
queryset = super().get_queryset()
if self.request.method in ('GET',):
queryset = queryset.prefetch_related('terminal').annotate(terminal_display=F('terminal__name'))
elif self.request.method in ('PATCH',):
# postgres reports an error for statements that use select_for_update for out join
# so we need to use select_for_update only for have not prefetch_related and annotate
queryset = queryset.select_for_update()
return queryset
def perform_create(self, serializer):
if hasattr(self.request.user, 'terminal'):
serializer.validated_data["terminal"] = self.request.user.terminal