php分布式限流解决方案,一个分布式限流的解决方案！

这个项目干嘛的?

这个项目是一个Api限流的解决方案，采用的是令牌桶的方式。如果使用的Redis则是分布式限流，如果采用guava的LimitRater，则是本地限流。 分2给维度限流，一个是用户维度，一个Api维度，读者可自定义。仅支持Spring Boot项目。

怎么使用？

在boot-example工程，有完整的案例。可在Controller上限流，也可以在Spring Mvc的Interceptor或者Servlet的Filter上限流。

开启限流

在pom文件加上jar包：

io.github.forezp

distributed-limit-core

1.0.4

有2中方式，本地限流，只需要配置limit.type=local；采用Reis限流，配置limit.type=redis，以及redis的配置，如下：

#limit.type: local

limit.type: redis

spring:

redis:

host: localhost

port: 6379

# password: ee

database: 1

pool:

max-active: 8

max-wait: -1

max-idle: 500

min-idle: 0

timeout: 0

Controller上使用，基于注解、AOP

在Controller上加 @Limit注解，其中identifier为识别身份的，key为限流的key,limtNum为限制的次数，seconds为多少秒，后2个配置的作用是在多少秒最大的请求次数 。其中identifier和key支持Spel表达式。如果仅API纬度，则identifier 为空即可；如果仅用户纬度，key为空即可。

@RestController

public class TestController {

@GetMapping("/test")

@Limit(identifier = "forezp", key = "test", limtNum = 10, seconds = 1)

public String Test() {

return "11";

}

}

仅次操作就可以限流了。

另外如果是以注解的形式进行限流，如果以identifier即请求用户维度去限流，可以动态的设置的identifier的值，示例如下：

@Component

public class IndentifierInterceptor extends HandlerInterceptorAdapter {

@Override

public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

//获取用户的信息，比如解析Token获取用户名，

// 这么做主要是在基于@Limit注解在Controller的时候，能都动态设置identifier信息

// 从而以用户维度进行限流

String identifier = "forezp";

IdentifierThreadLocal.set( identifier );

return true;

}

}

在Web层的Interceptor、Filter上使用

直接贴代码了，比较简单。

@Component

public class WebInterceptor extends HandlerInterceptorAdapter {

private Map limitEntityMap = Maps.newConcurrentMap();

@Autowired

LimitExcutor limitExcutor;

public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

//限流2个维度： 用户和api维度

//比如用户名

String identifier = "forezp";

//api维度

String key = request.getRequestURI();

String composeKey = KeyUtil.compositeKey( identifier, key );

LimitEntity limitEntity = limitEntityMap.get( composeKey );

if (limitEntity == null) {

limitEntity = new LimitEntity();

limitEntity.setIdentifier( identifier );

limitEntity.setKey( key );

//这可以在数据库中配置或者缓存中读取，在这里我写死

limitEntity.setSeconds( 1 );

limitEntity.setLimtNum( 10 );

limitEntityMap.putIfAbsent( composeKey, limitEntity );

}

if (!limitExcutor.tryAccess( limitEntity )) {

throw new LimitException( "you fail access, cause api limit rate ,try it later" );

}

return true;

}

}

注册一下Interceptor：

@Configuration

public class WebConfig extends WebMvcConfigurerAdapter {

@Autowired

IndentifierInterceptor indentifierInterceptor;

@Autowired

WebInterceptor webInterceptor;

/**

* 注册 拦截器

*/

@Override

public void addInterceptors(InterceptorRegistry registry) {

registry.addInterceptor( indentifierInterceptor );

registry.addInterceptor( webInterceptor );

}

}

联系我