opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-29 12:37:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(api): replace Any with precise types in db_migration_lock (#34891)

Browse Source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
corevibe555 2026-04-10 14:09:33 +03:00 committed by GitHub
parent 6612ba69b1
commit 130ad295d0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
api/libs/db_migration_lock.py
View File

@ -14,9 +14,15 @@ from __future__ import annotations
import logging import logging
import threading import threading
from typing import Any from typing import TYPE_CHECKING, Any
import redis
from redis.cluster import RedisCluster
from redis.exceptions import LockNotOwnedError, RedisError from redis.exceptions import LockNotOwnedError, RedisError
from redis.lock import Lock
if TYPE_CHECKING:
from extensions.ext_redis import RedisClientWrapper
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -38,21 +44,21 @@ class DbMigrationAutoRenewLock:
primary error/exit code. primary error/exit code.
""" """
_redis_client: Any _redis_client: redis.Redis | RedisCluster | RedisClientWrapper
_name: str _name: str
_ttl_seconds: float _ttl_seconds: float
_renew_interval_seconds: float _renew_interval_seconds: float
_log_context: str | None _log_context: str | None
_logger: logging.Logger _logger: logging.Logger
_lock: Any _lock: Lock | None
_stop_event: threading.Event | None _stop_event: threading.Event | None
_thread: threading.Thread | None _thread: threading.Thread | None
_acquired: bool _acquired: bool
def __init__( def __init__(
self, self,
redis_client: Any, redis_client: redis.Redis | RedisCluster | RedisClientWrapper,
name: str, name: str,
ttl_seconds: float = 60, ttl_seconds: float = 60,
renew_interval_seconds: float | None = None, renew_interval_seconds: float | None = None,
@ -127,7 +133,7 @@ class DbMigrationAutoRenewLock:
) )
self._thread.start() self._thread.start()
def _heartbeat_loop(self, lock: Any, stop_event: threading.Event) -> None: def _heartbeat_loop(self, lock: Lock, stop_event: threading.Event) -> None:
while not stop_event.wait(self._renew_interval_seconds): while not stop_event.wait(self._renew_interval_seconds):
try: try:
lock.reacquire() lock.reacquire()