软件技术-零基础-Golang注册验证与不记得密码

欢迎关注我的专栏( つ??ω??)つ【人工智能通识】

如何检查客户的验证码？如何编写不记得密码功能？

改进register.go

我们先在开头的数据结构中添加一个Code验证码字段：

type registerReqDS struct {    Email string    Pw    string    Code  string}

而后在下面的Register方法里面添加验证码格式的检查：

    codeRe, _ := regexp.Compile(`^[0-9]{6}$`)    fmt.Println(ds.Code, reflect.TypeOf(ds.Code))    if !codeRe.MatchString(ds.Code) {        util.WWrite(w, 1, "验证码格式错误", nil)        return    }

后面检查客户邮箱能否存在之前，我们添加检查验证码Code和数据库regVerify中数据能否匹配的检查：

    //检查验证码能否正确    dbcv := tool.MongoDBCLient.Database("myweb").Collection("regVerify")    var v bson.M    dbcv.FindOne(context.TODO(), bson.M{"Email": ds.Email}).Decode(&v)    if v["Code"] == nil || v["Code"] != ds.Code {        util.WWrite(w, 1, "验证码错误。", nil)        return    }

最后面我们可以顺带把客户注册时间戳也一起存入数据库：

    //写入数据库    u := bson.M{"Email": ds.Email, "Pw": ds.Pw, "Ts": time.Now().Unix()}

而后保存并运行，利用邮箱中收到的验证码进行注册。

整个regsiter.go的完整代码如下，可以对照梳理一下思路：

package apiimport (    "app/tool"    "app/util"    "context"    "encoding/json"    "fmt"    "net/http"    "reflect"    "regexp"    "time"    "go.mongodb.org/mongo-driver/bson"    "go.mongodb.org/mongo-driver/bson/primitive")type registerReqDS struct {    Email string    Pw    string    Code  string}//Register 注册接口解决函数func Register(w http.ResponseWriter, r *http.Request) {    ds := registerReqDS{}    json.NewDecoder(r.Body).Decode(&ds)    //格式检查    pwRe, _ := regexp.Compile(`^[0-9a-zA-Z_@]{6,18}$`)    if !pwRe.MatchString(ds.Pw) {        util.WWrite(w, 1, "密码格式错误。", nil)        return    }    codeRe, _ := regexp.Compile(`^[0-9]{6}$`)    fmt.Println(ds.Code, reflect.TypeOf(ds.Code))    if !codeRe.MatchString(ds.Code) {        util.WWrite(w, 1, "验证码格式错误", nil)        return    }    mailRe, _ := regexp.Compile(`^(([^<>()[\]\\.,;:\s@\"]+(\.[^<>()[\]\\.,;:\s@\"]+)*)|(\".+\"))@((\[[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\])|(([a-zA-Z\-0-9]+\.)+[a-zA-Z]{2,}))$`)    if !mailRe.MatchString(ds.Email) {        util.WWrite(w, 1, "邮箱格式错误。", nil)        return    }    //检查验证码能否正确    dbcv := tool.MongoDBCLient.Database("myweb").Collection("regVerify")    var v bson.M    dbcv.FindOne(context.TODO(), bson.M{"Email": ds.Email}).Decode(&v)    if v["Code"] == nil || v["Code"] != ds.Code {        util.WWrite(w, 1, "验证码错误。", nil)        return    }    //访问数据集    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)    dbc := tool.MongoDBCLient.Database("myweb").Collection("user")    defer cancel()    //验证客户邮箱能否存在    count, err := dbc.CountDocuments(context.TODO(), bson.M{"Email": ds.Email})    if err != nil {        util.WWrite(w, 1, "读取数据库失败。", nil)        return    }    if count > 0 {        util.WWrite(w, 1, "邮箱已存在。", nil)        return    }    //写入数据库    u := bson.M{"Email": ds.Email, "Pw": ds.Pw, "Ts": time.Now().Unix()}    res, err := dbc.InsertOne(ctx, u)    if err != nil {        util.WWrite(w, 1, "写入数据库失败。", nil)        return    }    d := res.InsertedID.(primitive.ObjectID).Hex()    util.WWrite(w, 0, "注册成功。", d)    return}

