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

生成带头像的二维码

import (
 "errors"
 "fmt"
 "github.com/nfnt/resize"
 "github.com/skip2/go-qrcode"
 "image"
 "image/draw"
 "image/png"
 "os"
 ) 

var err error 

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)
}

预览

合成到海报

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(426, 475) 

 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 ossClient() (bucket *alioss.Bucket, err error) {
 var (
 key string = "xxx"
 bucketName string = "xxx"
 secret string = "xxx"
 endpoint string = "xxx"
 ) 

 // 创建OSSClient实例。
 client, err = alioss.New(endpoint, key, secret) 

 if err != nil {
 fmt.Println("获取阿里云oss实例失败:", err)
 return nil, errors.New("资源服务器配置有误")
 }
 // 获取存储空间。
 bucket, err = client.Bucket(bucketName) 

 if err != nil {
 logger.Error("获取阿里云oss存储空间错误:", err)
 return nil, errors.New("获取存储服务器失败")
 }
 return bucket, nil
}

开始上传

将之前注释的那段删除 修改成这段

bucket, err = ossClient() 

if err != nil {
 fmt.Println("error:", err)
 return
} 

imgBuff := bytes.NewBuffer(nil) //开辟一个新的空buff
err = png.Encode(imgBuff, m) //img写入到buff 

if err != nil {
 fmt.Println("写入缓冲区失败",err)
 return
} 

err = bucket.PutObject("qrcode/2-1.png", imgBuff) 

fmt.Println("error:", err)

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。