首页 > 文章列表 > 在Gin框架中实现图片处理和添加水印功能

在Gin框架中实现图片处理和添加水印功能

图片处理 水印功能 Gin框架
328 2024-03-26

在如今的互联网时代,图片的应用越来越广泛,图片处理也成为了Web应用开发中不可或缺的一部分。而且,为了减少盗图等问题,加水印也成为了很多Web应用的必需品。本文将介绍如何使用Gin框架实现图片处理及水印功能。

Gin框架简介

Gin是一个轻量级的Web框架,它具有快速、灵活、简洁等特点,因此在开发中越来越受到欢迎。Gin基于Go语言,可以很好地应用于高并发的Web应用开发中。

实现图片处理

在使用Gin框架实现图片处理功能之前,我们需要先安装一些需要用到的Go包:

  • github.com/gin-gonic/gin
  • github.com/nfnt/resize
  • github.com/disintegration/imaging

安装好上述Go包之后,我们开始实现图片处理功能,下面是示例代码:

package main

import (
    "github.com/gin-gonic/gin"
    "github.com/disintegration/imaging"
    "github.com/nfnt/resize"
    "net/http"
)

func main() {
    router := gin.Default()

    router.GET("/image", func(context *gin.Context) {
        path := context.Query("path")
        width := context.Query("width")
        height := context.Query("height")
        mode := context.Query("mode")
        if path == "" {
            context.String(http.StatusBadRequest, "缺少参数path")
            return
        }
        src, err := imaging.Open(path)
        if err != nil {
            context.String(http.StatusBadRequest, "打开图片失败:%s", err.Error())
            return
        }
        w, h := src.Bounds().Max.X, src.Bounds().Max.Y
        var img image.Image
        if width == "" && height == "" {
            img = imaging.Clone(src)
        } else if width == "" {
            h64, err := strconv.ParseUint(height, 10, 64)
            if err != nil {
                context.String(http.StatusBadRequest, "参数height错误:%s", err.Error())
                return
            }
            h = int(h64)
            img = resize.Resize(0, h, src, resize.Lanczos3)
        } else if height == "" {
            w64, err := strconv.ParseUint(width, 10, 64)
            if err != nil {
                context.String(http.StatusBadRequest, "参数width错误:%s", err.Error())
                return
            }
            w = int(w64)
            img = resize.Resize(w, 0, src, resize.Lanczos3)
        } else {
            w64, err := strconv.ParseUint(width, 10, 64)
            if err != nil {
                context.String(http.StatusBadRequest, "参数width错误:%s", err.Error())
                return
            }
            h64, err := strconv.ParseUint(height, 10, 64)
            if err != nil {
                context.String(http.StatusBadRequest, "参数height错误:%s", err.Error())
                return
            }
            w = int(w64)
            h = int(h64)

            var m resize.InterpolationFunction
            switch mode {
            case "nearest_neighbor":
                m = resize.NearestNeighbor
            case "bilinear":
                m = resize.Bilinear
            case "bicubic":
                m = resize.Bicubic
            case "mitchell_netravali":
                m = resize.MitchellNetravali
            case "lanczos2":
                m = resize.Lanczos2
            case "lanczos3":
                m = resize.Lanczos3
            default:
                m = resize.Lanczos3
            }
            img = resize.Resize(w, h, src, m)
        }
        buf := new(bytes.Buffer)
        err = imaging.Encode(buf, img, imaging.JPEG)
        if err != nil {
            context.String(http.StatusInternalServerError, "编码图片失败:%s", err.Error())
            return
        }
        context.Data(http.StatusOK, "image/jpeg", buf.Bytes())
    })

    router.Run(":8080")
}

在代码中,我们实现了一个/image的接口,通过URL参数path、width、height、mode来控制图片的大小和质量。其中,path表示图片的路径,width和height控制图片的宽度和高度,mode用来指定图片缩放的算法,例如:nearest_neighbor(最近邻插值法)、bilinear(双线性插值法)、bicubic(双三次插值法)、mitchell_netravali(mitchell和netravali插值法)、lanczos2(lanczos resize插值法)、lanczos3(lanczos resize插值法),默认为Lanczos3算法。

实现图片水印

在实现图片水印功能之前,我们还需要安装一个Go包:

  • github.com/disintegration/imaging

安装好该包后,接下来是实现代码:

package main

import (
    "image"
    "image/color"
    "image/draw"
    "net/http"

    "github.com/disintegration/imaging"
    "github.com/gin-gonic/gin"
)

func main() {
    router := gin.Default()

    router.GET("/watermark", func(c *gin.Context) {
        path := c.Query("path")
        name := c.Query("name")
        xstr := c.Query("x")
        ystr := c.Query("y")
        x, y := 0, 0
        if xstr != "" {
            val, err := strconv.Atoi(xstr)
            if err != nil {
                c.String(http.StatusBadRequest, "参数x错误:%s", err.Error())
                return
            }
            x = val
        }
        if ystr != "" {
            val, err := strconv.Atoi(ystr)
            if err != nil {
                c.String(http.StatusBadRequest, "参数y错误:%s", err.Error())
                return
            }
            y = val
        }
        if path == "" || name == "" {
            c.String(http.StatusBadRequest, "缺少参数")
            return
        }
        src, err := imaging.Open(path)
        if err != nil {
            c.String(http.StatusBadRequest, "打开图片失败:%s", err.Error())
            return
        }
        water := imaging.New(50, 50, color.NRGBA{255, 0, 0, 255})
        draw.Draw(water, water.Bounds(), &image.Uniform{color.Transparent}, image.Point{}, draw.Src)
        draw.Draw(water, water.Bounds(), image.NewUniform(color.RGBA{255, 0, 0, 255}), image.Point{}, draw.Src)
        bounds := src.Bounds()
        image := imaging.Overlay(src, water, image.Point{x, y}, 1.0)
        buf := new(bytes.Buffer)
        err = imaging.Encode(buf, image, imaging.JPEG)
        if err != nil {
            c.String(http.StatusInternalServerError, "编码图片失败:%s", err.Error())
            return
        }
        c.Data(http.StatusOK, "image/jpeg", buf.Bytes())
    })

    router.Run(":8080")
}

在代码中,我们实现了一个/watermark的接口,通过URL参数path、name、x、y来控制图片的水印位置和质量。其中,path表示图片的路径,name指定水印文本,x和y表示水印的位置,需要注意的是x和y的值是相对于图片左上角的偏移量。

结语

本文介绍了如何使用Gin框架实现图片处理及水印功能。其中图片处理主要是通过resize包来实现的,而图片水印则是通过图像合成技术来实现的。通过本文的介绍,相信大家可以更好地应用Gin框架来开发高效、灵活的Web应用,同时也能提升在图片处理和水印方面的技术水平。