diff --git a/controllers/base.go b/controllers/base.go index 4f534d4..0383c1b 100644 --- a/controllers/base.go +++ b/controllers/base.go @@ -114,6 +114,7 @@ func (c *BaseController) Logined() *BaseController { } else { logs.Warn("登录成功") c.PtPin = v.(string) + logs.Info(models.Config.Master) if strings.EqualFold(models.Config.Master, v.(string)) { c.Master = true } diff --git a/controllers/login.go b/controllers/login.go index 046a019..b674689 100644 --- a/controllers/login.go +++ b/controllers/login.go @@ -330,6 +330,7 @@ func (c *LoginController) IsAdmin() { c.Ctx.Redirect(302, "/") c.StopRun() } else { + c.SetSession("pin", pin) c.Ctx.WriteString("登录") } diff --git a/models/db.go b/models/db.go index b8c10de..17d1bcd 100644 --- a/models/db.go +++ b/models/db.go @@ -42,6 +42,7 @@ func initDB() { &UserAgent{}, &Env{}, &Wish{}, + &Token{}, ) keys = make(map[string]bool) pins = make(map[string]bool) @@ -273,3 +274,34 @@ func CheckIn(pin, key string) int { } return 2 } + +func setToken(token *Token) error { + tx := db.Begin() + if err := tx.Create(token).Error; err != nil { + tx.Rollback() + return err + } + return tx.Commit().Error +} + +//func getToken(pin string) (*Token, error) { +// token := &Token{} +// db.Where(CreateAt+" = ?", pin).First(token) +// format := "2006-01-02 15:04:05" +// sqlUpdatedAt, _ := time.ParseInLocation(format, "2021-03-24 15:00:00", time.Local) +// fmt.Println("测试时间:", sqlUpdatedAt) +// +// t := time.Now() +// t_zero := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, t.Location()) +// fmt.Println("当天凌晨时间:", t_zero) +// +// t_ := sqlUpdatedAt.Sub(t_zero) +// fmt.Println("测试时间到当天凌晨时间:", t_) +// if t_ > 0 { +// fmt.Println("未超有效时间!!!") +// } else { +// fmt.Println("超过有效时间!!!") +// } +// +// return token, db.Where(CreateAt+" = ?", pin).First(ck).Error +//} diff --git a/models/toekn.go b/models/toekn.go new file mode 100644 index 0000000..597d8b8 --- /dev/null +++ b/models/toekn.go @@ -0,0 +1,11 @@ +package models + +import ( + "time" +) + +type Token struct { + ID int + CreatedAt time.Time + Token string +}