快捷導(dǎo)航

Go語言簡易圖像驗證碼生成的實戰(zhàn)案例

更新時間：2025年08月14日 09:57:50 作者：程序員愛釣魚

本篇我們將使用Go語言結(jié)合第三方庫 github.com/mojocn/base64Captcha,快速實現(xiàn)一個簡易圖像驗證碼生成接口,具有一定的參考價值,感興趣的可以了解一下

在 Web 應(yīng)用中，驗證碼（CAPTCHA）常用于防止機器人批量提交請求，比如注冊、登錄、評論等功能。本篇我們將使用 Go 語言和 Gin 框架，結(jié)合第三方庫 github.com/mojocn/base64Captcha，快速實現(xiàn)一個簡易圖像驗證碼生成接口。

一、功能目標(biāo)

提供一個生成驗證碼的 API，返回驗證碼圖片（Base64 編碼）和驗證碼 ID。
前端展示驗證碼圖片，并在提交時攜帶驗證碼 ID 和用戶輸入。
提供一個校驗驗證碼的 API。

二、安裝依賴

首先安裝 Gin 和 Base64Captcha：

go get github.com/gin-gonic/gin
go get github.com/mojocn/base64Captcha

三、代碼實現(xiàn)

package main

import (
	"github.com/gin-gonic/gin"
	"github.com/mojocn/base64Captcha"
	"net/http"
)

// 驗證碼存儲在內(nèi)存中（也可以換成 Redis）
var store = base64Captcha.DefaultMemStore

// 生成驗證碼
func generateCaptcha(c *gin.Context) {
	driver := base64Captcha.NewDriverDigit(80, 240, 5, 0.7, 80) // 高度80, 寬度240, 5位數(shù)字
	captcha := base64Captcha.NewCaptcha(driver, store)
	id, b64s, err := captcha.Generate()
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"error": "驗證碼生成失敗"})
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"captcha_id":   id,
		"captcha_image": b64s, // Base64 編碼的圖片
	})
}

// 校驗驗證碼
func verifyCaptcha(c *gin.Context) {
	var req struct {
		ID    string `json:"id"`
		Value string `json:"value"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}

	if store.Verify(req.ID, req.Value, true) { // true 表示驗證成功后清除
		c.JSON(http.StatusOK, gin.H{"message": "驗證成功"})
	} else {
		c.JSON(http.StatusBadRequest, gin.H{"message": "驗證碼錯誤"})
	}
}

func main() {
	r := gin.Default()
	r.GET("/captcha", generateCaptcha)
	r.POST("/verify", verifyCaptcha)
	r.Run(":8080")
}

四、運行與測試

運行服務(wù)：

go run main.go

1. 獲取驗證碼

curl http://localhost:8080/captcha

返回：

{
  "captcha_id": "ZffX7Xr7EccGdS4b",
  "captcha_image": "data:image/png;base64,iVBORw0KGgoAAAANSUhE..."
}

前端可直接用 <img src="captcha_image" /> 渲染驗證碼。

2. 校驗驗證碼

curl -X POST http://localhost:8080/verify \
  -H "Content-Type: application/json" \
  -d '{"id":"ZffX7Xr7EccGdS4b","value":"12345"}'

五、注意事項

驗證碼存儲
- 本示例使用內(nèi)存存儲，適合單機開發(fā)環(huán)境。
- 生產(chǎn)環(huán)境建議使用 Redis 等共享存儲。
驗證碼類型 base64Captcha 支持?jǐn)?shù)字、字母混合、中文等類型，可以根據(jù)業(yè)務(wù)需求選擇不同 Driver。
安全性
- 不能把驗證碼 ID 暴露給爬蟲（可配合 CSRF、限流等手段）。
- 驗證碼要有有效期，防止重放攻擊。