package cn.iocoder.yudao.server.framework.bizno;

import cn.iocoder.yudao.framework.common.util.number.BizNoUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;

import javax.sql.DataSource;
import java.util.List;
import java.util.Map;

/**
 * 已用业务编码集合兜底初始化。
 *
 * <p>业务编码为「随机 16 位纯数字」，唯一性靠 Redis 集合（{@link BizNoUtils#USED_SET_KEY}）查重。
 * 若 Redis 丢失该集合，已生成的历史编码会被"忘记"，之后可能被重新随机命中而重号。
 *
 * <p>因此：启动时若发现该集合 **不存在**，则从数据库中扫描所有业务编码列里的
 * 「16 位纯数字」编码，重新载入集合，恢复防重能力。
 *
 * <p>说明：
 * <ul>
 *     <li>仅在集合缺失时扫描（正常启动无额外开销）；</li>
 *     <li>best-effort，失败只告警、不影响启动；</li>
 *     <li>旧的带字母编码会被忽略。</li>
 * </ul>
 */
@Component
@Slf4j
public class BizCodeSetInitializer implements ApplicationRunner {

    /** 集合初始化标记（非 16 位，不会与真实编码冲突），用于在库内无编码时也标记为已初始化 */
    private static final String INIT_MARKER = "__initialized__";

    private final StringRedisTemplate stringRedisTemplate;
    private final JdbcTemplate jdbcTemplate;

    public BizCodeSetInitializer(StringRedisTemplate stringRedisTemplate, DataSource dataSource) {
        this.stringRedisTemplate = stringRedisTemplate;
        this.jdbcTemplate = new JdbcTemplate(dataSource);
    }

    @Override
    public void run(ApplicationArguments args) {
        if (Boolean.TRUE.equals(stringRedisTemplate.hasKey(BizNoUtils.USED_SET_KEY))) {
            log.info("[run][已用编码集合兜底] 集合已存在，跳过库内扫描");
            return;
        }
        long start = System.currentTimeMillis();
        long total = 0L;
        try {
            String schema = jdbcTemplate.queryForObject("SELECT DATABASE()", String.class);
            List<Map<String, Object>> columns = jdbcTemplate.queryForList(
                    "SELECT table_name, column_name FROM information_schema.columns " +
                            "WHERE table_schema = ? AND data_type IN ('varchar', 'char') " +
                            "AND (column_name = 'no' OR column_name = 'code' OR column_name = 'temp_code' " +
                            "     OR column_name LIKE '%\\_no' OR column_name LIKE '%\\_code')",
                    schema);
            for (Map<String, Object> column : columns) {
                String table = String.valueOf(column.get("table_name"));
                String col = String.valueOf(column.get("column_name"));
                if (table.toLowerCase().startsWith("qrtz_") || table.toLowerCase().startsWith("act_")) {
                    continue;
                }
                List<String> codes = jdbcTemplate.queryForList(
                        "SELECT `" + col + "` FROM `" + table + "` WHERE `" + col + "` REGEXP '^[0-9]{"
                                + BizNoUtils.CODE_LENGTH + "}$'",
                        String.class);
                if (!codes.isEmpty()) {
                    stringRedisTemplate.opsForSet().add(BizNoUtils.USED_SET_KEY, codes.toArray(new String[0]));
                    total += codes.size();
                }
            }
            // 标记为已初始化，避免下次启动重复扫描
            stringRedisTemplate.opsForSet().add(BizNoUtils.USED_SET_KEY, INIT_MARKER);
            log.info("[run][已用编码集合兜底] 集合缺失，已载入库内 {} 个编码", total);
        } catch (Exception e) {
            log.warn("[run][已用编码集合兜底] 初始化失败（不影响启动），请排查", e);
        } finally {
            log.info("[run][已用编码集合兜底] 库内扫描耗时 {} ms", System.currentTimeMillis() - start);
        }
    }

}