package cn.iocoder.yudao.module.mes.service.trace.consumer;
|
|
import cn.hutool.core.util.StrUtil;
|
import cn.hutool.crypto.digest.DigestUtil;
|
import cn.iocoder.yudao.framework.common.util.servlet.ServletUtils;
|
import cn.iocoder.yudao.framework.ip.core.Area;
|
import cn.iocoder.yudao.framework.ip.core.enums.AreaTypeEnum;
|
import cn.iocoder.yudao.framework.ip.core.utils.IPUtils;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.trace.consumer.MesTraceConsumerScanEventDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.trace.consumer.MesTraceConsumerScanEventMapper;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
import java.time.LocalDateTime;
|
|
/**
|
* 消费者扫码访问事件 Service 实现
|
*
|
* @author 超级管理员
|
*/
|
@Service
|
@Slf4j
|
public class MesTraceConsumerScanEventServiceImpl implements MesTraceConsumerScanEventService {
|
|
private static final int TRACE_CODE_MAX_LENGTH = 255;
|
private static final int NORMALIZED_CODE_MAX_LENGTH = 128;
|
|
@Resource
|
private MesTraceConsumerScanEventMapper eventMapper;
|
|
@Override
|
public MesTraceConsumerScanEventDO recordScan(String traceCode, String normalizedCode, String sourceType,
|
Long batchId, boolean success, String errorCode) {
|
try {
|
MesTraceConsumerScanEventDO event = new MesTraceConsumerScanEventDO();
|
event.setTraceCode(limit(traceCode, TRACE_CODE_MAX_LENGTH));
|
event.setNormalizedCode(limit(normalizedCode, NORMALIZED_CODE_MAX_LENGTH));
|
event.setSourceType(sourceType);
|
event.setChannel(limit(channel(), 32));
|
String[] region = region(ServletUtils.getClientIP());
|
event.setProvince(limit(region[0], 64));
|
event.setCity(limit(region[1], 64));
|
event.setBatchId(batchId);
|
event.setSuccess(success);
|
event.setErrorCode(limit(errorCode, 64));
|
event.setIpHash(hash(ServletUtils.getClientIP()));
|
event.setUserAgentHash(hash(ServletUtils.getUserAgent()));
|
event.setEventTime(LocalDateTime.now());
|
eventMapper.insert(event);
|
return event;
|
} catch (Exception ex) {
|
log.warn("[recordScan][记录消费者扫码访问事件失败] traceCode={}", limit(traceCode, 64), ex);
|
return null;
|
}
|
}
|
|
@Override
|
public MesTraceConsumerScanEventDO getEvent(Long id) {
|
return eventMapper.selectById(id);
|
}
|
|
/**
|
* 根据 User-Agent 识别扫码入口渠道
|
*/
|
private String channel() {
|
String ua = ServletUtils.getUserAgent();
|
if (StrUtil.isBlank(ua)) {
|
return null;
|
}
|
String lower = ua.toLowerCase();
|
if (lower.contains("micromessenger")) {
|
return "微信";
|
}
|
if (lower.contains("alipayclient")) {
|
return "支付宝";
|
}
|
return "浏览器";
|
}
|
|
/**
|
* 根据客户端 IP 解析省份、城市；解析失败返回空，不影响事件记录
|
*/
|
private String[] region(String ip) {
|
if (StrUtil.isBlank(ip)) {
|
return new String[]{null, null};
|
}
|
Area area = IPUtils.getArea(ip);
|
if (area == null) {
|
return new String[]{null, null};
|
}
|
String province = null;
|
String city = null;
|
Area node = area;
|
while (node != null) {
|
Integer type = node.getType();
|
if (AreaTypeEnum.PROVINCE.getType().equals(type)) {
|
province = node.getName();
|
} else if (AreaTypeEnum.CITY.getType().equals(type)) {
|
city = node.getName();
|
}
|
node = node.getParent();
|
}
|
return new String[]{province, city};
|
}
|
|
private String hash(String value) {
|
return value == null || value.isBlank() ? null : DigestUtil.sha256Hex(value);
|
}
|
|
private String limit(String value, int maxLength) {
|
if (value == null || value.length() <= maxLength) {
|
return value;
|
}
|
return value.substring(0, maxLength);
|
}
|
|
}
|
