Golang Http 驗證碼示例實現

更新時間：2020年08月21日 08:22:34 作者：xialeistudio

這篇文章主要介紹了Golang Http 驗證碼示例，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

驗證碼（CAPTCHA）是“Completely Automated Public Turing test to tell Computers and Humans Apart”（全自動區(qū)分計算機和人類的圖靈測試）的縮寫，是一種區(qū)分用戶是計算機還是人的公共全自動程序。可以防止：惡意破解密碼、刷票、論壇灌水，有效防止某個黑客對某一個特定注冊用戶用特定程序暴力破解方式進行不斷的登陸嘗試，實際上用驗證碼是現在很多網站通行的方式，我們利用比較簡易的方式實現了這個功能。這個問題可以由計算機生成并評判，但是必須只有人類才能解答。由于計算機無法解答CAPTCHA的問題，所以回答出問題的用戶就可以被認為是人類。

傳統網站驗證碼工作機制

客戶端請求服務器獲取驗證碼圖片
服務器生成隨機串(驗證碼值)寫入Session，并將驗證碼值寫入到圖片中返回給客戶端
客戶端輸入圖片上的字符串提交給服務器驗證
服務器比對客戶端提交的字符串值和 Session 中是否匹配，如果匹配則通過驗證

由于服務器生成的驗證碼值從始至終均未返回給客戶端，因此，客戶端只能從圖片中識別驗證碼字符串，從而保證人機校驗邏輯。

Go的HTTP驗證碼

思路

Go 語言的 HTTP 服務器默認不支持 Session，因此驗證碼值需要換個思路存儲，以下是不使用 Session 的邏輯

客戶端請求服務器獲取驗證碼ID
服務器生成驗證碼 ID，并生成驗證碼值，將 ID 和值的映射關系記錄到內存或緩存，并將 ID 返回給客戶端
客戶端根據返回的 ID 請求服務器獲取驗證碼圖片
服務器獲取到驗證碼 ID，從內存或緩存中取出驗證碼值，將該值寫入圖片并將圖片返回給客戶端
客戶端提交驗證碼 ID（第1步獲得）和驗證碼值給服務器驗證
服務器獲取驗證碼 ID，從內存或緩存中取出驗證碼值與客戶端提交的驗證碼值比對

示例

安裝驗證碼依賴

github.com/dchest/captcha

代碼實現

package main
  
  import (
    "fmt"
    "github.com/dchest/captcha"
    "log"
    "net/http"
  )
  
  func main() {
    // 獲取驗證碼 ID
    http.HandleFunc("/captcha/generate", func(w http.ResponseWriter, r *http.Request) {
      id := captcha.NewLen(6)
      if _, err := fmt.Fprint(w, id); err != nil {
        log.Println("generate captcha error", err)
      }
    })
    // 獲取驗證碼圖片
    http.HandleFunc("/captcha/image", func(w http.ResponseWriter, r *http.Request) {
      id := r.URL.Query().Get("id")
      if id == "" {
        http.Error(w, "Bad Request", http.StatusBadRequest)
        return
      }
      w.Header().Set("Content-Type", "image/png")
      if err := captcha.WriteImage(w, id, 120, 80); err != nil {
        log.Println("show captcha error", err)
      }
    })
    // 業(yè)務處理
    http.HandleFunc("/login", func(w http.ResponseWriter, r *http.Request) {
      if err := r.ParseForm(); err != nil {
        log.Println("parseForm error", err)
        http.Error(w, "Internal Error", http.StatusInternalServerError)
        return
      }
      // 獲取驗證碼 ID 和驗證碼值
      id := r.FormValue("id")
      value := r.FormValue("value")
      // 比對提交的驗證碼值和內存中的驗證碼值
      if captcha.VerifyString(id, value) {
        fmt.Fprint(w, "ok")
      } else {
        fmt.Fprint(w, "mismatch")
      }
    })
    log.Fatal(http.ListenAndServe(":8080", nil))
  }

運行