从Go中的HTTP请求主体读取图像

时间:2011-12-27 19:54:58

标签: http go

我正在玩Go(有史以来第一次),我想构建一个工具来从互联网上检索图像并剪切它们(甚至调整大小)但我仍然坚持第一步。

package main

import (
  "fmt"
  "http"
)

var client = http.Client{}

func cutterHandler(res http.ResponseWriter, req *http.Request) {
  reqImg, err := client.Get("http://www.google.com/intl/en_com/images/srpr/logo3w.png")
  if err != nil {
    fmt.Fprintf(res, "Error %d", err)
    return
  }
  buffer := make([]byte, reqImg.ContentLength)
  reqImg.Body.Read(buffer)
  res.Header().Set("Content-Length", fmt.Sprint(reqImg.ContentLength)) /* value: 7007 */
  res.Header().Set("Content-Type", reqImg.Header.Get("Content-Type")) /* value: image/png */
  res.Write(buffer)
}

func main() {
  http.HandleFunc("/cut", cutterHandler)
  http.ListenAndServe(":8080", nil) /* TODO Configurable */
}

我可以申请图片(让我们使用Google徽标)并获取其类型和尺寸。

实际上,我只是重新编写图像(将其视为玩具“代理”),设置Content-Length和Content-Type并将字节切片写回来,但我在某处弄错了。看看它在Chromium 12.0.742.112(90304)上呈现的最终图像的样子:

Grotesque result

我还检查了下载的文件,它是一个7007字节的PNG图像。如果我们查看请求,它应该正常工作:

  

获取/削减HTTP / 1.1
  User-Agent:curl / 7.22.0(i486-pc-linux-gnu)libcurl / 7.22.0 OpenSSL / 1.0.0e zlib / 1.2.3.4 libidn / 1.23 libssh2 / 1.2.8 librtmp / 2.3
  主持人:127.0.0.1:8080
  接受: /

     

HTTP / 1.1 200 OK
  内容长度:7007
  内容类型:image / png
  日期:2011年12月27日星期二19:51:53 GMT

     

[PNG数据]

你认为我在这里做错了什么?

免责声明:我正在抓挠自己的痒,所以可能我使用的是错误的工具:)无论如何,我可以在Ruby上实现它,但在此之前我想尝试一下Go。

更新:仍在抓痒但是...我认为这将是一个很好的侧面项目,所以我打开它https://github.com/imdario/go-lazor如果它没用,至少有人可以找到有用的与用于开发它的参考。他们是给我的。

2 个答案:

答案 0 :(得分:8)

我尝试了你的代码并注意到你所服务的图像大小合适,但文件的内容超过了某一点都是0x00。

查看io.Reader documentation。要记住的重要一点是Read读取最多您请求的字节数。它可以读取更少,没有返回错误。 (您应该检查错误,但这不是问题。)

如果要确保缓冲区已满,请使用io.ReadFull。在这种情况下,只需使用io.Copy复制Reader的全部内容即可。

记住关闭HTTP请求主体也很重要。

我会用这种方式重写代码:

package main

import (
    "fmt"
    "http"
    "io"
)

var client = http.Client{}

func cutterHandler(res http.ResponseWriter, req *http.Request) {
    reqImg, err := client.Get("http://www.google.com/intl/en_com/images/srpr/logo3w.png")
    if err != nil {
        fmt.Fprintf(res, "Error %d", err)
        return
    }
    res.Header().Set("Content-Length", fmt.Sprint(reqImg.ContentLength))
    res.Header().Set("Content-Type", reqImg.Header.Get("Content-Type"))
    if _, err = io.Copy(res, reqImg.Body); err != nil {
        // handle error
    }
    reqImg.Body.Close()
}

func main() {
    http.HandleFunc("/cut", cutterHandler)
    http.ListenAndServe(":8080", nil) /* TODO Configurable */
}

答案 1 :(得分:8)

我认为你过得太快了服务事物部分

专注于第一步,下载图片

这里有一个小程序可以将该图像下载到内存中 它适用于我的2011-12-22周版本,对于r60.3,你只需要gofix导入。

package main

import (
    "log"
    "io/ioutil"
    "net/http"
)

const url = "http://www.google.com/intl/en_com/images/srpr/logo3w.png"

func main() {
    // Just a simple GET request to the image URL
    // We get back a *Response, and an error
    res, err := http.Get(url)

    if err != nil {
        log.Fatalf("http.Get -> %v", err)
    }

    // We read all the bytes of the image
    // Types: data []byte
    data, err = ioutil.ReadAll(res.Body)

    if err != nil {
        log.Fatalf("ioutil.ReadAll -> %v", err)
    }

    // You have to manually close the body, check docs
    // This is required if you want to use things like
    // Keep-Alive and other HTTP sorcery.
    res.Body.Close()

    // You can now save it to disk or whatever...
    ioutil.WriteFile("google_logo.png", data, 0666)

    log.Println("I saved your image buddy!")
}

瞧!

这会将图像记录到data内的内存中 完成后,您可以对其进行解码,裁剪并返回浏览器。

希望这有帮助。