利用Go语言上传图像并生成缩略图

来源:互联网 发布:java多线程实例代码 编辑:程序博客网 时间:2024/05/29 14:13

承前文:Go语言中对图像进行缩放


//利用Go语言上传图像并生成缩略图

func upload(w http.ResponseWriter, req *http.Request, link string) {
// Upload of a new image.
// Copied from Moustachio demo.
f, _, err := req.FormFile("image")
if err != nil {
fmt.Fprintf(w, "You need to select an image to upload.\n")
return
}
defer f.Close()


i, _, err := image.Decode(f)
if err != nil {
panic(err)
}


// Convert image to 128x128 gray+alpha.
b := i.Bounds()
const max = 128
// If it's gigantic, it's more efficient to downsample first
// and then resize; resizing will smooth out the roughness.
var i1 *image.RGBA
if b.Dx() > 4*max || b.Dy() > 4*max {
w, h := 2*max, 2*max
if b.Dx() > b.Dy() {
h = b.Dy() * h / b.Dx()
} else {
w = b.Dx() * w / b.Dy()
}
i1 = resize.Resample(i, b, w, h)
} else {
// "Resample" to same size, just to convert to RGBA.
i1 = resize.Resample(i, b, b.Dx(), b.Dy())
}
b = i1.Bounds()


// Encode to PNG.
dx, dy := 128, 128
if b.Dx() > b.Dy() {
dy = b.Dy() * dx / b.Dx()
} else {
dx = b.Dx() * dy / b.Dy()
}
i128 := resize.ResizeRGBA(i1, i1.Bounds(), dx, dy)


var buf bytes.Buffer
if err := png.Encode(&buf, i128); err != nil {
panic(err)
}


h := md5.New()
h.Write(buf.Bytes())
tag := fmt.Sprintf("%x", h.Sum(nil))[:32]


ctxt := fs.NewContext(req)
if err := ctxt.Write("qr/upload/"+tag+".png", buf.Bytes()); err != nil {
panic(err)
}


// Redirect with new image tag.
// Redirect to draw with new image tag.
http.Redirect(w, req, req.URL.Path+"?"+url.Values{"i": {tag}, "url": {link}}.Encode(), 302)
}