package jnpf.audit.sdk;
|
|
import cn.hutool.http.useragent.UserAgent;
|
import cn.hutool.http.useragent.UserAgentUtil;
|
import jnpf.audit.AuditApiConsts;
|
import jnpf.audit.model.AuditEventDTO;
|
import jnpf.base.UserInfo;
|
import jnpf.util.IpUtil;
|
import jnpf.util.RandomUtil;
|
import jnpf.util.ServletUtil;
|
import jnpf.util.UserProvider;
|
import org.springframework.web.context.request.RequestContextHolder;
|
|
import java.util.Date;
|
|
/**
|
* 审计上下文采集器（Task 7 Step 1）：只填 DTO 的空缺字段，不覆盖调用方已给值。
|
*
|
* <p>操作人/租户取自 {@link UserProvider}（Sa-Token 登录上下文）；IP/浏览器/OS 取自请求上下文，
|
* 工具与平台 {@code jnpf.RequestLogAspect} 同款（{@link IpUtil#getIpAddr()} / {@link IpUtil#getIpCity(String)} /
|
* {@link ServletUtil#getUserAgent()} + hutool {@link UserAgentUtil#parse(String)}）。
|
*
|
* <p>关键约束（终验#6）：operationId **不由本采集器生成、也不写 MDC**——生成与 MDC 写入统一归
|
* {@link AuditTxHolder}；采集器若先写 MDC，会让 holder 读到的 previousMdc 变成新值本身，恢复等于没恢复。
|
*/
|
public class AuditContextCollector {
|
|
/** 无登录上下文时的操作人占位（异步/系统触发链路） */
|
private static final String OPERATOR_SYSTEM = "SYSTEM";
|
|
private final String appName;
|
|
public AuditContextCollector(String appName) {
|
this.appName = appName;
|
}
|
|
/** 填充空缺字段并返回同一 dto（供 AuditClient 冻结后交 holder 缓冲）。 */
|
public AuditEventDTO fill(AuditEventDTO dto) {
|
if (isBlank(dto.getClientEventId())) {
|
dto.setClientEventId(RandomUtil.uuId()); // 平台雪花（= String.valueOf(YitIdHelper.nextId())）
|
}
|
if (dto.getEventTime() == null) {
|
dto.setEventTime(new Date());
|
}
|
if (isBlank(dto.getAppName())) {
|
dto.setAppName(appName);
|
}
|
fillOperator(dto);
|
fillRequest(dto);
|
return dto;
|
}
|
|
/**
|
* 操作人/租户：有登录用户则填 userId/userName/organizeId/tenantId；无则 operator=SYSTEM。
|
*
|
* <p>租户回落（M2 阻断项修复）：单租户部署下 {@code user.getTenantId()} 返回**空串**，
|
* 无会话链路更是无从取得——两种情况都在最后统一回落到 {@link AuditApiConsts#DEFAULT_TENANT_ID}，
|
* 否则服务端 validate 会以「tenantId 不能为空」拒收，事件进 %RETRY% 直至 DLQ。
|
* 回落只在空白时发生，多租户部署下真实租户号不受影响。
|
*/
|
private void fillOperator(AuditEventDTO dto) {
|
UserInfo user = currentUser();
|
if (user != null && !isBlank(user.getUserId())) {
|
if (isBlank(dto.getOperatorId())) {
|
dto.setOperatorId(user.getUserId());
|
}
|
if (isBlank(dto.getOperatorName())) {
|
dto.setOperatorName(user.getUserName());
|
}
|
if (isBlank(dto.getOperatorOrgId())) {
|
dto.setOperatorOrgId(user.getOrganizeId());
|
}
|
if (isBlank(dto.getTenantId())) {
|
dto.setTenantId(user.getTenantId());
|
}
|
} else if (isBlank(dto.getOperatorId())) {
|
// 无登录上下文：operator=SYSTEM（tenantId 由下方哨兵回落兜底）
|
dto.setOperatorId(OPERATOR_SYSTEM);
|
}
|
if (isBlank(dto.getTenantId())) {
|
dto.setTenantId(AuditApiConsts.DEFAULT_TENANT_ID);
|
}
|
}
|
|
/** IP/浏览器/OS：仅在有请求上下文时采集，best-effort，采集失败静默降级不影响审计主流程。 */
|
private void fillRequest(AuditEventDTO dto) {
|
if (RequestContextHolder.getRequestAttributes() == null) {
|
return; // 无请求上下文（异步/系统触发）
|
}
|
try {
|
if (isBlank(dto.getIp())) {
|
String ip = IpUtil.getIpAddr();
|
dto.setIp(ip);
|
if (isBlank(dto.getIpRegion())) {
|
dto.setIpRegion(IpUtil.getIpCity(ip));
|
}
|
}
|
String ua = ServletUtil.getUserAgent();
|
if (!isBlank(ua)) {
|
UserAgent agent = UserAgentUtil.parse(ua);
|
if (isBlank(dto.getBrowser())) {
|
dto.setBrowser(join(agent.getBrowser().getName(), agent.getVersion()));
|
}
|
if (isBlank(dto.getOs())) {
|
dto.setOs(join(agent.getPlatform().getName(), agent.getOsVersion()));
|
}
|
}
|
} catch (Exception ignore) {
|
// 上下文采集属 best-effort，任何异常不得冒泡影响业务与审计主链路
|
}
|
}
|
|
private static UserInfo currentUser() {
|
try {
|
return UserProvider.getUser();
|
} catch (Exception e) {
|
return null; // 未登录/无 token（Sa-Token 可能抛异常），视为无用户
|
}
|
}
|
|
private static String join(String a, String b) {
|
String x = a == null ? "" : a.trim();
|
String y = b == null ? "" : b.trim();
|
String r = (x + " " + y).trim();
|
return r.isEmpty() ? null : r;
|
}
|
|
private static boolean isBlank(String s) {
|
return s == null || s.trim().isEmpty();
|
}
|
}
|
