利用ChatGPT编写一个Golang图像压缩函数实践整理

作者:袖梨 2026-08-19

平时做技术实践时,很多问题不是概念不会,而是细节没串起来。拿“利用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编写Golang图像压缩函数这部分内容适合结合实际项目边做边理解。先抓住核心思路,再逐步补上细节和边界处理,最后效果会更稳定,也更容易复用。

相关文章

精彩推荐