寻找更好的方法将内存映像保存到文件

时间:2016-10-07 23:42:07

标签: image go io

代码的目标是下载图像,将其粘贴到更大的父图像上并保存结果。

经过几次失败后,我最终得到了以下可行的代码。

但是,有没有比使用bytes.Bufferwriter更好的方法将目标图像保存到文件/将其传递给httpResponse?

package main

import (
    "image"
    "image/draw"
    "image/jpeg"
    "os"
    // "image/color"
    // "io/ioutil"
    // "fmt"
    "bufio"
    "bytes"
    "log"
    "net/http"
)

func main() {
    // Fetch an image.
    resp, err := http.Get("http://katiebrookekennels.com/wp-content/uploads/2014/10/dog-bone4.jpg")
    if err != nil {
        log.Panic(err)
    }
    defer resp.Body.Close()

    // Keep an in memory copy.
    myImage, err := jpeg.Decode(resp.Body)

    if err != nil {
        log.Panic(err)
    }

    // Prepare parent image where we want to position child image.
    target := image.NewRGBA(image.Rect(0, 0, 800, 800))
    // Draw white layer.
    draw.Draw(target, target.Bounds(), image.White, image.ZP, draw.Src)
    // Draw child image.
    draw.Draw(target, myImage.Bounds(), myImage, image.Point{0, 0}, draw.Src)

    // Encode to jpeg.
    var imageBuf bytes.Buffer
    err = jpeg.Encode(&imageBuf, target, nil)

    if err != nil {
        log.Panic(err)
    }

    // Write to file.
    fo, err := os.Create("img.jpg")
    if err != nil {
        panic(err)
    }
    fw := bufio.NewWriter(fo)

    fw.Write(imageBuf.Bytes())
}

1 个答案:

答案 0 :(得分:3)

jpeg.Encode()期望io.Writer将编码图像写入(JPEG格式)。 *os.Filehttp.ResponseWriter都会实现io.Writer,因此您也可以直接传递文件或HTTP响应编写器而不是bytes.Buffer

将图像直接保存到文件中:

f, err := os.Create("img.jpg")
if err != nil {
    panic(err)
}
defer f.Close()
jpeg.Encode(f, target, nil)

将其作为图像提供:

// w is of type http.ResponseWriter:
jpeg.Encode(w, target, nil)

您可以在这些问题中看到一些示例(保存/加载JPEG和PNG图像):

Draw a rectangle in Golang?

How to add a simple text label to an image in Go?

Change color of a single pixel - Go lang image