zhouxy108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-07-21 15:09:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

add TokenBucketRateLimiter

Browse Source
This commit is contained in:
Looly
2024-08-21 09:00:27 +08:00
parent f40092c95e
commit d2a00a6268
5 changed files with 115 additions and 144 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
hutool-core/src/test/java/org/dromara/hutool/core/thread/ratelimiter/SemaphoreRateLimiterTest.java
View File

@@ -10,8 +10,12 @@ public class SemaphoreRateLimiterTest {
@Test
void test() {
final RateLimiterConfig rateLimiterConfig = RateLimiterConfig.of(Duration.ofSeconds(5), Duration.ofMillis(300), 5);
final RateLimiter rateLimiter = new FixedRateLimiter(rateLimiterConfig);
final RateLimiterConfig rateLimiterConfig = RateLimiterConfig.of()
.setCapacity(5)
.setMaxReleaseCount(5)
.setRefreshPeriod(Duration.ofMillis(300))
.setTimeout(Duration.ofSeconds(5));
final TokenBucketRateLimiter rateLimiter = new TokenBucketRateLimiter(rateLimiterConfig);
final boolean b = rateLimiter.tryAcquire(5);
Assertions.assertTrue(b);
@@ -27,5 +31,7 @@ public class SemaphoreRateLimiterTest {
// 超过数量
final boolean b3 = rateLimiter.tryAcquire(1);
Assertions.assertFalse(b3);
rateLimiter.close();
}
}