golang 生成二维码海报的实现代码

 

目标:生成一个二维码,加上 logo,扫码结果为baidu.com

准备工作,准备一个logo,我是借用了CSDN的logo,保存为 avatar.png,置于main.go代码同目录下

main.go代码:

package main

import (
    "errors"
    "fmt"
    "image"
    "image/draw"
    "image/png"
    "os"

    "github.com/nfnt/resize"
    "github.com/skip2/go-qrcode"
)

var err error

func main() {
    img, _ := createAvatar()
    f, _ := os.Create("image.png")
    png.Encode(f, img)

}

func createAvatar() (image.Image, error) {
    var (
        bgImg      image.Image
        offset     image.Point
        avatarFile *os.File
        avatarImg  image.Image
    )

    bgImg, err = createQrCode("http://www.baidu.com")

    if err != nil {
        fmt.Println("创建二维码失败:", err)
        return nil, errors.New("创建二维码失败")
    }
    avatarFile, err = os.Open("./avatar.png")
    avatarImg, err = png.Decode(avatarFile)
    avatarImg = ImageResize(avatarImg, 40, 40)
    b := bgImg.Bounds()

    // 设置为居中
    offset = image.Pt((b.Max.X-avatarImg.Bounds().Max.X)/2, (b.Max.Y-avatarImg.Bounds().Max.Y)/2)

    m := image.NewRGBA(b)

    draw.Draw(m, b, bgImg, image.Point{X: 0, Y: 0}, draw.Src)

    draw.Draw(m, avatarImg.Bounds().Add(offset), avatarImg, image.Point{X: 0, Y: 0}, draw.Over)

    return m, err
}

func createQrCode(content string) (img image.Image, err error) {
    var qrCode *qrcode.QRCode

    qrCode, err = qrcode.New(content, qrcode.Highest)

    if err != nil {
        return nil, errors.New("创建二维码失败")
    }
    qrCode.DisableBorder = true

    img = qrCode.Image(150)

    return img, nil
}

func ImageResize(src image.Image, w, h int) image.Image {
    return resize.Resize(uint(w), uint(h), src, resize.Lanczos3)
}

会在当前目录下生成 image.png ,打开之,用手机扫描,会访问baidu.com

 

接下来,

将二维码合成到海报

准备一张海报背景图,我是从 baidu下的,结果大都是.jpg ,用画图另存为 bg.png,置于main.go目录下

main.go代码:

package main

import (
    "errors"
    "fmt"
    "image"
    "image/draw"
    "image/png"
    "os"
    "path"

    "github.com/nfnt/resize"
    "github.com/skip2/go-qrcode"
)

var err error

func main() {
    var (
        bgFile    *os.File
        bgImg     image.Image
        qrCodeImg image.Image
        offset    image.Point
    )

    // 01: 打开背景图片
    bgFile, err = os.Open("./bg.png")
    if err != nil {
        fmt.Println("打开背景图片失败", err)
        return
    }

    defer bgFile.Close()

    // 02: 编码为图片格式
    bgImg, err = png.Decode(bgFile)
    if err != nil {
        fmt.Println("背景图片编码失败:", err)
        return
    }

    // 03: 生成二维码
    qrCodeImg, err = createAvatar()
    if err != nil {
        fmt.Println("生成二维码失败:", err)
        return
    }

    offset = image.Pt(200, 200) //用于调整二维码在背景图片上的位置

    b := bgImg.Bounds()

    m := image.NewRGBA(b)

    draw.Draw(m, b, bgImg, image.Point{X: 0, Y: 0}, draw.Src)

    draw.Draw(m, qrCodeImg.Bounds().Add(offset), qrCodeImg, image.Point{X: 0, Y: 0}, draw.Over)

    // 上传至oss时这段要改
    i, _ := os.Create(path.Base("a2.png"))

    _ = png.Encode(i, m)

    defer i.Close()

}

func createAvatar() (image.Image, error) {
    var (
        bgImg      image.Image
        offset     image.Point
        avatarFile *os.File
        avatarImg  image.Image
    )

    bgImg, err = createQrCode("http://www.baidu.com")

    if err != nil {
        fmt.Println("创建二维码失败:", err)
        return nil, errors.New("创建二维码失败")
    }
    avatarFile, err = os.Open("./avatar.png")
    avatarImg, err = png.Decode(avatarFile)
    avatarImg = ImageResize(avatarImg, 40, 40)  //用于调整avatar.png在二维码中的大小
    b := bgImg.Bounds()

    // 设置为居中
    offset = image.Pt((b.Max.X-avatarImg.Bounds().Max.X)/2, (b.Max.Y-avatarImg.Bounds().Max.Y)/2)

    m := image.NewRGBA(b)

    draw.Draw(m, b, bgImg, image.Point{X: 0, Y: 0}, draw.Src)

    draw.Draw(m, avatarImg.Bounds().Add(offset), avatarImg, image.Point{X: 0, Y: 0}, draw.Over)

    return m, err
}

func createQrCode(content string) (img image.Image, err error) {
    var qrCode *qrcode.QRCode

    qrCode, err = qrcode.New(content, qrcode.Highest)

    if err != nil {
        return nil, errors.New("创建二维码失败")
    }
    qrCode.DisableBorder = true

    img = qrCode.Image(150)

    return img, nil
}

func ImageResize(src image.Image, w, h int) image.Image {
    return resize.Resize(uint(w), uint(h), src, resize.Lanczos3)
}

会生成a2.png

 

 

参考:https://www.yht7.com/news/29937

你可能感兴趣的:(golang 生成二维码海报的实现代码)