package com.zy.ai.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zy.ai.entity.LlmCallLog;
import com.zy.ai.mapper.LlmCallLogMapper;
import com.zy.ai.service.LlmCallLogService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

@Service("llmCallLogService")
@Slf4j
public class LlmCallLogServiceImpl extends ServiceImpl<LlmCallLogMapper, LlmCallLog> implements LlmCallLogService {

    private volatile boolean disabled = false;
    private volatile long lastRetryTime = 0;
    private static final long RETRY_INTERVAL_MS = 60_000; // 1 分钟后重试

    @Override
    public void saveIgnoreError(LlmCallLog logItem) {
        if (logItem == null) {
            return;
        }
        if (disabled) {
            // 定期重试，防止表后来创建了但 disabled 一直为 true
            long now = System.currentTimeMillis();
            if (now - lastRetryTime < RETRY_INTERVAL_MS) {
                return;
            }
            lastRetryTime = now;
            log.info("LLM调用日志之前已禁用，尝试重新写入...");
        }
        try {
            save(logItem);
            if (disabled) {
                disabled = false;
                log.info("LLM调用日志写入成功，已恢复日志记录");
            }
        } catch (Exception e) {
            String msg = e.getMessage() == null ? "" : e.getMessage();
            if (msg.contains("doesn't exist") || msg.contains("不存在")) {
                disabled = true;
                lastRetryTime = System.currentTimeMillis();
                log.warn("LLM调用日志表不存在，日志记录已暂停，请先执行建表SQL");
                return;
            }
            log.warn("写入LLM调用日志失败: {}", msg);
        }
    }
}