golang验证码登录

在互联网时代，验证码（CAPTCHA）已经成为了网站和APP登录的常用方式之一。它不仅能够有效防止恶意程序的自动登录、注入等攻击行为，还可以提升用户数据的安全性。而在Golang开发中，实现验证码登录也变得异常简单。

生成随机验证码

要实现验证码登录，首先需要生成一个随机的验证码。在Golang中，通过使用`math/rand`包和`time`包提供的方法，我们可以很方便地生成随机的验证码。

func generateCode(length int) string {
    rand.Seed(time.Now().UnixNano())
    chars := "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
    code := make([]byte, length)
    
    for i := 0; i < length; i++ {
        code[i] = chars[rand.Intn(len(chars))]
    }
    
    return string(code)
}

以上代码通过`rand.Intn`方法在指定的字符集`chars`中随机选择一个字符，并通过循环生成指定长度的验证码。最后将字符数组转换为字符串并返回该字符串作为验证码。

存储验证码

生成验证码之后，我们需要将其存储到某个地方，以便后续的验证过程中进行对比。一种常用的方式是将验证码存储在内存中的缓存中，例如使用`sync.Map`来存储验证码和对应的标识符。

type CodeStore struct {
    store *sync.Map
}

func (cs *CodeStore) Save(code, identifier string) {
    cs.store.Store(identifier, code)
}

func (cs *CodeStore) Verify(code, identifier string) bool {
    savedCode, ok := cs.store.Load(identifier)
    if !ok {
        return false
    }
    
    return code == savedCode.(string)
}

以上代码定义了一个`CodeStore`结构体，其中的`Save`方法用于将验证码和对应的标识符存储到`sync.Map`中，`Verify`方法用于验证用户输入的验证码是否正确。

验证码登录接口

有了验证码生成和存储的基础，我们就可以实现验证码登录的接口了。在Golang中，可以使用标准库提供的`net/http`包和`http.Handler`接口来实现一个登录的HTTP接口。

type LoginHandler struct {
    codeStore *CodeStore
}

func (lh *LoginHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    if r.Method != http.MethodPost {
        w.WriteHeader(http.StatusMethodNotAllowed)
        return
    }
    
    identifier := r.FormValue("identifier")
    code := r.FormValue("code")
    
    if lh.codeStore.Verify(code, identifier) {
        w.WriteHeader(http.StatusOK)
        w.Write([]byte("Verification passed"))
        // 执行登录操作
    } else {
        w.WriteHeader(http.StatusUnauthorized)
        w.Write([]byte("Verification failed"))
    }
}

以上代码定义了一个`LoginHandler`结构体，实现了`ServeHTTP`方法来处理登录请求。通过解析HTTP请求中的表单值，获取到用户输入的标识符和验证码，并通过`CodeStore`对象进行验证，如果验证通过则返回状态码200和"Verification passed"消息，否则返回401和"Verification failed"消息。

通过上述代码，我们可以实现一个简单而又安全的验证码登录功能。当然，在实际应用中，我们还可以对验证码进行定期清理、限制尝试次数、加入图形化界面等更多细节的优化。