AnnotatedRateLimiterFactory

@BConfiguration
public class ProxyManagerProvider {
    @BService
    public static ProxyManager<String> proxyManager(HikariSourceSupplier hikariSourceSupplier) {
        // Create a proxy to manager buckets, persisting with PostgreSQL,
        // see https://bucket4j.com/8.14.0/toc.html#postgresqlselectforupdatebasedproxymanager
        return Bucket4jPostgreSQL.selectForUpdateBasedBuilder(hikariSourceSupplier.getSource())
                // Bucket expiration, needs to be triggered manually,
                // see https://bucket4j.com/8.14.0/toc.html#expiration-policy
                .expirationAfterWrite(ExpirationAfterWriteStrategy.basedOnTimeForRefillingBucketUpToMax(Duration.ofMinutes(1)))
                // RateLimiter#createDefaultProxied uses a String key
                .primaryKeyMapper(PreparedStatement::setString)
                .build();
    }
}

@BService
public class AnnotatedProxiedRateLimiterFactory implements AnnotatedRateLimiterFactory {
    private final ProxyManager<String> proxyManager;

    public AnnotatedProxiedRateLimiterFactory(ProxyManager<String> proxyManager) {
        this.proxyManager = proxyManager;
    }

    @NotNull
    @Override
    public RateLimiter create(@NotNull RateLimitScope scope, @NotNull BucketConfigurationSupplier configurationSupplier, boolean deleteOnRefill) {
        return RateLimiter.createDefaultProxied(scope, proxyManager, configurationSupplier, deleteOnRefill);
    }
}