java - 如何在 Spring Boot 中为每个用户设置速率限制？

Question

我正在开发一个 Spring Boot Rest API，它处理大量传入的请求调用。我的 Controller 如下所示:

@RestController

public class ApiController {
    List<ApiObject>  apiDataList;   

    @RequestMapping(value="/data",produces={MediaType.APPLICATION_JSON_VALUE},method=RequestMethod.GET)
    public ResponseEntity<List<ApiObject>> getData(){                                       
        List<ApiObject> apiDataList=getApiData();
        return new ResponseEntity<List<ApiObject>>(apiDataList,HttpStatus.OK);
    }
    @ResponseBody 
    @Async  
    public List<ApiObject>  getApiData(){
        List<ApiObject>  apiDataList3=new List<ApiObject> ();
        //do the processing
        return apiDataList3;
    }
}

所以现在我想为每个用户设置一个速率限制。假设每个用户每分钟只能请求 5 个请求或类似的请求。如何将每个用户的速率限制设置为每分钟仅进行 5 次 api 调用，如果用户请求的次数超过此次数，我可以发回 429 响应？我们需要他们的 IP 地址吗？

任何帮助表示赞赏。

Answer 1

您在 Spring 中没有该组件。

您可以将其构建为解决方案的一部分。创建一个过滤器并在您的 spring 上下文中注册它。过滤器应检查传入调用并计算时间窗口内每个用户的传入请求。我会使用 token bucket algorithm因为它是最灵活的。

您可以构建一些独立于当前解决方案的组件。创建一个 API 网关来完成这项工作。您可以扩展 Zuul 网关，并再次使用 token 桶算法。

您可以使用已经内置的组件，例如 Mulesoft ESB，它可以充当 API 网关并支持速率限制和节流。自己没用过。

最后，您可以使用具有速率限制和节流等功能的 API 管理器。结帐 MuleSoft、WSO2、3Scale、Kong 等...(大多数会产生费用，有些是开源的并且有社区版)。