Golang Gin实践连载十五生成二维码、合并海报

如果对你有所帮助，欢迎点个 Star ?

在本章节，将实现如下功能细项：

1、生成二维码

2、合并海报（背景图 + 二维码）

实现

首先，你需要在 App 配置项中增加二维码及其海报的存储路径，我们约定配置项名称为，值为 qrcode/

经过多节连载的你应该能够完成，若有不懂可参照

工具包

考虑生成二维码这一动作贴合工具包的定义，且有公用的可能性，新建 pkg/qrcode/qrcode.go 文件，写入内容：

package qrcode
import (
    "image/jpeg"
    "github.com/boombuler/barcode"
    "github.com/boombuler/barcode/qr"
    "github.com/EDDYCJY/go-gin-example/pkg/file"
    "github.com/EDDYCJY/go-gin-example/pkg/setting"
    "github.com/EDDYCJY/go-gin-example/pkg/util"
)
type QrCode struct {
    URL    string
    Width  int
    Height int
    Ext    string
    Level  qr.ErrorCorrectionLevel
    Mode   qr.Encoding
}
const (
    EXT_JPG = ".jpg"
)
func NewQrCode(url string, width, height int, level qr.ErrorCorrectionLevel, mode qr.Encoding) *QrCode {
    return &QrCode{
        URL:    url,
        Width:  width,
        Height: height,
        Level:  level,
        Mode:   mode,
        Ext:    EXT_JPG,
    }
}
func GetQrCodePath() string {
    return setting.AppSetting.QrCodeSavePath
}
func GetQrCodeFullPath() string {
    return setting.AppSetting.RuntimeRootPath + setting.AppSetting.QrCodeSavePath
}
    return setting.AppSetting.PrefixUrl + "/" + GetQrCodePath() + name
}
    return util.EncodeMD5(value)
}
func (q *QrCode) GetQrCodeExt() string {
    return q.Ext
}
func (q *QrCode) CheckEncode(path string) bool {
    src := path + GetQrCodeFileName(q.URL) + q.GetQrCodeExt()
    if file.CheckNotExist(src) == true {
        return false
    }
    return true
}
func (q *QrCode) Encode(path string) (string, string, error) {
    name := GetQrCodeFileName(q.URL) + q.GetQrCodeExt()
    src := path + name
    if file.CheckNotExist(src) == true {
        code, err := qr.Encode(q.URL, q.Level, q.Mode)
        if err != nil {
            return "", "", err
        }
        code, err = barcode.Scale(code, q.Width, q.Height)
        if err != nil {
            return "", "", err
        }
        f, err := file.MustOpen(name, path)
        if err != nil {
            return "", "", err
        }
        defer f.Close()
        err = jpeg.Encode(f, code, nil)
        if err != nil {
            return "", "", err
        }
    }
    return name, path, nil
}

这里主要聚焦 func (q *QrCode) Encode 方法，做了如下事情：

获取二维码生成路径
创建二维码
缩放二维码到指定大小
新建存放二维码图片的文件
将图像（二维码）以 JPEG 4：2：0 基线格式写入文件

另外在 jpeg.Encode(f, code, nil) 中，第三个参数可设置其图像质量，默认值为 75

路由方法

1、第一步

在 routers/api/v1/article.go 新增 GenerateArticlePoster 方法用于接口开发

在 routers/router.go 的 apiv1 中新增 apiv1.POST("/articles/poster/generate", v1.GenerateArticlePoster) 路由

3、第三步

修改 GenerateArticlePoster 方法，编写对应的生成逻辑，如下：

const (
    QRCODE_URL = "https://github.com/EDDYCJY/blog#gin%E7%B3%BB%E5%88%97%E7%9B%AE%E5%BD%95"
)
func GenerateArticlePoster(c *gin.Context) {
    appG := app.Gin{c}
    qrc := qrcode.NewQrCode(QRCODE_URL, 300, 300, qr.M, qr.Auto)
    path := qrcode.GetQrCodeFullPath()
    _, _, err := qrc.Encode(path)
        appG.Response(http.StatusOK, e.ERROR, nil)
        return
    }
    appG.Response(http.StatusOK, e.SUCCESS, nil)
}

通过 POST 方法访问 http://127.0.0.1:8000/api/v1/articles/poster/generate?token=$token（注意 $token）

通过检查两个点确定功能是否正常，如下：

1、访问结果是否 200

2、本地目录是否成功生成二维码图片

合并海报

在这一节，将实现二维码图片与背景图合并成新的一张图，可用于常见的宣传海报等业务场景

背景图

service 方法

打开 service/article_service 目录，新建 article_poster.go 文件，写入内容：

这里重点留意 func (a *ArticlePosterBg) Generate() 方法，做了如下事情：

获取二维码存储路径
生成二维码图像
检查合并后图像（指的是存放合并后的海报）是否存在
若不存在，则生成待合并的图像 mergedF
打开事先存放的背景图 bgF
打开生成的二维码图像 qrF
解码 bgF 和 qrF 返回 image.Image
创建一个新的 RGBA 图像
在 RGBA 图像上绘制背景图（bgF）
在已绘制背景图的 RGBA 图像上，在指定 Point 上绘制二维码图像（qrF）
将绘制好的 RGBA 图像以 JPEG 4：2：0 基线格式写入合并后的图像文件（mergedF）

新增错误码，

路由方法

打开 routers/api/v1/article.go 文件，修改 GenerateArticlePoster 方法，编写最终的业务逻辑（含生成二维码及合并海报），如下：

const (
    QRCODE_URL = "https://github.com/EDDYCJY/blog#gin%E7%B3%BB%E5%88%97%E7%9B%AE%E5%BD%95"
)
func GenerateArticlePoster(c *gin.Context) {
    appG := app.Gin{c}
    article := &article_service.Article{}
    qr := qrcode.NewQrCode(QRCODE_URL, 300, 300, qr.M, qr.Auto) // 目前写死 gin 系列路径，可自行增加业务逻辑
    posterName := article_service.GetPosterFlag() + "-" + qrcode.GetQrCodeFileName(qr.URL) + qr.GetQrCodeExt()
    articlePoster := article_service.NewArticlePoster(posterName, article, qr)
    articlePosterBgService := article_service.NewArticlePosterBg(
        "bg.jpg",
        articlePoster,
        &article_service.Rect{
            X0: 0,
            Y0: 0,
            X1: 550,
            Y1: 700,
        },
        &article_service.Pt{
            X: 125,
            Y: 298,
        },
    )
    _, filePath, err := articlePosterBgService.Generate()
    if err != nil {
        appG.Response(http.StatusOK, e.ERROR_GEN_ARTICLE_POSTER_FAIL, nil)
        return
    }
    appG.Response(http.StatusOK, e.SUCCESS, map[string]string{
        "poster_url":      qrcode.GetQrCodeFullUrl(posterName),
        "poster_save_url": filePath + posterName,
    })
}