mirror of https://github.com/jumpserver/jumpserver
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
import re
|
|
from importlib import import_module
|
|
|
|
from django.conf import settings
|
|
from django.contrib.sessions.backends.cache import (
|
|
SessionStore as DjangoSessionStore
|
|
)
|
|
from django.core.cache import cache, caches
|
|
|
|
from common.utils import get_logger
|
|
from jumpserver.utils import get_current_request
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
class SessionStore(DjangoSessionStore):
|
|
ignore_urls = [
|
|
r'^/api/v1/users/profile/',
|
|
r'^/api/v1/authentication/user-session/'
|
|
]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.ignore_pattern = re.compile('|'.join(self.ignore_urls))
|
|
|
|
def save(self, *args, **kwargs):
|
|
request = get_current_request()
|
|
if request is None or not self.ignore_pattern.match(request.path):
|
|
try:
|
|
super().save(*args, **kwargs)
|
|
except Exception as e:
|
|
logger.info(f'SessionStore save error: {e}')
|
|
|
|
|
|
class RedisUserSessionManager:
|
|
JMS_SESSION_KEY = 'jms_session_key'
|
|
|
|
def __init__(self):
|
|
self.client = cache.client.get_client()
|
|
|
|
def add_or_increment(self, session_key):
|
|
self.client.hincrby(self.JMS_SESSION_KEY, session_key, 1)
|
|
|
|
def decrement(self, session_key):
|
|
self.client.hincrby(self.JMS_SESSION_KEY, session_key, -1)
|
|
|
|
def remove(self, session_key):
|
|
try:
|
|
self.client.hdel(self.JMS_SESSION_KEY, session_key)
|
|
session_store = import_module(settings.SESSION_ENGINE).SessionStore(session_key)
|
|
session_store.delete()
|
|
except Exception:
|
|
pass
|
|
|
|
def check_active(self, session_key):
|
|
count = self.client.hget(self.JMS_SESSION_KEY, session_key)
|
|
count = 0 if count is None else int(count.decode('utf-8'))
|
|
return count > 0
|
|
|
|
def get_active_keys(self):
|
|
session_keys = []
|
|
for k, v in self.client.hgetall(self.JMS_SESSION_KEY).items():
|
|
count = int(v.decode('utf-8'))
|
|
if count <= 0:
|
|
continue
|
|
key = k.decode('utf-8')
|
|
session_keys.append(key)
|
|
return session_keys
|
|
|
|
@staticmethod
|
|
def get_keys():
|
|
session_store_cls = import_module(settings.SESSION_ENGINE).SessionStore
|
|
cache_key_prefix = session_store_cls.cache_key_prefix
|
|
keys = caches[settings.SESSION_CACHE_ALIAS].iter_keys('*')
|
|
return [k.replace(cache_key_prefix, '') for k in keys]
|
|
|
|
|
|
user_session_manager = RedisUserSessionManager()
|