目录
一、逻辑说明
二、示例代码
一、逻辑说明
- 在使用OkHttpClient时,可以通过检查Response对象的内容类型(Content-Type)来确定返回的响应是否是文件流。如果文件流是以特定的Content-Type(如"application/octet-stream")返回的,则可以将响应的body作为输入流处理,设置Content-Disposition头信息,提示浏览器以附件方式下载文件。如果是非文件流,我们将直接返回响应数据(如JSON、文本等)。请根据你使用的具体web框架和前端要求进行适当的调整,并确保正确设置相应的头信息和Content-Type。
二、示例代码
- 以下是一个示例代码片段,演示如何使用OkHttpClient来处理返回的文件流、JSON、文本等:
@RestController
@RequestMapping("/xxx/xx/xxx")
public class JyReportPdfController {
@GetMapping("/xxx")
public void getJyReportPdf(@RequestParam("key") String key, HttpServletResponse res) {
// 请求
OkHttpClient client = new OkHttpClient().newBuilder()
.build();
// 请求体 根据自己的需求更换get、post请求及请求变量
RequestBody body = new MultipartBody.Builder().setType(MultipartBody.FORM)
.addFormDataPart("key", key)
.build();
Request request = new Request.Builder()
.url("https://example.com/file")
.method("POST", body)
.build();
try (Response response = client.newCall(request).execute()) {
//执行成功
if (response.isSuccessful()) {
String contentType = response.header("Content-Type");
// 检查Content-Type来判断响应是否为文件流
if (contentType != null && contentType.equals("application/octet-stream")) {
// 获取文件流
InputStream inputStream = response.body().byteStream();
// 返回文件流响应
res.setHeader("Content-Disposition", "attachment; fileName=" + URLEncoder.encode(key, "UTF-8"));
res.setContentType("application/octet-stream");
OutputStream outputStream = res.getOutputStream();
byte[] buffer = new byte[1024];
int len;
while ((len = inputStream.read(buffer)) > 0) {
outputStream.write(buffer, 0, len);
}
outputStream.flush();
inputStream.close();
outputStream.close();
} else {
// 处理非文件流的响应,如JSON、文本等
res.setContentType("application/json;charset=UTF-8");
String responseData = response.body().string();
res.getWriter().write(responseData);
}
} else {
// 处理请求失败的情况
res.setContentType("application/json;charset=UTF-8");
String state = "{\"state\":\"10\"}";
res.getWriter().write(state);
}
} catch (IOException e) {
// 处理IO异常
e.printStackTrace();
}
}
}