【WEB系列】WebClient之策略设置

在前面介绍WebClient的常见参数中，有一个exchangeStrategies参数设置，通过它我们可以设置传输数据的内存占用大小限制，避免内存问题；也可以通过它设置数据的编解码

# I. 项目环境

本项目借助SpringBoot 2.2.1.RELEASE + maven 3.5.3 + IDEA进行开发

# 1. 依赖

使用WebClient，最主要的引入依赖如下（省略掉了SpringBoot的相关依赖，如对于如何创建SpringBoot项目不太清楚的小伙伴，可以关注一下我之前的博文）

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-webflux</artifactId>
</dependency>

1
2
3
4

# 2. 测试接口

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Body implements Serializable {
    private static final long serialVersionUID = 1210673970258821332L;
    String name;
    Integer age;
}

@GetMapping(path = "get")
public Mono<String> get(String name, Integer age) {
    return Mono.just("req: " + name + " age: " + age);
}

@PostMapping(path = "body2")
public Mono<Body> postBody2(@RequestBody  Body body) {
    return Mono.just(body);
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

# II. 使用说明

# 1. MaxInMemorySize设置

默认情况下，内存中接收数据的buffering data size为256KB，可以根据实际的需要进行改大or调小

// 默认允许的内存空间大小为256KB，可以通过下面的方式进行修改
webClient = WebClient.builder().exchangeStrategies(
        ExchangeStrategies.builder().codecs(codec -> codec.defaultCodecs().maxInMemorySize(10)).build())
        .baseUrl("http://127.0.0.1:8080").build();

String argument = "这也是一个很长很长的文本，用于测试超出上限!";
Mono<String> ans = webClient.get().uri("/get?name={1}", argument).retrieve().bodyToMono(String.class)
        // 异常处理
        .doOnError(WebClientResponseException.class, err -> {
            System.out.println(err.getRawStatusCode() + "," + err.getResponseBodyAsString());
            throw new RuntimeException(err.getMessage());
        }).onErrorReturn("fallback");
ans.subscribe(s -> System.out.println("exchange strategy: " + ans));

1
2
3
4
5
6
7
8
9
10
11
12
13

# 2. 编解码设置

比如最常见的json编解码

WebClient webClient = WebClient.builder().exchangeStrategies(ExchangeStrategies.builder().codecs(codec -> {
    codec.customCodecs().decoder(new Jackson2JsonDecoder());
    codec.customCodecs().encoder(new Jackson2JsonEncoder());
}).build()).baseUrl("http://127.0.0.1:8080").build();
Body body = new Body("一灰灰😝", 18);
Mono<Body> ans =
        webClient.post().uri("/body2").contentType(MediaType.APPLICATION_JSON).bodyValue(body).retrieve()
                .bodyToMono(Body.class);
ans.subscribe(s -> System.out.println("retreive res: " + s));