后端透传代理实现
目录
后端透传代理实现
后端透传代理实现
参数设计
- 增加的一个target参数,用于表明需要代理的目标地址
- 其他query参数透传
- header参数透传
代码实现
@RestController
public class HomeController {
@GetMapping("/backend/proxy")
public ResponseEntity<?> proxy(@RequestParam MultiValueMap<String, String> params, @RequestHeader MultiValueMap<String, String> heads) {
if (!params.containsKey("target")) {
return ResponseEntity.status(HttpStatus.NOT_FOUND).body("{\"message\": \"target param is required\"}");
}
String url = params.remove("target").get(0);
System.out.println("url: " + url);
URI uri;
try {
uri = new URI(url);
} catch (URISyntaxException ex) {
return ResponseEntity.status(HttpStatus.NOT_FOUND).body("{\"message\": \"target value invalid\"}");
}
String result = WebClient.create().get().uri(uriBuilder -> uriBuilder.scheme(uri.getScheme())
.host(uri.getHost()).path(uri.getPath()).queryParams(params).build())
.headers(headers -> headers.addAll(heads)).retrieve()
.bodyToMono(String.class).block();
return ResponseEntity.ok(result);
}
}