背景
新的限流方法需要从请求中读取appid,实现appid级的限流。功能开发完在测试环境中跑没有问题,测试也没有发现问题。但上线后却直接抛异常,看了下日志,是个固定appid,从request读取的时候拿不到。仔细查看后,发现这个appid对应的客户采用的xml的格式请求,不是JSON的!
具体问题代码行:
RequestWrapper requestWrapper = new RequestWrapper(request);
String body = requestWrapper.getBody();
Map<String, Object> baseMap = OBJECT_MAPPER.readValue(body, new TypeReference<Map<String, Object>>() {});
如果请求是xml,最后一行jackson读值的时候就会报错。
解决方法
解决的思路是,拿到请求后,先判断请求头里的请求格式,如果是xml格式,则先转成JSON。后续流程保持一致。从网上找到两种xml转JSON的方法,详见参考文件。
具体的实现代码如下:
if(StringUtils.isBlank(appId)) {
RequestWrapper requestWrapper = new RequestWrapper(request);
String body = requestWrapper.getBody();
//判断区分application/xml格式还是application/json格式
String contentType = request.getHeader(ParamKeys.CONTENT_TYPE);
//XML格式读取
if(StringUtils.isNotBlank(contentType) && contentType.contains(ParamKeys.APPLICATION_XML) && StringUtils.isNotBlank(body)) {
Document document = DocumentHelper.parseText(body);
Element root = document.getRootElement();
appId = StringUtils.isNotBlank(root.elementText(ParamKeys.APP_ID_PARAM2)) ? root.elementText(ParamKeys.APP_ID_PARAM2) : root.elementText(ParamKeys.APP_ID_PARAM3);
}
//json格式读取
if(StringUtils.isBlank(appId) && StringUtils.isNotBlank(body)) {
Map<String, Object> baseMap = OBJECT_MAPPER.readValue(body, new TypeReference<Map<String, Object>>() {});
appId = StringUtils.isNotBlank((String)baseMap.get(ParamKeys.APP_ID_PARAM2))?(String)baseMap.get(ParamKeys.APP_ID_PARAM2) : (String)baseMap.get(ParamKeys.APP_ID_PARAM3);
}
}
另外使用dom4j需要先引包,以gradle为例:
List dom = [
"org.dom4j:dom4j:2.1.3"
]
参考文献:
https://blog.csdn.net/qq_21856521/article/details/89187481
https://zhuanlan.zhihu.com/p/326424733