如何动态启用/禁用Bucket4J限流器及获取桶当前容量？

一、动态启用/禁用限流的优化方案 #

1. 使用官方无限带宽配置（推荐） #

Bucket4J 内置了 UnlimitedBandwidth 类，专门用于实现完全不限流的场景，比手动设置超大带宽更规范，还能避免数值过大引发的潜在问题。

修改禁用逻辑后的代码：

import io.github.bucket4j.UnlimitedBandwidth;

public void configure(int permitsPerSecond) {
    BucketConfiguration currentConfiguration;
    if (permitsPerSecond > 0) {
        // 启用限流
        Refill refill = Refill.intervally(permitsPerSecond, Duration.ofSeconds(1L));
        Bandwidth currentLimit = Bandwidth.classic(permitsPerSecond, refill);
        currentConfiguration = BucketConfiguration.builder()
                .addLimit(currentLimit)
                .build();
    } else {
        // 禁用限流：使用官方无限带宽实现
        currentConfiguration = BucketConfiguration.builder()
                .addLimit(UnlimitedBandwidth.INSTANCE)
                .build();
    }
    bucket.replaceConfiguration(currentConfiguration, TokensInheritanceStrategy.RESET);
}

2. 维护开关变量控制（临时启停更灵活） #

如果需要在不修改桶配置的前提下临时禁用限流，可以单独维护一个开关变量，实现更细粒度的控制：

private volatile boolean rateLimitEnabled = true; // 限流开关

public void setRateLimitEnabled(boolean enabled) {
    this.rateLimitEnabled = enabled;
}

public boolean allowExecution() {
    if (!rateLimitEnabled) {
        return true; // 禁用时直接放行所有请求
    }
    return bucket.tryConsume(1L);
}

这种方式无需频繁修改桶配置，适合临时启停限流的场景，可与配置式限流结合使用。

二、获取桶当前容量的方法 #

1. 获取当前可用令牌数（剩余容量） #

直接调用 Bucket 提供的 getAvailableTokens() 方法即可：

public long getCurrentAvailableTokens() {
    return bucket.getAvailableTokens();
}

2. 获取桶的最大容量（配置的总令牌数） #

若要获取桶的最大允许令牌数，需从当前配置中提取：

public long getBucketMaxCapacity() {
    BucketConfiguration config = bucket.getConfiguration();
    // 假设配置中只有一个带宽限制，实际可根据你的配置逻辑调整
    return config.getBandwidths().iterator().next().getCapacity();
}

三、补充说明 #

Bucket4J 本身没有提供直接的 bucket.enable(boolean) 方法，但通过上述两种方案可以实现等效的动态启用/禁用效果。其中 UnlimitedBandwidth 是官方推荐的无限制场景实现，开关变量法则更适合临时启停的需求。

内容的提问来源于stack exchange，提问作者Oscar Besga Panel