package com.ruoyi.datacollection.config;
|
|
import com.ruoyi.common.core.service.IConfigService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.ObjectProvider;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
/**
|
* 数据采集模块<b>总开关</b>：值存放在 {@code sys_config} 参数配置表中，<b>运行时读取、可热切换</b>。
|
*
|
* <h3>参数配置</h3>
|
* <pre>
|
* 参数键名：datacollection.enabled
|
* 参数键值：true = 启用（默认，等价于不配置这一行）
|
* false = 停采
|
* </pre>
|
* 位置：系统管理 → 参数设置 → 新增/修改。
|
* <pre>
|
* insert into sys_config (config_name, config_key, config_value, config_type, create_by, create_time, remark)
|
* values ('数据采集模块开关', 'datacollection.enabled', 'true', 'Y', 'admin', sysdate(), 'true启用，false停采（默认true）');
|
* </pre>
|
*
|
* <h3>取值优先级</h3>
|
* <ol>
|
* <li>{@code sys_config} 里 {@code datacollection.enabled} 的值（<b>主用</b>，
|
* 来自 {@code IConfigService} 接口，实现在 ruoyi-system 的 {@code SysConfigServiceImpl}）；</li>
|
* <li>{@link DataCollectionProperties#isEnabled()}（<b>兜底</b>：仅当 sys_config 里没有这一行、
|
* 或非 Spring 环境取不到配置服务时使用，默认 true）。</li>
|
* </ol>
|
*
|
* <h3>为什么读不到配置时按 true（fail-open）</h3>
|
* 读 {@code sys_config} 需要走数据库，数据库抖动时如果 fail-closed，
|
* 结果就是"数据库抖一下，当天 HIS 数据全都不采了" —— 静默少数据。
|
* 这里只打 WARN 并放行。要停采必须显式配置 false。
|
*
|
* <h3>为什么有 10 秒缓存</h3>
|
* {@code SysConfigServiceImpl#selectConfigByKey} 当前每次都查库，而本开关在每轮定时采集
|
* 开始时都会被读到。这里加一个进程内 TTL 缓存，把查库降到"每个实例每 10 秒最多一次"。
|
* 代价是参数改动最多 10 秒生效（多实例部署时各自计时）。
|
*/
|
@Slf4j
|
@Component
|
public class DataCollectionSwitch {
|
|
/**
|
* sys_config 里的参数键名
|
*/
|
public static final String CONFIG_KEY = "datacollection.enabled";
|
|
/**
|
* 进程内缓存有效期（毫秒）
|
*/
|
private static final long CACHE_TTL_MILLIS = 10_000L;
|
|
private final DataCollectionProperties properties;
|
|
/**
|
* 参数配置服务（ruoyi-common 的接口，实现在 ruoyi-system）。
|
* <p>
|
* 用 {@code ObjectProvider} 惰性取：非 Spring 环境/独立部署等"容器里没有实现"的场景下
|
* 不能因此启动失败。
|
*/
|
private final ObjectProvider<IConfigService> configServiceProvider;
|
|
/**
|
* 最近一次解析结果
|
*/
|
private volatile Boolean cachedEnabled;
|
|
/**
|
* 最近一次解析的时间戳
|
*/
|
private volatile long cachedAt;
|
|
/**
|
* Spring 使用的构造器。
|
* <p>
|
* ⚠️ <b>必须显式标注 {@code @Autowired}</b>：本类有两个公开构造器
|
* （另一个给测试/非 Spring 环境用），不标注的话 Spring 会因"构造器不唯一"
|
* 退回查找无参构造器，启动即报 {@code No default constructor found}。
|
* 该规则已由 {@code sql/_probe/verify_spring_beans.py} 静态把关。
|
*/
|
@Autowired
|
public DataCollectionSwitch(DataCollectionProperties properties,
|
ObjectProvider<IConfigService> configServiceProvider) {
|
this.properties = properties;
|
this.configServiceProvider = configServiceProvider;
|
}
|
|
/**
|
* 供<b>测试/非 Spring 环境</b>直接构造：{@code configValue} 就是 sys_config 里的键值，
|
* 传 {@code null} 表示"参数行不存在"（此时回落到 {@link DataCollectionProperties}）。
|
*/
|
public DataCollectionSwitch(DataCollectionProperties properties, final String configValue) {
|
this(properties, new FixedProvider(configValue));
|
// 非 Spring 环境下没有"改配置后等 10 秒"的概念，直接禁用缓存，保证断言即时可见
|
this.cachedEnabled = resolve();
|
this.cachedAt = System.currentTimeMillis() + CACHE_TTL_MILLIS * 1000L;
|
}
|
|
/**
|
* 数据采集模块是否启用（运行时热读，最多 10 秒延迟）
|
*/
|
public boolean isEnabled() {
|
long now = System.currentTimeMillis();
|
Boolean cached = this.cachedEnabled;
|
if (cached != null && now - this.cachedAt < CACHE_TTL_MILLIS) {
|
return cached;
|
}
|
boolean resolved = resolve();
|
this.cachedEnabled = resolved;
|
this.cachedAt = now;
|
return resolved;
|
}
|
|
/**
|
* 清掉进程内缓存，下次调用立即重新读 sys_config（供测试/需要立即生效的场景）
|
*/
|
public void invalidateCache() {
|
this.cachedEnabled = null;
|
}
|
|
/**
|
* 解析当前开关值：sys_config 优先，其次 yml 兜底
|
*/
|
private boolean resolve() {
|
String value = readFromSysConfig();
|
if (value != null) {
|
if (!"true".equalsIgnoreCase(value) && !"false".equalsIgnoreCase(value)) {
|
log.warn("【数据采集】sys_config 中 {} 的值不是 true/false（实际=「{}」），将按「不是 false 即启用」处理",
|
CONFIG_KEY, value);
|
}
|
boolean enabled = !"false".equalsIgnoreCase(value);
|
log.debug("【数据采集】开关取自 sys_config：{}={} -> enabled={}", CONFIG_KEY, value, enabled);
|
return enabled;
|
}
|
boolean enabled = properties.isEnabled();
|
log.debug("【数据采集】sys_config 中未配置 {}，回落 datacollection.enabled={}", CONFIG_KEY, enabled);
|
return enabled;
|
}
|
|
/**
|
* 读 sys_config 的原始键值；没有实现、没有这一行、或读库失败都返回 {@code null}
|
*/
|
private String readFromSysConfig() {
|
if (configServiceProvider == null) {
|
return null;
|
}
|
try {
|
IConfigService configService = configServiceProvider.stream().findFirst().orElse(null);
|
if (configService == null) {
|
return null;
|
}
|
String value = configService.selectConfigByKey(CONFIG_KEY);
|
if (value == null || value.trim().isEmpty()) {
|
return null;
|
}
|
return value.trim();
|
} catch (Exception e) {
|
// 容器没有实现、数据源异常、SQL 报错……一律按"参数缺失"处理（fail-open），绝不影响采集主流程
|
log.warn("【数据采集】读取 sys_config 的 {} 失败，本次按未配置处理（fail-open）：{}",
|
CONFIG_KEY, e.toString());
|
return null;
|
}
|
}
|
|
/**
|
* 固定值的 {@code ObjectProvider}：只用于非 Spring 环境/测试
|
*/
|
private static final class FixedProvider implements ObjectProvider<IConfigService> {
|
|
private final IConfigService configService;
|
|
private FixedProvider(final String configValue) {
|
this.configService = configValue == null ? null : configKey -> configValue;
|
}
|
|
@Override
|
public IConfigService getObject(Object... args) {
|
return configService;
|
}
|
|
@Override
|
public IConfigService getIfAvailable() {
|
return configService;
|
}
|
|
@Override
|
public IConfigService getIfUnique() {
|
return configService;
|
}
|
|
@Override
|
public IConfigService getObject() {
|
return configService;
|
}
|
|
@Override
|
public java.util.stream.Stream<IConfigService> stream() {
|
return configService == null ? java.util.stream.Stream.empty() : java.util.stream.Stream.of(configService);
|
}
|
}
|
}
|
