利用Zookeeper来实现分布式环境下的原子计数器和API每天调用次数限制
String apiType = "1";
指定新的限制值:
CountLimit.setLimit(apiType, 1000000);
获取现有限制值:
CountLimit.getLimit(apiType);
String apiType = "1";
// 调用API没有响应的情况
AtomicCounter.getNoResponseCount(apiType);
// 调用API有响应, 但是内容错误的情况
AtomicCounter.getWrongContentCount(apiType);
// 调用API出现异常的情况, 比如网络中断, 服务中断等等
AtomicCounter.getExceptionCount(apiType);
// 成功的情况
AtomicCounter.getResponseSuccessCount(apiType);
// 假设限额为一百万, 客户端请求了两百万, 那么这里的beyond的值就是一百万
AtomicCounter.getBeyondCount(apiType);
以上方法调用的是调用当日的情况, 如果想查看具体某一天的情况, 则指定某一天即可, 如:
AtomicCounter.getNoResponseCount("20161118", apiType);
AtomicCounter.getWrongContentCount("20161118", apiType);
AtomicCounter.getExceptionCount("20161118", apiType);
AtomicCounter.getResponseSuccessCount("20161118", apiType);
AtomicCounter.getBeyondCount("20161118", apiType);
String apiType = "1";
// 检查请求是否超配额
if (AtomicCounter.getResponseSuccessCount(apiType) < CountLimit.getLimit(apiType)) {
// 没有超过配额, 继续调用
callApi(request, apiType);
} else {
// 超过配额, 忽略调用请求
AtomicCounter.beyond(1, apiType);
}