日志速率限制器

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Function;
 
class Logger {
 
    // 使用 ConcurrentHashMap 存储每条消息和它最近一次打印的时间戳
    private ConcurrentHashMap<String, AtomicInteger> messageTimestampMap;
 
    /** 初始化 logger 对象 */
    public Logger() {
        messageTimestampMap = new ConcurrentHashMap<>();
    }
 
    /**
     * 判断是否应该打印消息
     * @param timestamp 当前时间戳（秒）
     * @param message 消息内容
     * @return 是否应该打印消息
     */
    public boolean shouldPrintMessage(int timestamp, String message) {
        return messageTimestampMap.compute(message, (key, value) -> {
            if (value == null || timestamp - value.get() >= 10) {
                return new AtomicInteger(timestamp);
            }
            return value;
        }).get() == timestamp;
    }
}