使用Go实现邮箱验证码API功能

go get github.com/jordan-wright/email
go get -u github.com/gin-gonic/gin
go get github.com/redis/go-redis/v9

package main
import (
	"fmt"
	"net/smtp"
	"github.com/jordan-wright/email"
)
func main() {
	targetMailBox := ""          // 目标邮箱
	smtpServer := "smtp.163.com" // smtp服务器
	emailAddr := ""              // 要发件的邮箱地址
	smtpKey := ""                // 获取的smtp密钥
	em := email.NewEmail()
	em.From = fmt.Sprintf("Go-Cloud-Disk <%s>", emailAddr) // 发件人
	em.To = []string{targetMailBox}                        // 目标邮箱
	// email title
	em.Subject = "Email Confirm Test" // 标题
	// build email content
	em.Text = []byte("test") // 内容
	// 调用接口发送邮件
    // 此处端口号不一定为25使用对应邮箱时需要具体更换
	em.Send(smtpServer+":25", smtp.PlainAuth("", emailAddr, smtpKey, smtpServer))
}

package main
import (
	"context"
	"os"
	"strconv"
	"github.com/redis/go-redis/v9"
)
var RedisClient *redis.Client
// init redis
func Redis() {
	db, _ := strconv.ParseUint(os.Getenv("REDIS_DB"), 10, 64) //可不填
	client := redis.NewClient(&redis.Options{
		Addr:       "127.0.0.1:6379", // Redis默认地址
		Password:   "", // Redis密码，一般建立时不设置
		DB:         int(db),
		MaxRetries: 1,
	})
	_, err := client.Ping(context.Background()).Result()
	if err != nil {
		panic("can't connect redis")
	}
	RedisClient = client
}

func VerifyEmailFormat(email string) bool {
	pattern := `^[^\s@]+@[^\s@]+\.[^\s@]+$` //match email
	reg := regexp.MustCompile(pattern)
	return reg.MatchString(email)
}

// SendConfirmMessage传入目标邮箱以及对应的Code码即可发送验证码邮件
func SendConfirmMessage(targetMailBox string, code string) error {
	em := email.NewEmail()
	em.From = fmt.Sprintf("Go-Cloud-Disk <%s>", os.Getenv("EMAIL_ADDR"))
	em.To = []string{targetMailBox}
	// 邮箱标题
	em.Subject = "Email Confirm Code " + code
	// 邮件内容
	emailContentCode := "you confirm code is " + code + ", Your code will expire in 30 minutes"
	emailContentEmail := "you confirm email is " + targetMailBox
	emailContent := emailContentCode + "\n" + emailContentEmail
	em.Text = []byte(emailContent)
	// 调用发送邮件函数
	em.Send(os.Getenv("EMAIL_SMTP_SERVER")+":25", smtp.PlainAuth("", os.Getenv("EMAIL_ADDR"), os.Getenv("EMAIL_SECRET_KEY"), os.Getenv("EMAIL_SMTP_SERVER")))
	return nil
}

type UserSendConfirmEmailService struct {
	UserEmail string `json:"email" form:"email" binding:"required"`
}

func getConfirmCode() string {
	var confirmCode int
	for i := 0; i < 6; i++ {
		confirmCode = confirmCode*10 + (rand.Intn(9) + 1) //随机函数获取值
	}
    // 转换成字符串
	confirmCodeStr := strconv.Itoa(confirmCode)
	return confirmCodeStr
}

func (service *UserSendConfirmEmailService) SendConfirmEmail() string {
	// 检查是否为正确邮箱
	if !VerifyEmailFormat(service.UserEmail) {
		return "NotEmail"
	}
	// 检查是否在3分钟内发送过邮件
	if RedisClient.Get(context.Background(), "send-email:"+service.UserEmail).Val() != "" {
		return "HasSendCode"
	}
    // 获取六位数邮箱验证码
	code := getConfirmCode()
    // 将其存储至Redis中，由于Redis为KV键值对存储所以需要定义前缀方便使用
	RedisClient.Set(context.Background(), "email:"+service.UserEmail, code, time.Minute*30)
    // 发送邮件，此处为方便起见没有处理返回值
	SendConfirmMessage(service.UserEmail, code)
	// 设置每个邮箱发送邮件的时间 此处设置为3分钟，由于Redis为KV键值对存储所以需要定义前缀方便使用
	RedisClient.Set(context.Background(), "send-email:"+service.UserEmail, code, time.Minute*3)
	return "true"
}

func ConfirmUserEmail(c *gin.Context) {
	var service UserSendConfirmEmailService
	if err := c.ShouldBind(&service); err != nil { //使用ShouldBind绑定传入参数
		c.JSON(200, "ParmaErr")
		return
	}
	res := service.SendConfirmEmail()
	c.JSON(200, res)
}

func main() {
	Redis() // 初始化Redis
	server := gin.Default()
	server.POST("email", ConfirmUserEmail) // 注册函数至localhost:3030/email
	server.Run(":3030") // 服务具体运行端口
}

code := cache.RedisClient.Get(context.Background(), "email:"+service.UserEmail).Val()