package com.vincent.rsf.httpaudit.service;

import com.vincent.rsf.httpaudit.mapper.HttpAuditConfigMapper;
import com.vincent.rsf.httpaudit.props.HttpAuditDbConfigHolder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Scheduled;

import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 审计配置刷新
 */
@Slf4j
public class HttpAuditDbConfigService {

    private final HttpAuditConfigMapper httpAuditConfigMapper;

    public HttpAuditDbConfigService(HttpAuditConfigMapper httpAuditConfigMapper) {
        this.httpAuditConfigMapper = httpAuditConfigMapper;
    }

    @PostConstruct
    public void init() {
        refresh();
    }

    @Scheduled(fixedDelayString = "${http-audit.db-config-refresh-ms:60000}")
    public void refresh() {
        try {
            List<Map<String, Object>> rows = httpAuditConfigMapper.listEnabledConfig();
            Map<String, String> map = new HashMap<>();
            for (Map<String, Object> row : rows) {
                Object keyObj = row.get("config_key");
                if (keyObj == null) {
                    continue;
                }
                String key = String.valueOf(keyObj).trim();
                if (key.isEmpty()) {
                    continue;
                }
                Object valObj = row.get("config_val");
                map.put(key, valObj == null ? "" : String.valueOf(valObj));
            }
            HttpAuditDbConfigHolder.replace(map);
        } catch (Exception e) {
            log.warn("http-audit 配置刷新失败，使用内存/默认配置", e);
        }
    }
}