创建图片服务接口及ByteArrayOutputStream空数组问题求助

核心问题分析 #

你的代码里有两个关键问题导致字节数组为空：

1. 路径参数未正确绑定：@GetMapping里的{name}路径参数，方法参数String name没加@PathVariable注解，可能导致name值为空，进而读取错误的文件路径；
2. ImageIO.write参数不匹配：Files.probeContentType返回的是MIME类型（比如image/png、image/jpeg），但ImageIO.write的第二个参数需要的是图片格式的后缀名（比如png、jpg），格式不匹配导致写入失败，最终输出流为空。

另外还有潜在风险：如果图片格式是ImageIO不支持的类型（比如webp），ImageIO.read会返回null，同样会导致写入失败。

修复后的代码 #

@GetMapping(path = "/get/image/{name}")
public ResponseEntity<byte[]> displayImage(@PathVariable String name) throws IOException {
    // 绑定路径参数
    String photoPathFromDatabase = productRepository.findPhotoByName(name);
    Path path = Paths.get(photoPathFromDatabase);
    File imageFile = path.toFile();

    // 校验文件是否存在
    if (!imageFile.exists()) {
        return ResponseEntity.notFound().build();
    }

    ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
    BufferedImage image = ImageIO.read(imageFile);
    
    // 校验图片是否读取成功
    if (image == null) {
        return ResponseEntity.status(HttpStatus.UNSUPPORTED_MEDIA_TYPE).build();
    }

    // 将MIME类型转换为ImageIO支持的格式名
    String mimeType = Files.probeContentType(path);
    String formatName = mimeType.split("/")[1];
    // 针对jpeg的特殊处理，ImageIO识别"jpeg"或"jpg"
    if ("jpeg".equals(formatName)) {
        formatName = "jpg";
    }

    // 写入字节流并校验结果
    boolean writeSuccess = ImageIO.write(image, formatName, byteArrayOutputStream);
    if (!writeSuccess) {
        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build();
    }

    byte[] bytes = byteArrayOutputStream.toByteArray();
    return ResponseEntity
            .ok()
            .contentType(MediaType.valueOf(mimeType))
            .body(bytes);
}

额外优化建议 #

可以直接读取文件字节流，绕开ImageIO的格式限制，更高效且支持更多格式：

@GetMapping(path = "/get/image/{name}")
public ResponseEntity<Resource> displayImage(@PathVariable String name) throws IOException {
    String photoPathFromDatabase = productRepository.findPhotoByName(name);
    Path path = Paths.get(photoPathFromDatabase);
    FileSystemResource resource = new FileSystemResource(path);
    
    if (!resource.exists()) {
        return ResponseEntity.notFound().build();
    }

    String mimeType = Files.probeContentType(path);
    return ResponseEntity
            .ok()
            .contentType(MediaType.valueOf(mimeType))
            .body(resource);
}

这种方式不需要将图片转为BufferedImage，直接返回文件资源，避免了ImageIO的格式兼容性问题，性能也更好。

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