尝试在内存中按需转换图像并返回转换后的图像

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《尝试在内存中按需转换图像并返回转换后的图像》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

我正在尝试将图像即时转换为 png，源可以是 tiff、jpeg 或 png，并且我希望它始终以 png 形式返回。

目前，我在 http 响应中获取图像时遇到问题，因此任何指针都将受到高度赞赏，因为 res.write 中的当前错误是“无法使用 'destimage'（类型错误）作为类型 []字节”

如果可能的话，图像处理最好在内存中完成，而不是在文件系统上完成。

import (
    "bufio"
    "bytes"
    "crypto/tls"
    "log"
    "net/http"

    "github.com/sunshineplan/imgconv"
)

var client = http.Client{}

func main() {

    http.HandleFunc("/photo", cutterHandler)
    http.ListenAndServe(":8000", nil)
}

func cutterHandler(res http.ResponseWriter, req *http.Request) {
    url := "https://proxy.local/Image.aspx?id=CAB96C9C-DA56-4BC9-8626-FC5C95FF4D95"
    http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
    reqImg, err := client.Get(url)
    if err != nil {
        log.Println(res, "Error %d", err)
        req.Response.StatusCode = http.StatusInternalServerError
        return
    }

    srcImage, err := imgconv.Decode(reqImg.Body)
    if err != nil {
        log.Println(err)
        req.Response.StatusCode = http.StatusInternalServerError
        return
    }

    convertedImage := imgconv.Resize(srcImage, imgconv.ResizeOption{Width: 128, Height: 128})
    var b bytes.Buffer
    buffer := bufio.NewWriter(&b)
    destImage := imgconv.Write(buffer, convertedImage, imgconv.FormatOption{Format: imgconv.PNG})
    res.WriteHeader(http.StatusOK)
    res.Header().Set("Content-Type", "image/png")
    res.Write(destImage)
}

正确答案

问题是 ResponseWriter.Write() 方法接受 bytes（[]byte）的切片，但您试图传递 error type（内置接口类型）。

如果我尝试构建您的代码，我会收到以下错误消息：

在源代码中，我看到“destimage”变量在第 41 行初始化，并具有 error type（方法 imgconv.Write() 返回错误）。

destimage := imgconv.write(buffer, convertedimage, imgconv.formatoption{format: imgconv.png})

要修复该错误，您需要更改第 41 行的 imgconv.write() 方法调用。可以通过以下方式完成：

convertedimage := imgconv.resize(srcimage, imgconv.resizeoption{width: 128, height: 128})

res.header().set("content-type", "image/png")
res.writeheader(http.statusok)

err = imgconv.write(res, convertedimage, imgconv.formatoption{format: imgconv.png})
if err != nil {
    log.println(err)
}

为了演示，我写了 another version，您可能会觉得有用。

package main

import (
    "bytes"
    "crypto/tls"
    "fmt"
    "io"
    "log"
    "net/http"
    "strconv"

    "github.com/sunshineplan/imgconv"
)

var client = http.client{
    transport: clonetransport(),
}

func main() {
    imageurl := imageurl("https://go.dev/blog/go-brand/go-logo/jpg/go-logo_blue.jpg")
    http.handle("/logos/golang.png", imageurl)
    log.fatal(http.listenandserve(":8080", nil))
}

type imageurl string

func (imageurl imageurl) servehttp(w http.responsewriter, req *http.request) {
    if req.method != http.methodget {
        w.writeheader(http.statusmethodnotallowed)
        return
    }

    res, err := client.get(string(imageurl))

    if err != nil {
        log.printf("fetch %q image: %v", imageurl, err)
        w.writeheader(http.statusinternalservererror)
        return
    }

    if res.statuscode != http.statusok {
        log.printf("fetch %q image: %s [%d]", imageurl, res.status, res.statuscode)
        w.writeheader(http.statusinternalservererror)
        return
    }

    buffer := new(bytes.buffer)
    err = convert(buffer, res.body)

    defer res.body.close()

    if err != nil {
        log.printf("convert %q image: %v", imageurl, err)
        w.writeheader(http.statusinternalservererror)
        return
    }

    w.header().set("content-type", "image/png")
    w.header().set("content-length", strconv.formatint(int64(buffer.len()), 10))
    w.writeheader(http.statusok)

    if _, err = buffer.writeto(w); err != nil {
        log.printf("write data: %v", err)
    }
}

func convert(w io.writer, r io.reader) error {
    srcimage, err := imgconv.decode(r)

    if err != nil {
        return fmt.errorf("decode image: %w", err)
    }

    img := imgconv.resize(srcimage, imgconv.resizeoption{width: 128, height: 128})
    err = imgconv.write(w, img, imgconv.formatoption{format: imgconv.png})

    if err != nil {
        return fmt.errorf("encode image: %w", err)
    }

    return nil
}

func clonetransport() *http.transport {
    transport := http.defaulttransport.(*http.transport).clone()
    transport.tlsclientconfig = &tls.config{
        insecureskipverify: true,
    }

    return transport
}

此外，another variant的convert()函数不支持cmyk (tiff)，使用/x/image/tiff和resize包。

import (
    "fmt"
    "image"
    "image/png"
    "io"

    _ "golang.org/x/image/tiff"
    _ "image/jpeg"

    "github.com/nfnt/resize"
)

func convert(w io.Writer, r io.Reader) error {
    img, _, err := image.Decode(r)

    if err != nil {
        return fmt.Errorf("decode image: %w", err)
    }

    resizedImage := resize.Resize(128, 128, img, resize.Bicubic)
    err = png.Encode(w, resizedImage)

    if err != nil {
        return fmt.Errorf("encode image: %w", err)
    }

    return nil
}

到这里，我们也就讲完了《尝试在内存中按需转换图像并返回转换后的图像》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！