Spring Boot & Dubbo微服务架构:高效传输服务端生成文件
在Spring Boot和Dubbo构建的微服务架构中,如何将服务端生成的流文件高效地返回给前端是一个常见问题。服务提供方返回的是文件输入流,需要将其转换为可供前端下载的格式。本文提供一种解决方案。
解决方案:
服务提供方:
- 使用ByteArrayOutputStream接收文件流。
- 将文件流写入ByteArrayOutputStream。
- 使用ByteArrayOutputStream.toByteArray()获取文件字节数组。
- 返回该字节数组。
服务消费方:
- 调用服务提供方接口,获取文件字节数组。
- 设置HttpServletResponse的contentType为"application/octet-stream",指定二进制流。
- 获取HttpServletResponse.getOutputStream()。
- 将字节数组写入输出流。
代码示例:
服务提供方 (示例):
@Service public class FileServiceImpl implements FileService { @Override public byte[] generateFile() { ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); try { // 文件生成逻辑,将文件内容写入outputStream // ... 例如:outputStream.write(someBytes); } catch (IOException e) { // 处理异常 e.printStackTrace(); } return outputStream.toByteArray(); } }
服务消费方 (示例):
@RestController public class FileController { @Autowired private FileService fileService; @GetMapping("/file") public void downloadFile(HttpServletResponse response) throws IOException { byte[] fileBytes = fileService.generateFile(); response.setContentType("application/octet-stream"); response.setHeader("Content-Disposition", "attachment; filename="downloaded_file.txt""); // 设置文件名 OutputStream out = response.getOutputStream(); out.write(fileBytes); out.flush(); out.close(); } }
此方案中,服务提供方将文件流转换为字节数组,服务消费方再将字节数组写入HttpServletResponse的输出流,从而实现文件下载。 注意添加文件名设置,增强用户体验。 记得处理潜在的IOException。