Go项目实战之验证码的生成和使用

goshop开源项目的更新

备注：前面项目中用到的代码已经分享到GitHub中去了，并且以后所有项目中会出现的代码都会提交上去，欢迎查阅。 感兴趣的可以点个star哦~ 

今天咱们做一个登录功能中的验证码功能：

1. 封装验证码生成，验证码验证等功能，以便多复用

2. 封装session的中间件

接下来咱们就来实现以上的问题和功能：

• 先封装验证码业务部分

// 生成验证码
func Captcha(ctx *gin.Context) {
   // 默认验证码生成长度6位
	l := captcha.DefaultLen
	// 这里使用viper读取yaml的配置文件参数
	// w 验证码宽度     h验证码高度     l 验证码生成位数
	w, _ := strconv.Atoi(viper.GetString("captcha.w"))
	h, _ := strconv.Atoi(viper.GetString("captcha.h"))
	l, _ = strconv.Atoi(viper.GetString("captcha.length"))
	// 初始化 验证码长度
	captchaId := captcha.NewLen(l)
	// 开启session
	session := sessions.Default(ctx)
	// 设置session，存储验证码数值
	session.Set("captcha", captchaId)
	// 保存到session
	_ = session.Save()
	// 验证码生成
	_ = Serve(ctx.Writer, ctx.Request, captchaId, ".png", viper.GetString("captcha.lang"), false, w, h)
}

// 验证码的header头设置，生成页面显示
func Serve(w http.ResponseWriter, r *http.Request, id, ext, lang string, download bool, width, height int) error {
	w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
	w.Header().Set("Pragma", "no-cache")
	w.Header().Set("Expires", "0")

	var content bytes.Buffer
	switch ext {
	case ".png":
		w.Header().Set("Content-Type", "image/png")
		_ = captcha.WriteImage(&content, id, width, height)
	case ".wav":
		w.Header().Set("Content-Type", "audio/x-wav")
		_ = captcha.WriteAudio(&content, id, lang)
	default:
		return captcha.ErrNotFound
	}
   // 是否下载验证码
	if download {
		w.Header().Set("Content-Type", "application/octet-stream")
	}
	// 写入到页面中
	http.ServeContent(w, r, id+ext, time.Time{}, bytes.NewReader(content.Bytes()))
	return nil
}

// 验证
func CaptchaVerify(c *gin.Context, code string) bool {
	session := sessions.Default(c)
	if captchaId := session.Get("captcha"); captchaId != nil {
		session.Delete("captcha")
		_ = session.Save()
		// 验证 生成的验证码数值
		if captcha.VerifyString(captchaId.(string), code) {
			return true
		} else {
			return false
		}
	} else {
		return false
	}
}

• 验证码生成业务代码封装好了，接下来就是如何访问显示生成的验证码图片

// 首先要在对应的route文件中，配置一下路由，进行验证码的生成和显示
authGroup.GET("/captcha", func(ctx *gin.Context) {
	captcha.Captcha(ctx)
})

• 在对应的登录逻辑中，加入相应的逻辑：

captcha := params["captchaCode"].(string)
// 验证验证码
if !captcha2.CaptchaVerify(ctx, captcha) {
	utils.Fail(ctx, "验证码不正确", nil)
	return
}

• 接下来：封装一下session中间件

// 中间件，处理session
func Session(keyPairs string) gin.HandlerFunc {
	store := SessionConfig()
	return sessions.Sessions(keyPairs, store)
}
func SessionConfig() sessions.Store {
	sessionMaxAge := 3600
	sessionSecret := "topgoer"
	var store sessions.Store
	store = cookie.NewStore([]byte(sessionSecret))
	store.Options(sessions.Options{
		MaxAge: sessionMaxAge, //seconds
		Path:   "/",
	})
	return store
}

• 最后开启一下session

// 开启session
r.Use(middleware.Session("topgoer"))

走到这里，就全部实现了，验证码的生成，显示，验证，和使用等相关功能！

到了这一步，咱们就实现了上面需求的功能。 更多功能请持续关注！！！！！

知识星球

星球地址：t.zsxq.com/03MJM7YfI

关注公众号「程序员小乔」