不记得密码页面

假如客户不记得了密码怎样办？
我们实际上可以参照注册流程来思考。
复制regsiter.html文件成为page/fogotPw.html，修改它看起来这样：

客户输入邮箱，点击发送验证码，而后只需能输入邮箱里面收到的正确验证码即可以重置密码了。

整个页面代码的变化主要有：

• 去掉了客户协议选项，在checkBtn()函数中也去掉了agree检测。
• 原来的sendRegPost方法改为sendRstPwPost方法，'/api/Register'也改为'/api/ResetPw'
• sendVerify方法中'/api/SendRegVerifyMail'改为'/api/SendRstPwMail'.

添加sendRstPwMail.go

复制sendRegVerifyMail.go的内容成为api/sendRstPwMail.go文件，而后修改：

• 修改数据结构为type sendRstPwReqDS struct
• 修改函数名为func SendRstPwReqDS
• 修改ds := sendRstPwReqDS{}
• 修改数据集为Database("myweb").Collection("rstPwVerify")
• 修改发送邮件内容SendMail(ds.Email, "您在www.myweb.com的重置码是:"+s, "来自Myweb的重置验证码")
• 在邮箱格式检查之后，我们增加邮箱能否注册的检查：

    //检查邮箱能否已经注册，没注册不发送验证码    dbcu := tool.MongoDBCLient.Database("myweb").Collection("user")    count, _ := dbcu.CountDocuments(context.TODO(), bson.M{"Email": ds.Email})    if count == 0 {        util.WWrite(w, 1, "这个邮箱还没有注册。", nil)        return    }

而后再app.go中增加服务路径：
http.HandleFunc("/api/SendRstPwMail", api.SendRstPwMail)

而后运行代码，尝试访问页面http://localhost:8088/page/resetPw.html输入已经注册的邮箱，点击发送验证码按钮，成功后检查邮箱中的验证码。

添加resetPw.go

复制register.go的内容成为api/resetPw.go文件，而后修改：

• 修改数据格式type resetPwReqDS construt和ds := resetPwReqDS{}
• 修改函数名func ResetPw
• 将原有验证码检查部分修改为.Database("myweb").Collection("rstPwVerify")
• 将后续部分修改为以下新内容：

    //升级数据库中的密码    dbc := tool.MongoDBCLient.Database("myweb").Collection("user")    filter := bson.M{"Email": ds.Email}    up := bson.M{"$set": bson.M{"Pw": ds.Pw}}    _, err := dbc.UpdateOne(context.TODO(), filter, up)    if err != nil {        util.WWrite(w, 1, "写入数据库失败。", nil)        return    }    //返回修改的账号    var nu bson.M    dbc.FindOne(context.TODO(), bson.M{"Email": ds.Email}).Decode(&nu)    util.WWrite(w, 0, "修改成功。", nu["_id"])    return

这段代码注意几个内容：

• 我们使用了UpdateOne方法升级已有数据，
• 注意这里带有$set的格式表示要修改的内容bson.M{"$set": bson.M{"Pw": ds.Pw}}
• 重新使用了FindOne最后获取了客户的_id属性返回给客户，这在后续会使用到。

尽管我们能够利用login.html页面登录，但是切换新页面之后我们怎样知道客户依然处于登录状态呢？不能让客户每个页面都登录，在下一篇我们详情如何利用cookie来实现客户的自动登录。

欢迎关注我的专栏( つ??ω??)つ【人工智能通识】

每个人的智能新时代

假如您发现文章错误，请不吝留言指正；
假如您觉得有用，请点喜欢；
假如您觉得很有用，欢迎转载~

END