乐趣区

关于chatgpt:让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

此文章首发于我的独立博客:https://blog.xiaoz.org/archives/18626

退出移动版