>備註：前面專案中用到的程式碼已經分享到`GitHub`中去了，並且以後所有專案中會出現的程式碼都會提交上去，歡迎查閱。

> 地址 [`goshop`](https://gitee.com/jobhandsome/goshop/) 感興趣的可以點個`star`哦~  [`goshop開源專案的更新` ](https://gitee.com/jobhandsome/goshop/)

#### 今天我們們做一個`登入功能中的驗證碼功能`：
1. **封裝驗證碼生成，驗證碼驗證等功能，以便多複用**
2. **封裝session的中介軟體**
#### 接下來我們們就來實現以上的問題和功能：

**先封裝驗證碼業務部分**

```go
// 生成驗證碼
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
	}
}
```

**驗證碼生成業務程式碼封裝好了，接下來就是如何訪問顯示生成的驗證碼圖片**
```go
// 首先要在對應的route檔案中，配置一下路由，進行驗證碼的生成和顯示
authGroup.GET("/captcha", func(ctx *gin.Context) {
	captcha.Captcha(ctx)
})
```

**在對應的登入邏輯中，加入相應的邏輯：**
```go
captcha := params["captchaCode"].(string)
// 驗證驗證碼
if !captcha2.CaptchaVerify(ctx, captcha) {
	utils.Fail(ctx, "驗證碼不正確", nil)
	return
}
```

**接下來：封裝一下session中介軟體**
```go
// 中介軟體，處理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
}
```
**最後在router檔案中，開啟一下session**
```go
// 開啟session
r.Use(middleware.Session("topgoer"))
```

**說到這裡，就全部實現了，驗證碼的生成，顯示，驗證，和使用等相關功能！**
### 更多功能請持續關注！！！！！ 
### 歡迎各位加我的微信（`程式設計師小喬`）跟我一起完成並推動專案的發展
![在這裡插入圖片描述](https://img-blog.csdnimg.cn/1787826e21c54f0c8bdf87ff91c0de2a.jpeg#pic_center)

Go專案實戰之驗證碼的生成和使用（易擴充套件方式）【goshop開源專案 | 20220516更新】

相關文章