Commit 0a54e567 authored by 陈炎's avatar 陈炎

🎨 红旗数据api-优化4

parent 0e98bcdb
......@@ -59,7 +59,7 @@ public class CommonDataController extends BaseController {
@GetMapping("/activity")
public BaseResponse<BasePageVO<ActivityInfoVO>> getActionData(HttpServletRequest request, @Max(value = 500,message = "单页数据page_size过大") @RequestParam(name = "page_size", defaultValue = "100", required = false) Integer pageSize,
@RequestParam(name = "current_page", defaultValue = "1", required = false) Integer pageNum) {
BaseResponse check = baseService.baseLimit(request, dateType, limit, limitPeriod);
BaseResponse check = baseService.baseLimit(dateType, limit, limitPeriod);
if (check != null) {
return check;
}
......@@ -70,7 +70,7 @@ public class CommonDataController extends BaseController {
@ApiOperation(value = "获取token", notes = "token数据", produces = "application/json")
@PostMapping("/token")
public BaseResponse<TokenVO> getToken(@RequestBody @Valid TokenDTO dto) {
BaseResponse check = baseService.baseLimit(null, dateType, limit, limitPeriod);
BaseResponse check = baseService.baseLimit(dateType, limit, limitPeriod);
if (check != null) {
return check;
}
......
......@@ -38,11 +38,8 @@ public class BaseService {
return null;
}
public BaseResponse baseLimit(HttpServletRequest request, String dataType, Integer limit, Integer limitPeriod) {
public BaseResponse baseLimit(String dataType, Integer limit, Integer limitPeriod) {
String header = "token";
if (request != null) {
header = (String) request.getAttribute(BaseController.header);
}
String fawKey = String.format("%s:%s:limit", dataType, header);
RRateLimiter rateLimiter = redissonClient.getRateLimiter(fawKey);
rateLimiter.trySetRate(RateType.OVERALL, limit,limitPeriod, RateIntervalUnit.SECONDS);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment