mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-07-03 07:39:39 +00:00
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import os
|
||
from ..base.device import Device
|
||
from django.core.cache import cache
|
||
from redis_lock import Lock as RedisLock
|
||
|
||
|
||
class PiicoDevice(Device):
|
||
name = "piico"
|
||
|
||
def __init__(self):
|
||
self.open()
|
||
|
||
# 默认去lib路径检索
|
||
def open(self, driver_path="libpiico_ccmu.so"):
|
||
super().open(driver_path)
|
||
|
||
def reset_key_store(self):
|
||
redis_client = cache.client.get_client()
|
||
server_hostname = os.environ.get("SERVER_HOSTNAME")
|
||
RESET_LOCK_KEY = f"spiico:{server_hostname}:reset"
|
||
LOCK_EXPIRE_SECONDS = 300
|
||
|
||
if self._driver is None:
|
||
raise Exception("no driver loaded", 0)
|
||
if self.__device is None:
|
||
raise Exception("device not open", 0)
|
||
|
||
# ---- 分布式锁（Redis-Lock 实现 Redlock） ----
|
||
lock = RedisLock(
|
||
redis_client,
|
||
RESET_LOCK_KEY,
|
||
expire=LOCK_EXPIRE_SECONDS, # 锁自动过期
|
||
auto_renewal=False, # 不自动续租
|
||
)
|
||
|
||
# 尝试获取锁，拿不到直接返回
|
||
if not lock.acquire(blocking=False):
|
||
return
|
||
# ---- 真正执行 reset ----
|
||
ret = self._driver.SPII_ResetModule(self.__device)
|
||
if ret != 0:
|
||
raise Exception("reset device failed", ret)
|