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

perf: Lock username is not case sensitive

pull/14218/head
feng 2024-09-20 18:03:43 +08:00 committed by Bryan
parent 6e8affcdd6
commit d09eb3c4fa
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/users/utils.py
View File

@ -110,6 +110,7 @@ class BlockUtil:
BLOCK_KEY_TMPL: str BLOCK_KEY_TMPL: str
def __init__(self, username): def __init__(self, username):
username = username.lower()
self.block_key = self.BLOCK_KEY_TMPL.format(username) self.block_key = self.BLOCK_KEY_TMPL.format(username)
self.key_ttl = int(settings.SECURITY_LOGIN_LIMIT_TIME) * 60 self.key_ttl = int(settings.SECURITY_LOGIN_LIMIT_TIME) * 60
@ -125,6 +126,7 @@ class BlockUtilBase:
BLOCK_KEY_TMPL: str BLOCK_KEY_TMPL: str
def __init__(self, username, ip): def __init__(self, username, ip):
username = username.lower()
self.username = username self.username = username
self.ip = ip self.ip = ip
self.limit_key = self.LIMIT_KEY_TMPL.format(username, ip) self.limit_key = self.LIMIT_KEY_TMPL.format(username, ip)
@ -158,6 +160,7 @@ class BlockUtilBase:
@classmethod @classmethod
def unblock_user(cls, username): def unblock_user(cls, username):
username = username.lower()
key_limit = cls.LIMIT_KEY_TMPL.format(username, '*') key_limit = cls.LIMIT_KEY_TMPL.format(username, '*')
key_block = cls.BLOCK_KEY_TMPL.format(username) key_block = cls.BLOCK_KEY_TMPL.format(username)
# Redis 尽量不要用通配 # Redis 尽量不要用通配
@ -166,6 +169,7 @@ class BlockUtilBase:
@classmethod @classmethod
def is_user_block(cls, username): def is_user_block(cls, username):
username = username.lower()
block_key = cls.BLOCK_KEY_TMPL.format(username) block_key = cls.BLOCK_KEY_TMPL.format(username)
return bool(cache.get(block_key)) return bool(cache.get(block_key))