关于chatgpt:让ChatGPT帮我写了一个Golang图像压缩函数

40次阅读

共计 2017 个字符，预计需要花费 6 分钟才能阅读完成。

最近在应用 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 答复，哪些问题须要咱们本人解决。