perf: 优化 m2m json field

pull/10327/head
ibuler 2023-05-08 14:09:44 +08:00
parent a112d3c99d
commit 1ec4cbdf38
4 changed files with 5 additions and 51 deletions

View File

@ -37,7 +37,8 @@ class LoginAssetCheckAPI(CreateAPIView):
'account_username': self.serializer.validated_data.get('account_username'), 'account_username': self.serializer.validated_data.get('account_username'),
'action': LoginAssetACL.ActionChoices.review 'action': LoginAssetACL.ActionChoices.review
} }
acl = LoginAssetACL.filter_queryset(**kwargs).valid().first() acl = LoginAssetACL.objects.filter(**kwargs).valid().first()
if acl: if acl:
need_review = True need_review = True
response_data = self._get_response_data_of_need_review(acl) response_data = self._get_response_data_of_need_review(acl)

View File

@ -1,19 +1,13 @@
from django.core.validators import MinValueValidator, MaxValueValidator from django.core.validators import MinValueValidator, MaxValueValidator
from django.db import models from django.db import models
from django.db.models import Q
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from common.db.fields import JSONManyToManyField from common.db.fields import JSONManyToManyField
from common.db.models import JMSBaseModel from common.db.models import JMSBaseModel
from common.utils import contains_ip from orgs.mixins.models import OrgModelMixin
from orgs.mixins.models import OrgModelMixin, OrgManager
__all__ = [ __all__ = [
'ACLManager', 'BaseACL', 'UserAssetAccountBaseACL',
'BaseACL',
'BaseACLQuerySet',
'UserAssetAccountBaseACL',
'UserAssetAccountACLQuerySet'
] ]
@ -37,41 +31,6 @@ class BaseACLQuerySet(models.QuerySet):
return self.inactive() return self.inactive()
class UserAssetAccountACLQuerySet(BaseACLQuerySet):
def filter_user(self, username):
q = Q(users__username_group__contains=username) | \
Q(users__username_group__contains='*')
return self.filter(q)
def filter_asset(self, name=None, address=None):
queryset = self.filter()
if name:
q = Q(assets__name_group__contains=name) | \
Q(assets__name_group__contains='*')
queryset = queryset.filter(q)
if address:
ids = [
q.id for q in queryset
if contains_ip(address, q.assets.get('address_group', []))
]
queryset = queryset.filter(id__in=ids)
return queryset
def filter_account(self, username):
q = Q(accounts__username_group__contains=username) | \
Q(accounts__username_group__contains='*')
return self.filter(q)
class ACLManager(models.Manager):
def valid(self):
return self.get_queryset().valid()
class OrgACLManager(OrgManager, ACLManager):
pass
class BaseACL(JMSBaseModel): class BaseACL(JMSBaseModel):
name = models.CharField(max_length=128, verbose_name=_('Name')) name = models.CharField(max_length=128, verbose_name=_('Name'))
priority = models.IntegerField( priority = models.IntegerField(
@ -84,7 +43,7 @@ class BaseACL(JMSBaseModel):
is_active = models.BooleanField(default=True, verbose_name=_("Active")) is_active = models.BooleanField(default=True, verbose_name=_("Active"))
ActionChoices = ActionChoices ActionChoices = ActionChoices
objects = ACLManager.from_queryset(BaseACLQuerySet)() objects = BaseACLQuerySet.as_manager()
class Meta: class Meta:
ordering = ('priority', 'date_updated', 'name') ordering = ('priority', 'date_updated', 'name')
@ -99,8 +58,6 @@ class UserAssetAccountBaseACL(BaseACL, OrgModelMixin):
assets = JSONManyToManyField('assets.Asset', default=dict, verbose_name=_('Assets')) assets = JSONManyToManyField('assets.Asset', default=dict, verbose_name=_('Assets'))
accounts = models.JSONField(default=list, verbose_name=_("Account")) accounts = models.JSONField(default=list, verbose_name=_("Account"))
objects = OrgACLManager.from_queryset(UserAssetAccountACLQuerySet)()
class Meta(BaseACL.Meta): class Meta(BaseACL.Meta):
unique_together = ('name', 'org_id') unique_together = ('name', 'org_id')
abstract = True abstract = True

View File

@ -1,11 +1,9 @@
from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ugettext_lazy as _
from .base import UserAssetAccountBaseACL from .base import UserAssetAccountBaseACL
class LoginAssetACL(UserAssetAccountBaseACL): class LoginAssetACL(UserAssetAccountBaseACL):
class Meta(UserAssetAccountBaseACL.Meta): class Meta(UserAssetAccountBaseACL.Meta):
verbose_name = _('Login asset acl') verbose_name = _('Login asset acl')
abstract = False abstract = False

View File

@ -10,7 +10,6 @@
""" """
import uuid import uuid
from functools import reduce
from django.db import models from django.db import models
from django.db import transaction from django.db import transaction
@ -55,7 +54,6 @@ def output_as_string(field_name):
class MultiTableChildQueryset(QuerySet): class MultiTableChildQueryset(QuerySet):
def bulk_create(self, objs, batch_size=None): def bulk_create(self, objs, batch_size=None):
assert batch_size is None or batch_size > 0 assert batch_size is None or batch_size > 0
if not objs: if not objs: