import time
import logging
from server.chat import utils
from fastapi.responses import JSONResponse

def stop(message_id: str):
    try:
        # start_time = time.time()
        # tatus = None
        # # 重试十秒
        # while time.time() - start_time < 10:
        status = utils.get_shared_variable(message_id)
        #     if status is not None:
        #         break
        #     logging.info(f"status未赋值, messageId: {message_id}")
        #     time.sleep(0.5)  # 等待0.5秒后重试

        if status is None:
            return JSONResponse(content={"error": "未找到messegeId", "messageId": message_id}, status_code=500)
        
        status["status"] = False
        utils.set_shared_variable(message_id, status)
        logging.info(f"停止请求成功，messageId: {message_id}")
        return JSONResponse(content={"event": "请求已成功停止", "messageId": message_id})
    except Exception as e:
        return JSONResponse(content={"error": str(e),"messageId": message_id}, status_code=500)