Spring Boot对接GCS获取文件时如何避免文件下载到本地目录

问题原因 #

你当前代码显式创建了项目根目录下的本地文件，通过FileOutputStream将GCS返回的文件流写入本地磁盘，之后再读取本地文件转字节数组返回，落盘逻辑完全冗余。

方案一：直接返回字节数组（适配现有代码逻辑） #

直接将GCS返回的媒体流写入内存的ByteArrayOutputStream，全程不触碰本地磁盘，服务层代码修改如下：

public byte[] getImageBytes(String fileName) throws IOException {
    // 如需校验文件存在性可保留元信息查询，否则可省略
    storage.objects().get(bucketName, fileName).execute();
    
    try (InputStream gcsIs = storage.objects()
                .get(bucketName, fileName)
                .executeMediaAsInputStream();
         ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
        
        byte[] buffer = new byte[4096];
        int readLen;
        while ((readLen = gcsIs.read(buffer)) != -1) {
            baos.write(buffer, 0, readLen);
        }
        return baos.toByteArray();
    }
}

对应控制器层修改如下，去掉本地文件读取逻辑：

@GetMapping("/get/image/{id}")
public ResponseEntity<byte[]> getImage(@PathVariable("id") Long id) {
    try {
        String fileName = imageService.findImageById(id);
        byte[] imageBytes = gcsService.getImageBytes(fileName);

        return ResponseEntity.ok()
                // 建议根据GCS对象元数据动态获取Content-Type，避免硬编码导致非JPG图片无法展示
                .contentType(MediaType.IMAGE_JPEG)
                .body(imageBytes);
    } catch (IOException e) {
        e.printStackTrace();
        return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
    }
}

方案二：流式响应（高并发/大图片场景推荐） #

如果服务承载的图片体积较大、请求量高，方案一会把整个图片加载到内存，容易造成内存压力。可以直接将GCS的输入流写入响应输出流，内存占用仅为缓冲区大小，性能更优。
服务层直接返回媒体输入流：

public InputStream getImageStream(String fileName) throws IOException {
    return storage.objects()
            .get(bucketName, fileName)
            .executeMediaAsInputStream();
}

控制器层使用StreamingResponseBody实现流转发：

@GetMapping("/get/image/{id}")
public ResponseEntity<StreamingResponseBody> getImage(@PathVariable("id") Long id) {
    try {
        String fileName = imageService.findImageById(id);
        StorageObject fileMeta = storage.objects().get(bucketName, fileName).execute();
        InputStream imageStream = gcsService.getImageStream(fileName);

        StreamingResponseBody responseBody = os -> {
            // try-with-resources自动关闭流，避免连接泄漏
            try (InputStream is = imageStream) {
                byte[] buffer = new byte[4096];
                int readLen;
                while ((readLen = is.read(buffer)) != -1) {
                    os.write(buffer, 0, readLen);
                }
            }
        };

        return ResponseEntity.ok()
                .contentType(MediaType.parseMediaType(fileMeta.getContentType()))
                .body(responseBody);
    } catch (IOException e) {
        e.printStackTrace();
        return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
    }
}

注意事项 #

• 所有流操作必须用try-with-resources包裹，自动释放资源，避免HTTP连接和内存泄漏
• 不要硬编码响应Content-Type为IMAGE_JPEG，从GCS对象元数据中读取真实MIME类型，可兼容png、webp、gif等所有图片格式
• 代码上线前可清理项目根目录下之前落盘的冗余图片文件，释放磁盘空间

内容的提问来源于stack exchange，提问作者Specialized