Go image.Decode 怎样根据格式读取图片

在 Go 里读取用户上传的 PNG 或 JPEG，不需要先根据文件名写一串分支。image.Decode 会从输入内容判断格式，并返回统一的 image.Image、格式名和错误。关键前提是：先用空白导入注册需要的解码器；面对不可信图片时，再用 image.DecodeConfig 先看宽高，避免一上来就把超大图完整读进内存。

本文解决：根据图片内容自动识别 PNG/JPEG、读取格式名，并在完整解码前加一道尺寸边界。

让 image.Decode 认识 PNG 和 JPEG

image 包只提供统一入口，具体格式由各自的包在初始化时注册。因此代码里通常保留两个空白导入：_ "image/png" 和 _ "image/jpeg"。空白标识符表示不直接调用包里的名字，但保留初始化副作用。

package main

import (
	"fmt"
	"image"
	"os"

	// 空白导入用于注册 PNG 和 JPEG 解码器。
	_ "image/jpeg"
	_ "image/png"
)

func main() {
	file, err := os.Open("input.png")
	if err != nil {
		panic(err) // 打开失败时不要继续把空文件交给解码器。
	}
	defer file.Close() // 文件句柄由当前函数统一释放。

	img, format, err := image.Decode(file)
	if err != nil {
		panic(err) // 内容损坏或没有匹配的解码器都会从这里返回。
	}
	fmt.Println(format, img.Bounds().Dx(), img.Bounds().Dy())
}

这里的 format 通常是 png 或 jpeg，而 img 只暴露统一的图像接口。也就是说，后续业务可以围绕边界、颜色模型或缩略图处理，而不必把读取逻辑绑定到扩展名。

不可信图片先做尺寸预检

直接 Decode 的问题不在格式判断，而在它可能尝试构造很大的图像对象。官方文档建议处理任意来源图片时先调用 DecodeConfig。它只读取配置头，可以先根据宽高拒绝明显超出业务上限的输入。

package main

import (
	"fmt"
	"image"
	"io"
	"os"

	// 通过初始化注册这两种格式的解码器。
	_ "image/jpeg"
	_ "image/png"
)

func decodeBounded(path string) (image.Image, string, error) {
	file, err := os.Open(path)
	if err != nil {
		return nil, "", err // 打不开文件时直接返回原始错误。
	}
	defer file.Close()

	cfg, format, err := image.DecodeConfig(file)
	if err != nil {
		return nil, "", err // 头部损坏或格式未注册时停止。
	}
	const maxPixels = 20 * 1000 * 1000
	if cfg.Width  maxPixels/cfg.Height {
		return nil, format, fmt.Errorf("image is too large: %dx%d", cfg.Width, cfg.Height)
	}
	if _, err := file.Seek(0, io.SeekStart); err != nil {
		return nil, format, err // DecodeConfig 已经消费了文件位置，解码前必须复位。
	}
	img, format, err := image.Decode(file)
	return img, format, err
}

注意，DecodeConfig 已经从 reader 读取了一部分内容。对可回退的文件，使用 Seek(0, io.SeekStart) 后再 Decode；对网络流或不可回退的 reader，则应先把输入限制在可接受大小内并缓存到可重复读取的介质，再做同样的两阶段处理。

文件名不是格式判断的依据

扩展名只能作为用户提示，不能当作内容事实。文件名叫 photo.jpg，内容仍可能是 PNG，甚至可能是损坏数据。让解码器从字节头判断格式，能把改名、无扩展名和上传来源不稳定的情况统一起来。

如果业务只允许 PNG/JPEG，可以在 Decode 或 DecodeConfig 返回后检查 format；如果要支持 GIF，则额外空白导入 image/gif。不要把没有导入的格式名写进允许列表，否则最终仍会在解码处报错。

几个容易忽略的边界

• 没有注册解码器：看到“unknown format”时，先检查对应格式包是否空白导入。
• 重复读取：DecodeConfig 与 Decode 之间必须让 reader 回到开头；不可 Seek 的输入要先设计缓存边界。
• 整数溢出：用 maxPixels/cfg.Height 这样的除法判断，比先计算 Width*Height 更稳妥。
• 错误处理：格式名只能在没有错误时作为成功结果使用；错误返回时应保留原始错误供上层记录。

一句话总结：用空白导入准备格式注册，用 image.Decode 做内容识别，用 DecodeConfig 先守住尺寸边界；如果同一个输入要走两次，就明确设计 reader 的回退方式。

相关问题

为什么只导入 image 包仍然识别不了 PNG？因为 image 包没有自动加载所有格式，PNG/JPEG 解码器需要各自的空白导入。

DecodeConfig 会返回完整图片吗？不会，它返回图像配置和格式信息，适合在完整解码前做尺寸判断。

image.Decode 是按扩展名判断吗？不是，解码器根据输入内容注册的格式信息进行匹配。