import org.isomorphism.util.TokenBuckets; //导入依赖的package包/类
public GatewayFilter apply(int capacity, int refillTokens, int refillPeriod, TimeUnit refillUnit) {

		final TokenBucket tokenBucket = TokenBuckets.builder()
				.withCapacity(capacity)
				.withFixedIntervalRefillStrategy(refillTokens, refillPeriod, refillUnit)
				.build();

		return (exchange, chain) -> {
			//TODO: get a token bucket for a key
			log.debug("TokenBucket capacity: " + tokenBucket.getCapacity());
			boolean consumed = tokenBucket.tryConsume();
			if (consumed) {
				return chain.filter(exchange);
			}
			exchange.getResponse().setStatusCode(HttpStatus.TOO_MANY_REQUESTS);
			return exchange.getResponse().setComplete();
		};
	}
 

import org.isomorphism.util.TokenBuckets; //导入依赖的package包/类
private static TokenBucket createBucket(@Nonnegative final long bits) {
	// Instead of refilling once per second, we refill 10 times per second.
	// It makes for a smoother bandwidth curve under very low rates,
	// e.g., less than 100 kbps.
	final long bytes = bits / Byte.SIZE;
	final long refillRate = bytes / REFILL_FACTOR;
	// if the division caused an underflow, revert to base units.
	if (refillRate < 1) {
		return TokenBuckets.builder()
			.withCapacity(bytes)
			.withYieldingSleepStrategy()
			.withFixedIntervalRefillStrategy(bytes, BASE_INTERVAL, BASE_UNIT)
			.build();
	}
	return TokenBuckets.builder()
		.withCapacity(bytes)
		.withYieldingSleepStrategy()
		.withFixedIntervalRefillStrategy(refillRate, REFILL_INTERVAL, REFILL_UNIT)
		.build();
}
 

import org.isomorphism.util.TokenBuckets; //导入依赖的package包/类
/**
 * Increased Token Bucket, when <b>isKnownBot</b> is <b>true</b>
 * @return increased token bucket
 */
private TokenBucket setIncreasedMessageBucket() {
	return TokenBuckets.builder()
			.withCapacity(50)
			.withFixedIntervalRefillStrategy(1, 600, TimeUnit.MILLISECONDS)
			.build();
}
 

import org.isomorphism.util.TokenBuckets; //导入依赖的package包/类
/**
 * Default Token Bucket, when <b>isKnownBot</b> is <b>false</b>
 * @return default token bucket
 */
private TokenBucket setDefaultMessageBucket() {
	return TokenBuckets.builder()
			.withCapacity(20)
			.withFixedIntervalRefillStrategy(1, 1500, TimeUnit.MILLISECONDS)
			.build();
}
 

import org.isomorphism.util.TokenBuckets; //导入依赖的package包/类
/**
 * Joining the channel
 * @param channel channel name
 */
public void joinChannel(String channel) {
	if (!channelCache.containsKey(channel)) {
		sendCommand("join", "#" + channel);
		channelCache.put(channel, new ChannelCache(this, channel));

		Logger.debug(this, "Joining Channel [%s].", channel);

		TokenBucket modBucket = TokenBuckets.builder()
				.withCapacity(100)
				.withFixedIntervalRefillStrategy(1, 300, TimeUnit.MILLISECONDS)
				.build();
		modMessageBucket.put(channel, modBucket);
	}
}
 

import org.isomorphism.util.TokenBuckets; //导入依赖的package包/类
public void updateBuckets() {
	int oldNbWorkers = nbWorkers;
	nbWorkers = discoveryClient.getInstances(ServiceIds.worker()).size();

	if (oldNbWorkers != nbWorkers) {
		if (politeness.getPageMeanRate() > 0) {
			pagePageRateBucket = TokenBuckets
					.builder()
					.withCapacity(
							politeness.getPagePeekRate() > 0 && politeness.getPagePeekRate() > politeness.getPageMeanRate() ? politeness.getPagePeekRate()
									/ nbWorkers : politeness.getPageMeanRate() / nbWorkers)
					.withFixedIntervalRefillStrategy(politeness.getPageMeanRate() / nbWorkers, 1, TimeUnit.SECONDS).build();
		} else {
			pagePageRateBucket = null;
		}

		if (politeness.getPageMeanRate() > 0) {
			bandwidthBucket = TokenBuckets
					.builder()
					.withCapacity(
							politeness.getMaxPeekBandwith() > 0 && politeness.getMaxPeekBandwith() > politeness.getMaxBandwith() ? politeness
									.getMaxPeekBandwith() / nbWorkers : politeness.getMaxBandwith() / nbWorkers)
					.withFixedIntervalRefillStrategy(politeness.getPageMeanRate() / nbWorkers, 1, TimeUnit.SECONDS).build();
		} else {
			bandwidthBucket = null;
		}
	}
}
 

import org.isomorphism.util.TokenBuckets; //导入依赖的package包/类
private void checkRateLimitAvailable(Request request, Response response) throws ExecutionException {
    TokenBucket tokenBucket = rateLimiters.get(currentUsername(), () -> TokenBuckets.builder()
            .withCapacity(requestsPerMinute)
            .withInitialTokens(requestsPerMinute)
            .withRefillStrategy(new FixedIntervalRefillStrategy(ticker, requestsPerMinute, 1, TimeUnit.MINUTES))
            .build());

    response.header("X-RateLimit-Limit", String.valueOf(requestsPerMinute));
    response.header("X-RateLimit-Remaining", String.valueOf(tokenBucket.getNumTokens()));

    if (!tokenBucket.tryConsume()) {
        throw HaltApiResponses.haltBecauseRateLimitExceeded();
    }
}