87 lines
1.5 KiB
Go
87 lines
1.5 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"github.com/loveuer/ushare/internal/model"
|
|
"github.com/loveuer/ushare/internal/opt"
|
|
"github.com/loveuer/ushare/internal/pkg/tool"
|
|
"github.com/pkg/errors"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type userManager struct {
|
|
sync.Mutex
|
|
ctx context.Context
|
|
um map[string]*model.User
|
|
}
|
|
|
|
func (um *userManager) Login(username string, password string) (*model.User, error) {
|
|
var (
|
|
now = time.Now()
|
|
)
|
|
|
|
if username != "admin" {
|
|
return nil, errors.New("账号或密码错误")
|
|
}
|
|
|
|
if !tool.ComparePassword(password, opt.Cfg.Auth) {
|
|
return nil, errors.New("账号或密码错误")
|
|
}
|
|
|
|
op := &model.User{
|
|
Id: 1,
|
|
Username: username,
|
|
LoginAt: now.Unix(),
|
|
Token: tool.RandomString(32),
|
|
}
|
|
|
|
um.Lock()
|
|
defer um.Unlock()
|
|
um.um[op.Token] = op
|
|
|
|
return op, nil
|
|
}
|
|
|
|
func (um *userManager) Verify(token string) (*model.User, error) {
|
|
um.Lock()
|
|
defer um.Unlock()
|
|
|
|
op, ok := um.um[token]
|
|
if !ok {
|
|
return nil, errors.New("未登录或凭证已失效, 请重新登录")
|
|
}
|
|
|
|
return op, nil
|
|
}
|
|
|
|
func (um *userManager) Start(ctx context.Context) {
|
|
um.ctx = ctx
|
|
|
|
go func() {
|
|
|
|
ticker := time.NewTicker(time.Minute)
|
|
|
|
for {
|
|
select {
|
|
case <-um.ctx.Done():
|
|
return
|
|
case now := <-ticker.C:
|
|
um.Lock()
|
|
for _, op := range um.um {
|
|
if now.Sub(time.UnixMilli(op.LoginAt)) > 8*time.Hour {
|
|
delete(um.um, op.Token)
|
|
}
|
|
}
|
|
um.Unlock()
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
var (
|
|
UserManager = &userManager{
|
|
um: make(map[string]*model.User),
|
|
}
|
|
)
|