让ChatGPT帮我写了一个Golang图像压缩函数
最近在使用Wails开发ImgURL客户端时，需要对图片进行压缩再上传，为了提高效率，节省时间，这次直接让ChatGPT来帮我完成这个需求。
提出需求
我对ChatGPT提出了如下要求：
请帮我编写一个能够接收图片路径并进行压缩的Golang函数。该函数需要支持jpg和png格式的图片，并且会校验MIME类型。如果图片类型不是jpg或png，则函数不会进行处理。压缩后的图片路径与原路径相同，比如/tmp/1.png经过压缩后依然为/tmp/1.png。函数的返回值为err。
第一次给我返回的代码中并没有去校验MIME类型，于是我再次让他跟进需求：
您似乎不是根据MIME类型来判断而是根据扩展名来判断的？请根据MIME类型来判断，如果不是匹配的MIME类型，返回nil，压缩后图片路径不改变，比如/tmp/1.png压缩后的图片路径依然为/tmp/1.png
这次给的代码看起来没问题，但实际运行的时候并没有压缩图片，原来MIME类型判断错了，继续提问：
为什么我调用函数后图片大小没有任何变化？
ChatGPT勇于承认错误：
抱歉给您带来困扰。我发现在代码中有一个错误，http.DetectContentType 方法应该接收文件的前512个字节而不是文件路径。下面是修改后的 compressImage 函数。
这次给的代码未做任何修改，可以成功运行并压缩图像。
package main
import (
"errors"
"image"
"image/jpeg"
"image/png"
"mime"
"net/http"
"os"
)
func compressImage(imagePath string) error {
file, err := os.Open(imagePath)
if err != nil {
return err
}
defer file.Close()
// 读取文件的前512个字节
buffer := make([]byte, 512)
, err = file.Read(buffer)
if err != nil {
return err
}
file.Seek(0, 0)
// 根据MIME类型检查
contentType := http.DetectContentType(buffer)
var img image.Image
var format string
switch contentType {
case "image/jpeg":
img, err = jpeg.Decode(file)
format = "jpeg"
case "image/png":
img, err = png.Decode(file)
format = "png"
default:
return nil
}
if err != nil {
return err
}
// 关闭并重新打开文件进行压缩
file.Close()
file, err = os.OpenFile(imagePath, os.O_WRONLY|os.O_TRUNC, 0644)
if err != nil {
return err
}
defer file.Close()
var opt jpeg.Options
opt.Quality = 75 // 设置压缩质量为75
switch format {
case "jpeg":
err = jpeg.Encode(file, img, &opt)
case "png":
encoder := png.Encoder{CompressionLevel: png.BestCompression}
err = encoder.Encode(file, img)
}
return err
}
func main() {
imagePath := "/tmp/1.png"
err := compressImage(imagePath)
if err != nil {
println("Error compressing image:", err.Error())
} else {
println("Image compressed successfully")
}
}
并且对代码做出了解释：
这个函数中，JPEG图片采用有损压缩，而PNG图片采用无损压缩。
对于JPEG图片，压缩质量设置为75（范围是1-100，数值越高，质量越好，但文件大小也越大）。
对于PNG图片，这个函数使用了最佳压缩级别（png.BestCompression）。这会进行无损压缩，保留图像的所有原始信息，但可能会导致压缩时间较长。
结语
正确使用ChatGPT确实能有效提高效率，但不能完全依赖ChatGPT。还需要自己的辨别能力，因为ChatGPT给出的答案也不完全正确。我们需要自己辨别和加以引导，才能得到正确的答案。将ChatGPT视为辅助工具则非常不错。此外，我们应该分辨出哪些问题能够由ChatGPT回答，哪些问题需要我们自己解决。
无法直接使用ChatGPT的朋友也可以试试AsBot：https://dwz.ovh/ab3iv
上一篇
使用 Wails：结合 Go 和 Web 技术轻松构建桌面应用程序
下一篇
CloudCone推出了CDN Nexus，覆盖亚洲节点，附简单上手指南