PCNUSMSE/transcript_service
0
1o
2 W��h�- � @ s� d Z ddlZddlZddlZddlmZmZmZmZm Z m3Z4 ddl m Z ddl
mZ ddlmZ G dd � d e �ZG d5d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� d�ZG dd� d�Ze� Zdefdfdee d e d e fd d �Z defdfdee d e d e fd d!�Z d"ed#e e eee f fd$d%�Z!d"ed#e e eee f fd&d'�Z"d#efd(d)�Z#dS )*uf 错误处理和容错机制模块6 7提供统一的错误处理、重试逻辑和异常恢复功能。8� N)�Any�Callable�Dict�Optional�Type�Union)�Enum� )�9get_config)�get_task_loggerc @ s` e Zd ZdZdZdZdZdZdZdZ dZ10d Z d11Z d Z
d Zd
ZdZdZdZdZdZdZdZdZdS )� ErrorCodeu 错误代码ZFILE_001ZFILE_002ZFILE_003ZFILE_004ZNET_001ZNET_002ZNET_003ZAPI_001ZAPI_002ZAPI_003ZAPI_004ZOSS_001ZOSS_002ZOSS_003ZSYS_001ZSYS_002ZSYS_003ZGEN_001ZGEN_002ZGEN_003N)�__name__�12__module__� __qualname__�__doc__�FILE_NOT_FOUNDZFILE_TOO_LARGEZFILE_FORMAT_UNSUPPORTEDZFILE_CORRUPTED�NETWORK_TIMEOUT�NETWORK_CONNECTION_ERRORZNETWORK_DNS_ERROR�API_KEY_INVALID�API_QUOTA_EXCEEDED�API_SERVICE_UNAVAILABLE�API_RATE_LIMITEDZOSS_ACCESS_DENIEDZOSS_BUCKET_NOT_FOUND�OSS_UPLOAD_FAILED�SYSTEM_OUT_OF_MEMORYZSYSTEM_DISK_FULL�SYSTEM_PERMISSION_DENIED�
UNKNOWN_ERRORZ
TIMEOUT_ERROR�VALIDATION_ERROR� r r �H/Users/chengpeng/MyProject/transcript_service/src/utils/error_handler.pyr s, r c sL e Zd ZdZejdfdededef� fdd�
Zdeee f fd d13�Z14� Z S ) �TranscriptServiceErroru 服务自定义异常基类N�message�15error_code�detailsc s0 t � �|� || _|| _|pi | _t�� | _dS )u� 初始化异常16 17 Args:18 message: 错误消息19 error_code: 错误代码20 details: 额外详情21 N)�super�__init__r r! r" �time� timestamp)�selfr r! r" �� __class__r r r$ 6 s22 23z TranscriptServiceError.__init__�returnc C s | j j| j| j| jd�S )u 转换为字典格式)r! r r" r& )r! �valuer r" r& �r' r r r �to_dictD s24 �z TranscriptServiceError.to_dict) r
r r r r r �strr r$ r r- �
__classcell__r r r( r r 3 s " r c @ � e Zd ZdZdS )�FileValidationErroru 文件验证错误N�r
r r r r r r r r1 N � r1 c @ r0 )� NetworkErroru 网络相关错误Nr2 r r r r r4 S r3 r4 c @ r0 )�APIErroru API调用错误Nr2 r r r r r5 X r3 r5 c @ r0 )�OSSErroru OSS操作错误Nr2 r r r r r6 ] r3 r6 c @ r0 )� SystemErroru 系统错误Nr2 r r r r r7 b r3 r7 c @ sL e Zd ZdZ ddeded ed25ed ef26d d
�Zdedefdd�ZdS )�
RetryStrategyu 重试策略� � �?� N@� @T� max_attempts�27base_delay� max_delay�exponential_base�jitterc C s" || _ || _|| _|| _|| _dS )u' 初始化重试策略28 29 Args:30 max_attempts: 最大重试次数31 base_delay: 基础延迟时间（秒）32 max_delay: 最大延迟时间（秒）33 exponential_base: 指数退避基数34 jitter: 是否添加随机抖动35 N)r= r> r? r@ rA )r' r= r> r? r@ rA r r r r$ j s36 37zRetryStrategy.__init__�attemptr* c C sF | j | j|d }t|| j�}| jr!ddl}|d|�� d 9 }|S )u� 计算延迟时间38 39 Args:40 attempt: 当前尝试次数（从1开始）41 42 Returns:43 延迟时间（秒）44 � r Ng �?)r> r@ �minr? rA �random)r' rB �delayrE r r r �calculate_delay� s z RetryStrategy.calculate_delayN)r9 r: r; r< T) r
r r r �int�float�boolr$ rG r r r r r8 g s& �����45�r8 c @ sT e Zd ZdZdd� Zdedefdd�Zdedefdd �Z dded e46defd d
�Z dS )� ErrorHandleru 错误处理器c
C sv t � | _tdd�| _tttjftt tj47ft j t
tjftt
tjftttjftttjfi| _tjtjtjtjtjh| _dS )u 初始化错误处理器ztranscript_service.error�� logger_nameN)r48 �configr �logger�FileNotFoundErrorr1 r r �PermissionErrorr7 r �asyncio� TimeoutErrorr4 r �ConnectionErrorr �49ValueErrorr r � RuntimeErrorr �
error_mappingr r r �retryable_errorsr, r r r r$ � s 5051 525354�55�zErrorHandler.__init__�errorr* c C s� t |t�r|S t|�}|| jv r | j| \}}|t|�|�S t|��� }d|v r0tt|�tj�S d|v r<t t|�tj56�S d|v rHt t|�tj �S d|v sPd|v rXt t|�tj
�S tt|�tj�S )u� 分类和包装错误57 58 Args:59 error: 原始异常60 61 Returns:62 分类后的服务异常63 �timeoutzpermission deniedzapi keyZquota�limit)�64isinstancer �typerW r. �lowerr4 r r r7 r r5 r r r )r' rY �65error_typeZexception_classr! Z error_msgr r r �classify_error� s 66 67 zErrorHandler.classify_errorc C s |j | jv S )u� 判断错误是否可重试68 69 Args:70 error: 服务异常71 72 Returns:73 是否可重试74 )r! rX )r' rY r r r � is_retryable� s zErrorHandler.is_retryable� �contextc C sN | � |�}d|� d|j� �}|jtjtjfv r | j�|� |S | j�|� |S )u� 处理错误75 76 Args:77 error: 原始异常78 context: 错误上下文79 80 Returns:81 处理后的服务异常82 u 错误处理 - z: ) r` r r! r r r rO � exceptionrY )r' rY rc �classified_errorZlog_msgr r r � handle_error� s 83 84 �zErrorHandler.handle_errorN)rb ) r
r r r r$ � Exceptionr r` rJ ra r. rf r r r r rK � s rK rb �strategy�85exceptionsrc c �( �du rt � �dtf� ��fdd� }|S )u� 异步函数重试装饰器86 87 Args:88 strategy: 重试策略89 exceptions: 需要重试的异常类型90 context: 上下文信息91 N�funcc � t �� ���� �fdd��}|S )Nc � s� �t dd�}td�jd �D ]e}z
�| i |��I d H W S �ys } zJt�|�}|�jks4t�|�sI|�� � d|� d�j� dt|�� �� |���|�}|� � � d|� d|d �d92t|�� �� t93� |�I d H W Y d }~qd }~ww t d t
j��� Nztranscript_service.retryrL rC u 最终失败 (尝试 �/z): u 第 u 次尝试失败，z.1fu 秒后重试: u 重试逻辑异常)r �ranger= �
error_handlerr` ra rY r. rG �warningrR �sleepr r r ��args�kwargsrO rB �ere rF �rc ri rk rh r r �wrapper s �9495&96& �� z/retry_async.<locals>.decorator.<locals>.wrapper�� functools�wraps�rk rx �rc ri rh �rk r � decorator � z retry_async.<locals>.decorator�r8 r �rh ri rc r r r} r � retry_async� � r� c rj )u� 同步函数重试装饰器97 98 Args:99 strategy: 重试策略100 exceptions: 需要重试的异常类型101 context: 上下文信息102 Nrk c rl )Nc s� t dd�}td�jd �D ]_}z103�| i |��W S �yl } zGt�|�}|�jks0t�|�sE|�� � d|� d�j� dt|�� �� |���|�}|� � � d|� d|d �d104t|�� �� t105� |� W Y d }~q
d }~ww t d t
j��rm )r ro r= rp r` ra rY r. rG rq r% rr r r r rs rw r r rx 4 s 106107&108&�� z.retry_sync.<locals>.decorator.<locals>.wrapperry r| r} r~ r r 3 r� z retry_sync.<locals>.decoratorr� r� r r} r �109retry_sync$ r� r� rk r* c 110 O s^ z
| |i |��}d|dfW S t y. } zt�|d| j� ��}dd|fW Y d}~S d}~ww )u� 安全执行函数111 112 Args:113 func: 要执行的函数114 *args: 位置参数115 **kwargs: 关键字参数116 117 Returns:118 (是否成功, 结果或None, 错误或None)119 TN� 执行 F�rg rp rf r
�rk rt ru �resultrv rY r r r � safe_executeP s ��r� c 120 � sf �z| |i |��I dH }d|dfW S t y2 } zt�|d| j� ��}dd|fW Y d}~S d}~ww )u� 安全执行异步函数121 122 Args:123 func: 要执行的异步函数124 *args: 位置参数125 **kwargs: 关键字参数126 127 Returns:128 (是否成功, 结果或None, 错误或None)129 NTr� Fr� r� r r r �safe_execute_asyncc s � ��r� c C s t S )uP 获取错误处理器实例130 131 Returns:132 错误处理器实例133 )rp r r r r �get_error_handlerv s r� )$r rR rz r% �typingr r r r r r �enumr � core.configr134 � utils.loggerr r rg r r1 r4 r5 r6 r7 r8 rK rp �tupler. r� r� rJ r� r� r� r r r r �<module> sR #-a���135�-���136� ,