1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
| package com.qianwen.core.redis.ratelimiter;
|
| import java.util.concurrent.TimeUnit;
| import com.qianwen.core.tool.function.CheckedSupplier;
| import com.qianwen.core.tool.utils.Exceptions;
|
| public interface RateLimiterClient {
| boolean isAllowed(String key, long max, long ttl, TimeUnit timeUnit);
|
| default boolean isAllowed(String key, long max, long ttl) {
| return isAllowed(key, max, ttl, TimeUnit.SECONDS);
| }
|
| default <T> T allow(String key, long max, long ttl, CheckedSupplier<T> supplier) {
| return (T) allow(key, max, ttl, TimeUnit.SECONDS, supplier);
| }
|
| default <T> T allow(String key, long max, long ttl, TimeUnit timeUnit, CheckedSupplier<T> supplier) {
| boolean isAllowed = isAllowed(key, max, ttl, timeUnit);
| if (isAllowed) {
| try {
| return (T) supplier.get();
| } catch (Throwable e) {
| throw Exceptions.unchecked(e);
| }
| }
| throw new RateLimiterException(key, max, ttl, timeUnit);
| }
| }
|
|
