package com.trafficaudit.llmintegration.service;
|
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.atomic.AtomicInteger;
|
|
@Slf4j
|
@Service
|
public class DesensitizeService {
|
|
private final Map<String, String> enterpriseMap = new ConcurrentHashMap<>();
|
private final Map<String, String> reverseMap = new ConcurrentHashMap<>();
|
private final AtomicInteger counter = new AtomicInteger(1);
|
|
public String maskEnterprise(String name) {
|
return enterpriseMap.computeIfAbsent(name, k -> {
|
String alias = "Company" + counter.getAndIncrement();
|
reverseMap.put(alias, name);
|
return alias;
|
});
|
}
|
|
public String unmaskEnterprise(String alias) {
|
return reverseMap.getOrDefault(alias, alias);
|
}
|
|
public void clear() {
|
enterpriseMap.clear();
|
reverseMap.clear();
|
counter.set(1);
|
}
|
}
|