security: 接口鉴权 + 跨域收紧 + AI 数据脱敏 + 日志脱敏

zhizhijie

2026-09-14 d1cfd849c6a61800dcd86ffb45f80882ae20b2e7

 traffic-audit-server/src/main/java/com/trafficaudit/llmintegration/service/DesensitizeService.java

@@ -7,6 +7,10 @@
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 企业名称脱敏：把企业名称替换为 Company1、Company2… 后发给大模型，回答返回前再还原。
 * 仅保存在内存中，不落库。
 */
@Slf4j
@Service
public class DesensitizeService {
@@ -16,6 +20,9 @@
    private final AtomicInteger counter = new AtomicInteger(1);

    public String maskEnterprise(String name) {
        if (name == null) {
            return null;
        }
        return enterpriseMap.computeIfAbsent(name, k -> {
            String alias = "Company" + counter.getAndIncrement();
            reverseMap.put(alias, name);
@@ -27,9 +34,23 @@
        return reverseMap.getOrDefault(alias, alias);
    }

    /** 把整段文本里出现过的脱敏别名还原为真实企业名称（回答返回给前端前调用） */
    public String unmaskAll(String text) {
        if (text == null || text.isEmpty() || reverseMap.isEmpty()) {
            return text;
        }
        String out = text;
        for (Map.Entry<String, String> e : reverseMap.entrySet()) {
            if (out.contains(e.getKey())) {
                out = out.replace(e.getKey(), e.getValue());
            }
        }
        return out;
    }

    public void clear() {
        enterpriseMap.clear();
        reverseMap.clear();
        counter.set(1);
    }
}
}