1. gin验证码

在开发的过程中，我们有些接口为了防止被恶意调用，我们会采用加验证码的方式，例如：发送短信的接口，为了防止短信接口被频繁调用造成损失；注册的接口，为了防止恶意注册。在这里为大家推荐一个验证码的类库，方便大家学习使用。

  1.      github.com/dchest/captcha

web端是怎么实现验证码的功能呢？

  • 提供一个路由，先在session里写入键值对（k->v），把值写在图片上，然后生成图片，显示在浏览器上面
  • 前端将图片中的内容发送给后后端，后端根据session中的k取得v，比对校验。如果通过继续下一步的逻辑，失败给出错误提示

API接口验证码实现方式类似，可以把键值对存储在起来，验证的时候把键值对传输过来一起校验。这里我只给出了web端的方法，爱动手的小伙伴可以自己尝试一下。

后端

  1. package main
  3. import (
  4.     "bytes"
  5.     "github.com/dchest/captcha"
  6.     "github.com/gin-contrib/sessions"
  7.     "github.com/gin-contrib/sessions/cookie"
  8.     "github.com/gin-gonic/gin"
  9.     "net/http"
  10.     "time"
  11. )
  13. // 中间件，处理session
  14. func Session(keyPairs string) gin.HandlerFunc {
  15.     store := SessionConfig()
  16.     return sessions.Sessions(keyPairs, store)
  17. }
  18. func SessionConfig() sessions.Store {
  19.     sessionMaxAge := 3600
  20.     sessionSecret := "topgoer"
  21.     var store sessions.Store
  22.     store = cookie.NewStore([]byte(sessionSecret))
  23.     store.Options(sessions.Options{
  24.         MaxAge: sessionMaxAge, //seconds
  25.         Path:   "/",
  26.     })
  27.     return store
  28. }
  30. func Captcha(c *gin.Context, length ...int) {
  31.     l := captcha.DefaultLen
  32.     w, h := 107, 36
  33.     if len(length) == 1 {
  34.         l = length[0]
  35.     }
  36.     if len(length) == 2 {
  37.         w = length[1]
  38.     }
  39.     if len(length) == 3 {
  40.         h = length[2]
  41.     }
  42.     captchaId := captcha.NewLen(l)
  43.     session := sessions.Default(c)
  44.     session.Set("captcha", captchaId)
  45.     _ = session.Save()
  46.     _ = Serve(c.Writer, c.Request, captchaId, ".png", "zh", false, w, h)
  47. }
  48. func CaptchaVerify(c *gin.Context, code string) bool {
  49.     session := sessions.Default(c)
  50.     if captchaId := session.Get("captcha"); captchaId != nil {
  51.         session.Delete("captcha")
  52.         _ = session.Save()
  53.         if captcha.VerifyString(captchaId.(string), code) {
  54.             return true
  55.         } else {
  56.             return false
  57.         }
  58.     } else {
  59.         return false
  60.     }
  61. }
  62. func Serve(w http.ResponseWriter, r *http.Request, id, ext, lang string, download bool, width, height int) error {
  63.     w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
  64.     w.Header().Set("Pragma", "no-cache")
  65.     w.Header().Set("Expires", "0")
  67.     var content bytes.Buffer
  68.     switch ext {
  69.     case ".png":
  70.         w.Header().Set("Content-Type", "image/png")
  71.         _ = captcha.WriteImage(&content, id, width, height)
  72.     case ".wav":
  73.         w.Header().Set("Content-Type", "audio/x-wav")
  74.         _ = captcha.WriteAudio(&content, id, lang)
  75.     default:
  76.         return captcha.ErrNotFound
  77.     }
  79.     if download {
  80.         w.Header().Set("Content-Type", "application/octet-stream")
  81.     }
  82.     http.ServeContent(w, r, id+ext, time.Time{}, bytes.NewReader(content.Bytes()))
  83.     return nil
  84. }
  86. func main() {
  87.     router := gin.Default()
  88.     router.LoadHTMLGlob("./*.html")
  89.     router.Use(Session("topgoer"))
  90.     router.GET("/captcha", func(c *gin.Context) {
  91.         Captcha(c, 4)
  92.     })
  93.     router.GET("/", func(c *gin.Context) {
  94.         c.HTML(http.StatusOK, "index.html", nil)
  95.     })
  96.     router.GET("/captcha/verify/:value", func(c *gin.Context) {
  97.         value := c.Param("value")
  98.         if CaptchaVerify(c, value) {
  99.             c.JSON(http.StatusOK, gin.H{"status": 0, "msg": "success"})
  100.         } else {
  101.             c.JSON(http.StatusOK, gin.H{"status": 1, "msg": "failed"})
  102.         }
  103.     })
  104.     router.Run(":8080")
  105. }

前端页面

  1. <!DOCTYPE html>
  2. <html lang="en">
  3. <head>
  4.     <meta charset="UTF-8">
  5.     <title>www.topgoer.com验证码</title>
  6. </head>
  7. <body>
  8. <img src="/captcha" onclick="this.src='/captcha?v='+Math.random()">
  9. </body>
  10. </html>

浏览器访问http://127.0.0.1:8080

访问http://127.0.0.1:8080/captcha/verify/5721 进行验证

  1.     {
  2.         "msg": "failed",
  3.         "status": 1
  4.     }

转自：https://www.jianshu.com/p/5ee3c2a71453