Service代码:
//class name: FileAccessService
/**
* 文件访问中转
* @param url 地址
* @param resp 响应
*/
public void relay(String url, HttpServletResponse resp){
if(url==null || "".equals(url.trim())){
return; // url为空时不处理
}
CloseableHttpClient httpClient = HttpClientBuilder.create().build();
CloseableHttpResponse response = null;
System.out.println("要转发的文件地址:"+url);
try {
response = httpClient.execute(new HttpGet(url));
HttpEntity entity = response.getEntity();
int status = response.getStatusLine().getStatusCode();
resp.setStatus(status);
if (status==200 && entity!=null) {
//Content-Type根据文件实际类型设置
resp.setHeader("Content-Type", "application/pdf;charset=utf-8");
resp.setHeader("Content-Length", entity.getContentLength()+"");
resp.setHeader("Content-Encoding", "UTF-8");
InputStream is = entity.getContent();
OutputStream os = resp.getOutputStream();
byte[] buffer = new byte[1024];
int ch;
while ((ch = is.read(buffer)) != -1) {
os.write(buffer, 0, ch);
}
is.close();
os.flush();
os.close();
}
} catch (IOException e) {
e.printStackTrace();
}finally {
try {
if (httpClient != null) {
httpClient.close();
}
if (response != null) {
response.close();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
Controller接口:
@GetMapping(value = "/relay")
public void fileRelay(@RequestParam("url") String url, HttpServletResponse response){
fileAccessService.relay(url, response);
}