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

🎨 红旗数据api-优化4

parent 0e98bcdb
...@@ -59,7 +59,7 @@ public class CommonDataController extends BaseController { ...@@ -59,7 +59,7 @@ public class CommonDataController extends BaseController {
@GetMapping("/activity") @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, 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) { @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) { if (check != null) {
return check; return check;
} }
...@@ -70,7 +70,7 @@ public class CommonDataController extends BaseController { ...@@ -70,7 +70,7 @@ public class CommonDataController extends BaseController {
@ApiOperation(value = "获取token", notes = "token数据", produces = "application/json") @ApiOperation(value = "获取token", notes = "token数据", produces = "application/json")
@PostMapping("/token") @PostMapping("/token")
public BaseResponse<TokenVO> getToken(@RequestBody @Valid TokenDTO dto) { 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) { if (check != null) {
return check; return check;
} }
......
...@@ -38,11 +38,8 @@ public class BaseService { ...@@ -38,11 +38,8 @@ public class BaseService {
return null; 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"; String header = "token";
if (request != null) {
header = (String) request.getAttribute(BaseController.header);
}
String fawKey = String.format("%s:%s:limit", dataType, header); String fawKey = String.format("%s:%s:limit", dataType, header);
RRateLimiter rateLimiter = redissonClient.getRateLimiter(fawKey); RRateLimiter rateLimiter = redissonClient.getRateLimiter(fawKey);
rateLimiter.trySetRate(RateType.OVERALL, limit,limitPeriod, RateIntervalUnit.SECONDS); 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