github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2026-03-18 11:02:09 +00:00
Code Issues Projects Releases Wiki Activity

perf: limit license asset count

Browse Source
This commit is contained in:
feng
2026-03-13 17:58:41 +08:00
parent ce2d0ff3b3
commit 144abf99e2
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
apps/assets/api/asset/asset.py
View File

@@ -32,6 +32,11 @@ __all__ = [
]
def get_license_asset_limit():
license_info = getattr(settings, 'XPACK_LICENSE_INFO', {}) or {}
return license_info.get('license', {}).get('count') or 0
class AssetFilterSet(BaseFilterSet):
platform = drf_filters.CharFilter(method='filter_platform')
is_gateway = drf_filters.BooleanFilter(method='filter_is_gateway')
@@ -153,10 +158,18 @@ class BaseAssetViewSet(OrgBulkModelViewSet):
error = _('Cannot create asset directly, you should create a host or other')
return Response({'error': error}, status=400)
if not settings.XPACK_LICENSE_IS_VALID and self.model.objects.order_by().count() >= 5000:
asset_count = self.model.objects.order_by().count()
if not settings.XPACK_LICENSE_IS_VALID and asset_count >= 5000:
error = _('The number of assets exceeds the limit of 5000')
return Response({'error': error}, status=400)
if settings.XPACK_LICENSE_IS_VALID:
license_asset_limit = get_license_asset_limit()
if license_asset_limit and asset_count >= license_asset_limit:
error = _('The number of assets exceeds the license limit')
return Response({'error': error}, status=400)
return super().create(request, *args, **kwargs)