3 changed files with 182 additions and 51 deletions
Unified View
Diff Options
-
+56 -51pom.xml
-
+25 -0src/main/java/com/inspect/simulator/utils/http/WebClientConfig.java
-
+101 -0src/main/java/com/inspect/simulator/utils/http/WebClientUtil.java
+ 56
- 51
pom.xml
View File
+ 25
- 0
src/main/java/com/inspect/simulator/utils/http/WebClientConfig.java
View File
| @ -0,0 +1,25 @@ | |||||
| package com.inspect.simulator.utils.http; | |||||
| import org.springframework.context.annotation.Bean; | |||||
| import org.springframework.context.annotation.Configuration; | |||||
| import org.springframework.http.client.reactive.ReactorClientHttpConnector; | |||||
| import org.springframework.web.reactive.function.client.WebClient; | |||||
| import reactor.netty.http.client.HttpClient; | |||||
| import java.time.Duration; | |||||
| @Configuration | |||||
| public class WebClientConfig { | |||||
| @Bean | |||||
| public WebClient webClient() { | |||||
| // 配置 HttpClient：连接超时、响应超时 | |||||
| HttpClient httpClient = HttpClient.create() | |||||
| .responseTimeout(Duration.ofSeconds(30)); // 响应超时 | |||||
| return WebClient.builder() | |||||
| .clientConnector(new ReactorClientHttpConnector(httpClient)) | |||||
| .codecs(configurer -> configurer.defaultCodecs().maxInMemorySize(1024 * 1024 * 10)) // 设置最大内存限制为10MB | |||||
| .build(); | |||||
| } | |||||
| } | |||||
+ 101
- 0
src/main/java/com/inspect/simulator/utils/http/WebClientUtil.java
View File
| @ -0,0 +1,101 @@ | |||||
| package com.inspect.simulator.utils.http; | |||||
| import com.alibaba.fastjson.JSONObject; | |||||
| import com.inspect.simulator.constant.Color; | |||||
| import lombok.extern.slf4j.Slf4j; | |||||
| import org.apache.commons.lang3.StringUtils; | |||||
| import org.springframework.beans.factory.annotation.Autowired; | |||||
| import org.springframework.http.HttpMethod; | |||||
| import org.springframework.http.MediaType; | |||||
| import org.springframework.stereotype.Service; | |||||
| import org.springframework.web.reactive.function.client.WebClient; | |||||
| import java.util.Map; | |||||
| @Slf4j | |||||
| @Service | |||||
| public class WebClientUtil { | |||||
| @Autowired | |||||
| private WebClient webClient; | |||||
| public WebClient.RequestBodySpec http(HttpMethod httpMethod, String uri) { | |||||
| log.info(Color.MAGENTA + "[WebClient] address: {} " + Color.END, uri); | |||||
| assert StringUtils.isNotBlank(uri); | |||||
| assert webClient != null; | |||||
| return webClient.method(httpMethod).uri(uri); | |||||
| } | |||||
| public <T> T get(String uri, Class<T> clazz, Map<String, String> headerMap) { | |||||
| JSONObject result = http(HttpMethod.GET, uri) | |||||
| .headers(httpHeaders -> { | |||||
| headerMap.forEach((k, v) -> { | |||||
| httpHeaders.add(k, v); | |||||
| }); | |||||
| }) | |||||
| .retrieve() | |||||
| .bodyToMono(JSONObject.class) | |||||
| .block(); | |||||
| log.info("[WebClient] get, RES:{}", result.toJSONString()); | |||||
| return this.checkResultCodeAndConvertResult(uri, result, clazz); | |||||
| } | |||||
| public <T> T getJson(String uri, Class<T> clazz) { | |||||
| JSONObject result = http(HttpMethod.GET, uri) | |||||
| .contentType(MediaType.APPLICATION_JSON).retrieve().bodyToMono(JSONObject.class).block(); | |||||
| log.info("[WebClient] getJson, RES:{}", result.toJSONString()); | |||||
| return this.checkResultCodeAndConvertResult(uri, result, clazz); | |||||
| } | |||||
| public JSONObject getResultJson(String uri) { | |||||
| JSONObject result = http(HttpMethod.GET, uri) | |||||
| .retrieve() | |||||
| .bodyToMono(JSONObject.class) | |||||
| .block(); | |||||
| return result; | |||||
| } | |||||
| public <T> T putJson(Object requestParam, String uri, Class<T> clazz) { | |||||
| JSONObject t = http(HttpMethod.PUT, uri) | |||||
| .contentType(MediaType.APPLICATION_JSON) | |||||
| .bodyValue(requestParam) | |||||
| .retrieve() | |||||
| .bodyToMono(JSONObject.class) | |||||
| .block(); | |||||
| log.debug("[WebClient] putJson, PARAM: {}, RES: {}", JSONObject.toJSONString(requestParam), t.toJSONString()); | |||||
| return checkResultCodeAndConvertResult(uri, t, clazz); | |||||
| } | |||||
| public <T> T postJson(Object requestParam, String uri, Class<T> clazz, Map<String, String> headerMap) { | |||||
| JSONObject result = http(HttpMethod.POST, uri) | |||||
| .contentType(MediaType.APPLICATION_JSON) | |||||
| .headers(httpHeaders -> { | |||||
| headerMap.forEach((k, v) -> { | |||||
| httpHeaders.add(k, v); | |||||
| }); | |||||
| }) | |||||
| .bodyValue(requestParam).retrieve() | |||||
| .bodyToMono(JSONObject.class) | |||||
| .block(); | |||||
| log.info("[WebClient] postJson, PARAM:{}, RES:{}", JSONObject.toJSONString(requestParam), result.toJSONString()); | |||||
| return checkResultCodeAndConvertResult(uri, result, clazz); | |||||
| } | |||||
| public <T> T delete(String uri, Class<T> clazz) { | |||||
| JSONObject result = http(HttpMethod.DELETE, uri) | |||||
| .contentType(MediaType.APPLICATION_FORM_URLENCODED) | |||||
| .retrieve() | |||||
| .bodyToMono(JSONObject.class) | |||||
| .block(); | |||||
| return checkResultCodeAndConvertResult(uri, result, clazz); | |||||
| } | |||||
| private <T> T checkResultCodeAndConvertResult(final String uri, JSONObject result, Class<T> clazz) { | |||||
| log.info("CHECK_RET, URI: {}, RES: {}", uri, result.toJSONString()); | |||||
| return result.toJavaObject(clazz); | |||||
| } | |||||
| } | |||||