import json
import logging
import time

from mqtt_pub import MQTTClient

# MQTT 配置
MQTT_BROKER = "8.137.54.85"
MQTT_PORT = 1883
MQTT_PUBLISH_TOPIC = "ai/tottle/uvmodule"
# MQTT_PUBLISH_TOPIC =  "ai/tottle/uvmodulecallback" # 回调topic
# MQTT_CLIENT_ID = "aitottle0123456789"

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

def send_mqtt_uv_request(task_id,s3_id,s3_url,model_func_id):
    mqtt_client=None
    try:
        mqtt_client = MQTTClient(MQTT_BROKER, MQTT_PORT, MQTT_PUBLISH_TOPIC)
    except Exception as e:
        logger.error(f"启动AI视频处理失败: {e}")
        raise Exception(f"send_mqtt_uv_request connect to mqtt failed: {str(e)}", status_code=500)



    # # 启动生产者线程
    # producer = threading.Thread(target=producer_thread)
    # producer.start()

    # # 启动消费者线程
    # consumer = threading.Thread(target=consumer_thread, args=(mqtt_client,))
    # consumer.start()

    # # 等待生产者线程完成
    # producer.join()
    if mqtt_client is not  None:
        request_data={
            "task_id":task_id,
            "s3_id":s3_id,
            "list_s3_url":s3_url,
            "list_func_id":model_func_id,
        }
        payload = json.dumps(request_data, ensure_ascii=False)
        mqtt_client.publish_message(payload)
        # 等待队列中的所有消息被处理
        # message_queue.join()
        time.sleep(0.5)
        # 关闭 MQTT 客户端
        # mqtt_client.close()
        # time.sleep(5)
        print("程序结束")
    else:
        print("send_mqtt_uv_request 方法执行失败")