Bucket4j API未提供上次消费时间，如何清理闲置用户桶？

Bucket4j原生并未提供直接获取上次令牌消费时间的API，因为它的核心设计专注于限流逻辑实现，没有将该时间状态作为对外暴露的内容。不过你可以通过以下两种方案实现需求，无需完全自行重构数据结构：

方案一：包装Bucket，自行维护消费时间 #

创建一个自定义包装类，内部持有Bucket实例和原子类型的时间字段，每次调用消费令牌的方法时更新时间：

public class UserBucket {
    private final Bucket bucket;
    private final AtomicLong lastConsumeTimestamp;

    public UserBucket(Bucket bucket) {
        this.bucket = bucket;
        this.lastConsumeTimestamp = new AtomicLong(System.currentTimeMillis());
    }

    // 代理Bucket的令牌消费方法，成功消费时更新时间
    public boolean tryConsume(long tokens) {
        boolean success = bucket.tryConsume(tokens);
        if (success) {
            lastConsumeTimestamp.set(System.currentTimeMillis());
        }
        return success;
    }

    // 按需代理其他Bucket方法（如consume、tryConsumeAndReturnRemaining等）
    public ConsumptionProbe tryConsumeAndReturnRemaining(long tokens) {
        ConsumptionProbe probe = bucket.tryConsumeAndReturnRemaining(tokens);
        if (probe.isConsumed()) {
            lastConsumeTimestamp.set(System.currentTimeMillis());
        }
        return probe;
    }

    // 获取上次消费时间戳
    public long getLastConsumeTimestamp() {
        return lastConsumeTimestamp.get();
    }

    // 暴露原始Bucket（若有需要）
    public Bucket getRawBucket() {
        return bucket;
    }
}

之后将你的存储结构改为ConcurrentHashMap<String, UserBucket>，定期遍历Map时，通过getLastConsumeTimestamp()判断是否需要删除超过5分钟未消费的条目。

方案二：利用Bucket4j的监听机制 #

如果使用的Bucket4j版本支持（2.x及以上），可以给每个Bucket绑定BucketListener，在令牌成功消费时触发回调，维护单独的时间存储表：

// 单独存储用户最后消费时间
private final ConcurrentHashMap<String, AtomicLong> userLastConsumeTime = new ConcurrentHashMap<>();

// 创建用户Bucket时绑定监听器
public Bucket createUserBucket(String userId) {
    Bucket bucket = Bucket.builder()
            .addLimit(Bandwidth.simple(10, Duration.ofMinutes(1))) // 示例限流规则
            .build();
    
    bucket.addListener(new BucketListener() {
        @Override
        public void onConsumed(long tokens) {
            // 更新该用户的最后消费时间
            userLastConsumeTime.computeIfAbsent(userId, k -> new AtomicLong()).set(System.currentTimeMillis());
        }
    });
    
    return bucket;
}

这种方式无需包装整个Bucket，但需要额外维护一份时间映射表，清理时直接遍历这个表，再同步删除ConcurrentHashMap中的对应Bucket条目即可。

补充建议 #

• 定期清理时，建议使用ScheduledExecutorService定时执行（比如每分钟一次），避免频繁遍历开销；
• 遍历大体积ConcurrentHashMap时，优先使用forEach分段遍历，减少锁竞争，避免主线程阻塞；
• 清理逻辑中，判断时间时用System.currentTimeMillis() - 5 * 60 * 1000作为阈值，过滤超时条目。

内容的提问来源于stack exchange，提问作者Deibys