diff --git a/backend/app/api/login.go b/backend/app/api/login.go index c412b5d..140fce1 100644 --- a/backend/app/api/login.go +++ b/backend/app/api/login.go @@ -24,7 +24,7 @@ func Sign(c *gin.Context) { } form.Username = strings.TrimSpace(form.Username) form.Code = strings.TrimSpace(form.Code) - + // 从数据库拿用户 s, err := public.NewSqlite("data/data.db", "") if err != nil { @@ -41,10 +41,10 @@ func Sign(c *gin.Context) { public.FailMsg(c, err.Error()) return } - + session := sessions.Default(c) now := time.Now() - + loginErrCount := session.Get("__loginErrCount") loginErrEnd := session.Get("__loginErrEnd") ErrCount := 0 @@ -57,9 +57,9 @@ func Sign(c *gin.Context) { if __loginErrEnd, ok := loginErrEnd.(time.Time); ok { ErrEnd = __loginErrEnd } - + // fmt.Println(ErrCount, ErrEnd) - + // 判断登录错误次数 switch { case ErrCount >= 5: @@ -91,7 +91,7 @@ func Sign(c *gin.Context) { } } } - + // 判断用户是否存在 if len(res) == 0 { session.Set("__loginErrCount", ErrCount+1) @@ -117,7 +117,7 @@ func Sign(c *gin.Context) { keyMd5 := md5.Sum([]byte(passwd)) passwdMd5 := hex.EncodeToString(keyMd5[:]) // fmt.Println(passwdMd5) - + if res[0]["password"] != passwdMd5 { session.Set("__loginErrCount", ErrCount+1) session.Set("__loginErrEnd", now) @@ -128,7 +128,7 @@ func Sign(c *gin.Context) { public.FailMsg(c, "密码错误") return } - + // session := sessions.Default(c) session.Set("__loginErrCount", 0) session.Delete("__loginErrEnd") @@ -137,7 +137,7 @@ func Sign(c *gin.Context) { _ = session.Save() // c.JSON(http.StatusOK, public.ResOK(0, nil, "登录成功")) // 设置cookie - c.SetCookie("must_code", "1", 0, "/", "", false, false) + c.SetCookie("must_code", "1", -1, "/", "", false, false) public.SuccessMsg(c, "登录成功") return } @@ -145,7 +145,7 @@ func Sign(c *gin.Context) { func GetCode(c *gin.Context) { _, bs64, code, _ := public.GenerateCode() session := sessions.Default(c) - + session.Set("_verifyCode", code) _ = session.Save() public.SuccessData(c, bs64, 0)