修改经过Spring Gateway的表单中的Json数据

背景

使用Spring Cloud Gateway作为网关时有时候一个请求是既包含excel又包含json的表单数据，出于各种层面考虑网关需要获取并更新其中的json数据
依赖

• Spring Boot版本：2.7.15
  
• Hutool: 5.8.21
  
• Java: 11
  

实现逻辑

实现分为2个部分

• 使用上文提到的ModifyRequestBodyGatewayFilterFactory类来修改请求体，这样最后就不用我们手动包装
  
• 核心service通过将表单转为String，然后根据其中的boundary进行分割，提取修改json报文部分后再进行组装
  

注意：示例代码的核心service处理的表单内容只是2个，Json数据的key指定为json，另一个excel文件流
自定义filter

1. @Component
2. @Slf4j
3. public class RequestModifyFilter implements GlobalFilter, Ordered {
4.     @Autowired
5.     private ModifyRequestBodyGatewayFilterFactory modifyRequestBodyFilter;
6.     @Autowired
7.     private JsonRequestBodyRewriteService jsonRequestBodyRewriteService;
8.     @Autowired
9.     private FormDataRequestBodyRewriteService formDataRequestBodyRewriteService;
11.     @Override
12.     public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
13.         MediaType mediaType = exchange.getRequest().getHeaders().getContentType();
14.         if (MediaType.APPLICATION_JSON.isCompatibleWith(mediaType)) {
15.             // 纯json报文处理逻辑
16.             return modifyRequestBodyFilter
17.                     .apply(
18.                             new ModifyRequestBodyGatewayFilterFactory.Config()
19.                                     .setRewriteFunction(byte[].class, byte[].class, jsonRequestBodyRewriteService))
20.                     .filter(exchange, chain);
21.         } else if (MediaType.MULTIPART_FORM_DATA.isCompatibleWith(mediaType)) {
22.             // form表单数据处理
23.             return modifyRequestBodyFilter
24.                     .apply(
25.                             new ModifyRequestBodyGatewayFilterFactory.Config()
26.                                     .setRewriteFunction(byte[].class, byte[].class, formDataRequestBodyRewriteService))
27.                     .filter(exchange, chain);
28.         } else {
29.             return filter(exchange, chain);
30.         }
32.     }
34.     @Override
35.     public int getOrder() {
36.         return OrderConstant.REQUEST_MODIFY_FILTER.getOrder();
37.     }
38. }

复制代码

核心service

1. @Service
2. @Slf4j
3. public class FormDataRequestBodyRewriteService implements RewriteFunction<byte[], byte[]> {
4.     private final String BOUNDARY_PREFIX_IN_CONTENT_TYPE = "----WebKitFormBoundary";
5.     private final String BOUNDARY_PREFIX_IN_FORM_DATA = "------WebKitFormBoundary";
6.     private final String BOUNDARY_SUFFIX = "--\r\n";
8.     @Override
9.     public Publisher<byte[]> apply(ServerWebExchange exchange, byte[] body) {
10.         String finalResultString = "";
12.         // 将表单转为字符串格式从而根据boundary分割表单数据。注意这里不能用默认编码
13.         String request = StrUtil.str(body, StandardCharsets.ISO_8859_1);
15.         // 获取boundary的随机字符信息
16.         String contentType = exchange.getRequest().getHeaders().getContentType().toString();
17.         String randomStr = contentType.substring(contentType.indexOf(BOUNDARY_PREFIX_IN_CONTENT_TYPE) + BOUNDARY_PREFIX_IN_CONTENT_TYPE.length());
19.         // 这里和前端约定json数据的表单key为json
20.         String keyPart = "^\r\nContent-Disposition: form-data; name="json"";
21.         Pattern r = Pattern.compile(keyPart);
23.         // 根据表单内分割线进行分割。并通过关键段落keyPart来找到目标json数据
24.         String[] split = request.split(BOUNDARY_PREFIX_IN_FORM_DATA + randomStr);
25.         for (int x = 0; x < split.length - 1; x++) {
26.             Matcher m = r.matcher(split[x]);
27.             if (m.find()) {
28.                 // 找到了json报文部分数据
29.                 String originalJsonString = split[x];
31.                 // 找到 JSON 数据的起始和结束位置
32.                 int startIndex = originalJsonString.indexOf("{"");
33.                 int endIndex = originalJsonString.indexOf(""}") + 2;
34.                 // 提取 JSON 数据
35.                 String jsonData = originalJsonString.substring(startIndex, endIndex);
36.                 log.info("原始报文为：{}", jsonData);
38.                 JSONObject jsonObject = JSONUtil.parseObj(jsonData);
39.                 jsonObject.set("empId", "2345");
40.                 jsonObject.set("department", "Engineering");
41.                 String modifiedString = originalJsonString.substring(0, startIndex) + jsonObject + originalJsonString.substring(endIndex);
42.                 log.info("修改后报文为：{}", modifiedString);
44.                 // 重新组装split数组
45.                 finalResultString = finalResultString + modifiedString + BOUNDARY_PREFIX_IN_FORM_DATA + randomStr;
46.             } else {
47.                 // 重组表单数据
48.                 finalResultString = finalResultString + split[x] + BOUNDARY_PREFIX_IN_FORM_DATA + randomStr;
49.             }
50.         }
52.         // 补上最后一截数据
53.         finalResultString = finalResultString + BOUNDARY_SUFFIX;
55.         return Mono.just(finalResultString.getBytes(StandardCharsets.ISO_8859_1));
56.     }
57. }

复制代码

相关代码

https://github.com/eastcukt/demo-gatway
其他

核心service获取表单中的json数据逻辑挺复杂，根本原因是没有合适的方法进行对象转换，如果有像使用@RequestPart(value = "json")注解一样方便的方法将会非常方便也不用自己截取，各位大佬有更方便的方法感谢分享一下
参考

https://blog.csdn.net/qq_36966137/article/details/128536391

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！