Java使用OkHttpClient处理响应数据（文件流、JSON、文本等）

目录

一、逻辑说明

二、示例代码

一、逻辑说明

• 在使用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();
        }
 
    }
}复制