#  Copyright 2026 zhaoxi826
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.


import asyncio
from functools import wraps
from kilostar.utils.error import RetryableError


def retry_on_retryable_error(max_retries=3, base_delay=1):
    """执行与 retry on retryable error 相关的核心业务流转操作。
    该方法封装了具体的算法策略或状态控制逻辑，确保操作能够在事务上下文中被原子且一致地执行。
    Args:        max_retries: 参与 retry on retryable error 逻辑运算或数据构建的上下文依赖对象。        base_delay: 参与 retry on retryable error 逻辑运算或数据构建的上下文依赖对象。
    Returns:        : 经由当前业务模型加工处理后所输出的具体数据实例或领域模型对象。"""

    def decorator(func):
        """执行与 decorator 相关的核心业务流转操作。
        该方法封装了具体的算法策略或状态控制逻辑，确保操作能够在事务上下文中被原子且一致地执行。
        Args:        func: 参与 decorator 逻辑运算或数据构建的上下文依赖对象。
        Returns:        : 经由当前业务模型加工处理后所输出的具体数据实例或领域模型对象。"""
        if asyncio.iscoroutinefunction(func):

            @wraps(func)
            async def async_wrapper(*args, **kwargs):
                """执行与 async wrapper 相关的核心业务流转操作。
                该方法封装了具体的算法策略或状态控制逻辑，确保操作能够在事务上下文中被原子且一致地执行。
                Returns:        : 经由当前业务模型加工处理后所输出的具体数据实例或领域模型对象。"""
                for attempt in range(max_retries):
                    try:
                        return await func(*args, **kwargs)
                    except RetryableError:
                        if attempt == max_retries - 1:
                            raise
                        await asyncio.sleep(base_delay * (2**attempt))

            return async_wrapper
        else:

            @wraps(func)
            def sync_wrapper(*args, **kwargs):
                """执行与 sync wrapper 相关的核心业务流转操作。
                该方法封装了具体的算法策略或状态控制逻辑，确保操作能够在事务上下文中被原子且一致地执行。
                Returns:        : 经由当前业务模型加工处理后所输出的具体数据实例或领域模型对象。"""
                import time

                for attempt in range(max_retries):
                    try:
                        return func(*args, **kwargs)
                    except RetryableError:
                        if attempt == max_retries - 1:
                            raise
                        time.sleep(base_delay * (2**attempt))

            return sync_wrapper

    return decorator