This commit is contained in:
shaoyongjun 2024-09-30 22:49:21 +08:00
parent 821dda5395
commit adfc73df29
16 changed files with 306 additions and 62 deletions

View File

@ -21,16 +21,19 @@ func InitUserGroup(g *echo.Group) {
} }
ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db) ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db)
platform := utils.GetPlatform(&c)
req.Platform = platform
err := service.Login.Register(ctx, *req) err := service.UserBiz.Register(ctx, *req)
if err != nil { if err != nil {
return c.JSON(http.StatusOK, utils.Error(err)) return c.JSON(http.StatusOK, utils.Error(err))
} }
//登录 //登录
token, _ := service.Login.Login(ctx, dto.LoginReq{ token, _ := service.UserBiz.Login(ctx, dto.LoginReq{
Account: req.Account, Account: req.Account,
Password: req.Password, Password: req.Password,
Platform: platform,
}) })
return c.JSON(http.StatusOK, utils.Ok(token)) return c.JSON(http.StatusOK, utils.Ok(token))
@ -44,8 +47,9 @@ func InitUserGroup(g *echo.Group) {
} }
ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db) ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db)
//设置platform
token, err := service.Login.Login(ctx, *req) req.Platform = utils.GetPlatform(&c)
token, err := service.UserBiz.Login(ctx, *req)
if err != nil { if err != nil {
return c.JSON(http.StatusOK, utils.Error(err)) return c.JSON(http.StatusOK, utils.Error(err))
} }
@ -60,7 +64,7 @@ func InitUserGroup(g *echo.Group) {
return c.JSON(http.StatusOK, utils.Error(err)) return c.JSON(http.StatusOK, utils.Error(err))
} }
err := service.Login.SendResetPwdCode(context.Background(), req.Account) err := service.UserBiz.SendResetPwdCode(context.Background(), req.Email)
if err != nil { if err != nil {
return c.JSON(http.StatusOK, utils.Error(err)) return c.JSON(http.StatusOK, utils.Error(err))
} }
@ -77,7 +81,7 @@ func InitUserGroup(g *echo.Group) {
ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db) ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db)
err := service.Login.ResetPwd(ctx, *req) err := service.UserBiz.ResetPwd(ctx, *req)
if err != nil { if err != nil {
return c.JSON(http.StatusOK, utils.Error(err)) return c.JSON(http.StatusOK, utils.Error(err))
} }
@ -88,7 +92,10 @@ func InitUserGroup(g *echo.Group) {
//根据 token 获取 登录信息 //根据 token 获取 登录信息
g.POST("findLoginResult", func(c echo.Context) error { g.POST("findLoginResult", func(c echo.Context) error {
ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db) ctx := context.WithValue(context.Background(), constant.DBKey, repository.Db)
loginResult := service.Login.GetLoginResult(ctx, &c) loginResult, err := service.UserBiz.GetLoginResult(ctx, &c)
if err != nil {
return c.JSON(http.StatusOK, utils.Error(err))
}
return c.JSON(http.StatusOK, utils.Ok(loginResult)) return c.JSON(http.StatusOK, utils.Ok(loginResult))
}) })
} }

View File

@ -1,6 +1,7 @@
# 本机配置 # 本机配置
fbServer: myServer:
appName: mylomen_server appName: mylomen_server
jwtSigningKey: t2vr8fyqh5mvfvrjdszb6ev3eas4a9kw
# fbConsul # fbConsul
fbConsul: fbConsul:

View File

@ -1,6 +1,7 @@
# 本机配置 # 本机配置
fbServer: myServer:
appName: mylomen_server appName: mylomen_server
jwtSigningKey: t6p4n4g79mceqbe2b6syq4k5ahupy2h9
# fbConsul # fbConsul
fbConsul: fbConsul:

View File

@ -42,6 +42,11 @@ type Redis struct {
WriteTimeout int64 `mapstructure:"writeTimeout" json:"writeTimeout" yaml:"writeTimeout"` WriteTimeout int64 `mapstructure:"writeTimeout" json:"writeTimeout" yaml:"writeTimeout"`
} }
type MyServer struct {
AppName string `mapstructure:"appName" json:"appName" yaml:"appName"`
JwtSigningKey string `mapstructure:"jwtSigningKey" json:"jwtSigningKey" yaml:"jwtSigningKey"`
}
type Conf struct { type Conf struct {
FbConsul *FbConsul `mapstructure:"fbConsul" json:"fbConsul" yaml:"fbConsul"` // fbConsul 配置 FbConsul *FbConsul `mapstructure:"fbConsul" json:"fbConsul" yaml:"fbConsul"` // fbConsul 配置
@ -50,6 +55,8 @@ type Conf struct {
PgSql *PgSql `mapstructure:"pgSql" json:"pgSql" yaml:"pgSql"` // pgSql 配置 PgSql *PgSql `mapstructure:"pgSql" json:"pgSql" yaml:"pgSql"` // pgSql 配置
Redis *Redis `mapstructure:"redis" json:"redis" yaml:"redis"` // redis 配置 Redis *Redis `mapstructure:"redis" json:"redis" yaml:"redis"` // redis 配置
MyServer *MyServer `mapstructure:"myServer" json:"myServer" yaml:"myServer"` // myServer 配置
} }
var Instance = initCf() var Instance = initCf()

View File

@ -3,5 +3,7 @@ package constant
const ( const (
THIRD_LOGIN_TOKEN = ":login:token:" THIRD_LOGIN_TOKEN = ":login:token:"
THIRD_LOGIN_SN = ":login:sn:"
G_RESET_PWD_CODE = "g:reset:pwd:email:" G_RESET_PWD_CODE = "g:reset:pwd:email:"
) )

View File

@ -8,7 +8,7 @@ type RegisterReq struct {
} }
type SendResetPwdCodeReq struct { type SendResetPwdCodeReq struct {
Account string `json:"account"` // 帐号 Email string `json:"email"` // 帐号
} }
type ResetPwdReq struct { type ResetPwdReq struct {
@ -20,9 +20,12 @@ type ResetPwdReq struct {
} }
type LoginReq struct { type LoginReq struct {
Account *string `json:"account"` // 帐号 Account *string `json:"account"` // 帐号
Email *string `json:"email"` // 邮箱 Password string `json:"password"` // 密码
Phone *string `json:"phone"` // 手机号码 Platform string `json:"platform"` // 平台
Email *string `json:"email"` // 邮箱
Phone *string `json:"phone"` // 手机号码
WxId *string `json:"wxId"` // 微信unionId WxId *string `json:"wxId"` // 微信unionId
QqId *string `json:"qqId"` // qqId QqId *string `json:"qqId"` // qqId
@ -30,7 +33,6 @@ type LoginReq struct {
GoogleId *string `json:"googleId"` // googleId GoogleId *string `json:"googleId"` // googleId
FacebookId *string `json:"facebookId"` // facebookId FacebookId *string `json:"facebookId"` // facebookId
Password string `json:"password"` // 密码
} }
type UserLoginToken struct { type UserLoginToken struct {

View File

@ -1,10 +1,51 @@
package dto package dto
type UserVO struct { import (
Sn string `json:"sn"` "github.com/golang-jwt/jwt/v5"
Token string `json:"token"` "time"
Account *string `json:"Account"` )
type UserVO struct {
Sn string `json:"sn"`
Token string `json:"token"`
Platform string `json:"platform"`
Account *string `json:"Account"`
Nickname *string `json:"nickname"` Nickname *string `json:"nickname"`
Avatar *string `json:"avatar"` Avatar *string `json:"avatar"`
} }
type LoginTokenVo struct {
jwt.RegisteredClaims
Sn string `json:"sn"`
Token string `json:"token"`
Platform string `json:"platform"`
}
func (vo *LoginTokenVo) SetExtByPlatform(platform string) *LoginTokenVo {
switch platform {
case "web":
{
vo.Platform = "web"
vo.ExpiresAt = jwt.NewNumericDate(time.Now().Add(time.Hour * 24 * 7))
}
case "android":
{
vo.Platform = "android"
vo.ExpiresAt = jwt.NewNumericDate(time.Now().Add(time.Hour * 24 * 30))
}
case "ios":
{
vo.Platform = "ios"
vo.ExpiresAt = jwt.NewNumericDate(time.Now().Add(time.Hour * 24 * 30))
}
default:
{
vo.Platform = "web"
vo.ExpiresAt = jwt.NewNumericDate(time.Now().Add(time.Hour * 7 * 30))
}
}
return vo
}

View File

@ -12,3 +12,7 @@ func GetAccessToken(c *echo.Context) string {
return (*c).Request().Header.Get("udt") return (*c).Request().Header.Get("udt")
} }
func GetPlatform(c *echo.Context) string {
return (*c).Request().Header.Get("platform")
}

36
common/xjwt/jwt.go Normal file
View File

@ -0,0 +1,36 @@
package xjwt
import (
"github.com/golang-jwt/jwt/v5"
"mylomen_server/common/config"
"mylomen_server/common/dto"
"time"
)
func GenJwtToken(claims dto.LoginTokenVo) string {
mySigningKey := []byte(config.Instance.MyServer.JwtSigningKey)
claims.Issuer = "mylomen.com"
claims.IssuedAt = jwt.NewNumericDate(time.Now())
//加密
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
ss, _ := token.SignedString(mySigningKey)
return ss
}
func ParseJwtToken(jwtToken string) (*dto.LoginTokenVo, error) {
if jwtToken == "" || len(jwtToken) > 256 {
return nil, nil
}
//parse
parseToken, err := jwt.ParseWithClaims(jwtToken, &dto.LoginTokenVo{}, func(token *jwt.Token) (interface{}, error) {
return []byte(config.Instance.MyServer.JwtSigningKey), nil
})
//check
if userClaims, ok := parseToken.Claims.(*dto.LoginTokenVo); ok && parseToken.Valid {
return userClaims, nil
} else {
return nil, err
}
}

109
common/xjwt/jwt_test.go Normal file
View File

@ -0,0 +1,109 @@
package xjwt
import (
"fmt"
"github.com/golang-jwt/jwt/v5"
"mylomen_server/common/dto"
"testing"
"time"
)
func TestGenAndParse(t *testing.T) {
mySigningKey := []byte("AllYourBase")
claims := dto.LoginTokenVo{
Sn: "mySn",
Token: "myToken",
RegisteredClaims: jwt.RegisteredClaims{
Audience: []string{"mylomen.com"},
ExpiresAt: jwt.NewNumericDate(time.Now().Add(1 * time.Hour)),
IssuedAt: jwt.NewNumericDate(time.Now()),
Issuer: "mylomen.com",
NotBefore: jwt.NewNumericDate(time.Now()),
Subject: "1",
},
}
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
ss, err := token.SignedString(mySigningKey)
fmt.Println(ss, err)
parseToken, err := jwt.ParseWithClaims(ss, &dto.LoginTokenVo{}, func(token *jwt.Token) (interface{}, error) {
return mySigningKey, nil
})
if userClaims, ok := parseToken.Claims.(*dto.LoginTokenVo); ok && parseToken.Valid {
t.Log(userClaims, userClaims.RegisteredClaims.Issuer)
} else {
fmt.Println(err)
t.Error("验证失败")
}
}
func TestHs256(t *testing.T) {
type User struct {
Id int64
Name string
}
type UserClaims struct {
User User
jwt.RegisteredClaims
}
// 1 jwt.NewWithClaims生成token
user := User{
Id: 101,
Name: "hisheng",
}
userClaims := UserClaims{
User: user,
RegisteredClaims: jwt.RegisteredClaims{},
}
token := jwt.NewWithClaims(jwt.SigningMethodHS256, userClaims)
// 2 把token加密
mySigningKey := []byte("ushjlwmwnwht")
ss, err := token.SignedString(mySigningKey)
t.Log(ss, err)
}
func TestHs256Parse(t *testing.T) {
tokenString := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVc2VyIjp7IklkIjoxMDEsIk5hbWUiOiJoaXNoZW5nIn19.ij1kWID03f_CiELe0fPLZJ-Y64dkf2nDE-f6nGERBSE"
type User struct {
Id int64
Name string
}
type UserClaims struct {
User User
jwt.RegisteredClaims
}
token, err := jwt.ParseWithClaims(tokenString, &UserClaims{}, func(token *jwt.Token) (interface{}, error) {
return []byte("ushjlwmwnwht"), nil
})
if userClaims, ok := token.Claims.(*UserClaims); ok && token.Valid {
t.Log(userClaims, userClaims.RegisteredClaims.Issuer)
} else {
t.Log(err)
}
}
func TestGenJwtTokenThenParse(t *testing.T) {
claims := dto.LoginTokenVo{}
claims.Sn = "mySn"
claims.Token = "myToken"
claims.RegisteredClaims.ExpiresAt = jwt.NewNumericDate(time.Now().Add(24 * time.Hour))
jwtToken := GenJwtToken(claims)
//parse
data, err := ParseJwtToken(jwtToken)
if err != nil {
t.Error(err)
return
}
if data.Sn != claims.Sn {
t.Error("Sn 不一致")
}
}

1
go.mod
View File

@ -45,6 +45,7 @@ require (
github.com/go-viper/mapstructure/v2 v2.0.0 // indirect github.com/go-viper/mapstructure/v2 v2.0.0 // indirect
github.com/gogo/protobuf v1.3.2 // indirect github.com/gogo/protobuf v1.3.2 // indirect
github.com/golang-jwt/jwt v3.2.2+incompatible // indirect github.com/golang-jwt/jwt v3.2.2+incompatible // indirect
github.com/golang-jwt/jwt/v5 v5.2.1 // indirect
github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
github.com/golang/protobuf v1.5.4 // indirect github.com/golang/protobuf v1.5.4 // indirect
github.com/google/pprof v0.0.0-20240910150728-a0b0bb1d4134 // indirect github.com/google/pprof v0.0.0-20240910150728-a0b0bb1d4134 // indirect

2
go.sum
View File

@ -92,6 +92,8 @@ github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=
github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q= github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY= github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I= github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
github.com/golang-jwt/jwt/v5 v5.2.1 h1:OuVbFODueb089Lh128TAcimifWaLhJwVflnrgM17wHk=
github.com/golang-jwt/jwt/v5 v5.2.1/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
github.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc= github.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=
github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE= github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE=

View File

@ -8,7 +8,10 @@ import (
type gUserRepository interface { type gUserRepository interface {
FindByReq(ctx context.Context, req dto.LoginReq) *UserDO FindByReq(ctx context.Context, req dto.LoginReq) *UserDO
//FindById 根据 id 查询用户
FindById(ctx context.Context, id int64) *UserDO FindById(ctx context.Context, id int64) *UserDO
FindBySn(ctx context.Context, sn string) *UserDO
//FindByAccount 根据 account 查询用户 //FindByAccount 根据 account 查询用户
FindByAccount(ctx context.Context, account string) *UserDO FindByAccount(ctx context.Context, account string) *UserDO
FindByEmail(ctx context.Context, email string) *UserDO FindByEmail(ctx context.Context, email string) *UserDO
@ -25,4 +28,4 @@ type gUserRepository interface {
UpdateById(ctx context.Context, data *UserDO) error UpdateById(ctx context.Context, data *UserDO) error
} }
var GUser gUserRepository = new(gUserRepositoryImpl) var GUser gUserRepository = new(userRepositoryImpl)

View File

@ -43,10 +43,10 @@ func (m *UserDO) TableName() string {
return "user_base" return "user_base"
} }
type gUserRepositoryImpl struct { type userRepositoryImpl struct {
} }
func (rp *gUserRepositoryImpl) FindByReq(ctx context.Context, req dto.LoginReq) *UserDO { func (rp *userRepositoryImpl) FindByReq(ctx context.Context, req dto.LoginReq) *UserDO {
if req.Account != nil { if req.Account != nil {
return rp.FindByAccount(ctx, *req.Account) return rp.FindByAccount(ctx, *req.Account)
} }
@ -73,7 +73,7 @@ func (rp *gUserRepositoryImpl) FindByReq(ctx context.Context, req dto.LoginReq)
return nil return nil
} }
func (rp *gUserRepositoryImpl) FindById(ctx context.Context, id int64) *UserDO { func (rp *userRepositoryImpl) FindById(ctx context.Context, id int64) *UserDO {
//验证参数 //验证参数
if id <= 0 { if id <= 0 {
return nil return nil
@ -90,7 +90,24 @@ func (rp *gUserRepositoryImpl) FindById(ctx context.Context, id int64) *UserDO {
return &model return &model
} }
func (rp *gUserRepositoryImpl) FindByAccount(ctx context.Context, account string) *UserDO { func (rp *userRepositoryImpl) FindBySn(ctx context.Context, sn string) *UserDO {
//验证参数
if sn == "" || len(sn) > 16 {
return nil
}
db, ok := ctx.Value("db").(*gorm.DB)
if !ok {
db = Db
}
var model UserDO
sqlErr := db.Model(&model).Where("sn=?", sn).First(&model).Error
if sqlErr != nil {
return nil
}
return &model
}
func (rp *userRepositoryImpl) FindByAccount(ctx context.Context, account string) *UserDO {
//验证参数 //验证参数
if account == "" || len(account) > 32 { if account == "" || len(account) > 32 {
return nil return nil
@ -107,7 +124,7 @@ func (rp *gUserRepositoryImpl) FindByAccount(ctx context.Context, account string
return &model return &model
} }
func (rp *gUserRepositoryImpl) FindByEmail(ctx context.Context, email string) *UserDO { func (rp *userRepositoryImpl) FindByEmail(ctx context.Context, email string) *UserDO {
//验证参数 //验证参数
if email == "" || len(email) > 64 { if email == "" || len(email) > 64 {
return nil return nil
@ -124,7 +141,7 @@ func (rp *gUserRepositoryImpl) FindByEmail(ctx context.Context, email string) *U
return &model return &model
} }
func (rp *gUserRepositoryImpl) FindByPhone(ctx context.Context, phone string) *UserDO { func (rp *userRepositoryImpl) FindByPhone(ctx context.Context, phone string) *UserDO {
//验证参数 //验证参数
if phone == "" || len(phone) > 16 { if phone == "" || len(phone) > 16 {
return nil return nil
@ -141,7 +158,7 @@ func (rp *gUserRepositoryImpl) FindByPhone(ctx context.Context, phone string) *U
return &model return &model
} }
func (rp *gUserRepositoryImpl) FindByWxId(ctx context.Context, wxId string) *UserDO { func (rp *userRepositoryImpl) FindByWxId(ctx context.Context, wxId string) *UserDO {
//验证参数 //验证参数
if wxId == "" || len(wxId) > 16 { if wxId == "" || len(wxId) > 16 {
return nil return nil
@ -158,7 +175,7 @@ func (rp *gUserRepositoryImpl) FindByWxId(ctx context.Context, wxId string) *Use
return &model return &model
} }
func (rp *gUserRepositoryImpl) FindByQqId(ctx context.Context, qqId string) *UserDO { func (rp *userRepositoryImpl) FindByQqId(ctx context.Context, qqId string) *UserDO {
//验证参数 //验证参数
if qqId == "" || len(qqId) > 32 { if qqId == "" || len(qqId) > 32 {
return nil return nil
@ -175,7 +192,7 @@ func (rp *gUserRepositoryImpl) FindByQqId(ctx context.Context, qqId string) *Use
return &model return &model
} }
func (rp *gUserRepositoryImpl) FindByGoogleId(ctx context.Context, googleId string) *UserDO { func (rp *userRepositoryImpl) FindByGoogleId(ctx context.Context, googleId string) *UserDO {
//验证参数 //验证参数
if googleId == "" || len(googleId) > 32 { if googleId == "" || len(googleId) > 32 {
return nil return nil
@ -192,7 +209,7 @@ func (rp *gUserRepositoryImpl) FindByGoogleId(ctx context.Context, googleId stri
return &model return &model
} }
func (rp *gUserRepositoryImpl) FindByFacebookId(ctx context.Context, facebookId string) *UserDO { func (rp *userRepositoryImpl) FindByFacebookId(ctx context.Context, facebookId string) *UserDO {
//验证参数 //验证参数
if facebookId == "" || len(facebookId) > 32 { if facebookId == "" || len(facebookId) > 32 {
return nil return nil
@ -210,7 +227,7 @@ func (rp *gUserRepositoryImpl) FindByFacebookId(ctx context.Context, facebookId
} }
// Create 创建用户 // Create 创建用户
func (rp *gUserRepositoryImpl) Create(ctx context.Context, data *UserDO) error { func (rp *userRepositoryImpl) Create(ctx context.Context, data *UserDO) error {
db, ok := ctx.Value("db").(*gorm.DB) db, ok := ctx.Value("db").(*gorm.DB)
if !ok { if !ok {
db = Db db = Db
@ -225,7 +242,7 @@ func (rp *gUserRepositoryImpl) Create(ctx context.Context, data *UserDO) error {
} }
// UpdateById 根据ID更新用户 // UpdateById 根据ID更新用户
func (rp *gUserRepositoryImpl) UpdateById(ctx context.Context, data *UserDO) error { func (rp *userRepositoryImpl) UpdateById(ctx context.Context, data *UserDO) error {
db, ok := ctx.Value("db").(*gorm.DB) db, ok := ctx.Value("db").(*gorm.DB)
if !ok { if !ok {
db = Db db = Db

View File

@ -4,7 +4,6 @@ import (
"context" "context"
"crypto/sha256" "crypto/sha256"
"encoding/hex" "encoding/hex"
"encoding/json"
"errors" "errors"
"github.com/google/uuid" "github.com/google/uuid"
"github.com/labstack/echo/v4" "github.com/labstack/echo/v4"
@ -14,6 +13,7 @@ import (
"mylomen_server/common/email" "mylomen_server/common/email"
"mylomen_server/common/logs" "mylomen_server/common/logs"
"mylomen_server/common/utils" "mylomen_server/common/utils"
"mylomen_server/common/xjwt"
"mylomen_server/infrastructure/convert" "mylomen_server/infrastructure/convert"
"mylomen_server/infrastructure/redis" "mylomen_server/infrastructure/redis"
"mylomen_server/infrastructure/repository" "mylomen_server/infrastructure/repository"
@ -21,22 +21,27 @@ import (
"time" "time"
) )
type login struct { type user struct {
} }
var Login login var UserBiz user
// 令牌桶大小为 100, 以每秒 10 个 Token 的速率向桶中放置 Token // 令牌桶大小为 100, 以每秒 10 个 Token 的速率向桶中放置 Token
var limiter = rate.NewLimiter(10, 10) var limiter = rate.NewLimiter(10, 10)
// Register 注册 // Register 注册
func (l login) Register(ctx context.Context, req dto.RegisterReq) error { func (l user) Register(ctx context.Context, req dto.RegisterReq) error {
acUser := repository.GUser.FindByReq(ctx, req.LoginReq) acUser := repository.GUser.FindByReq(ctx, req.LoginReq)
if acUser != nil && acUser.Deleted == false { if acUser != nil && acUser.Deleted == false {
return errors.New("user exist") return errors.New("user exist")
} }
//fill data
userDO := convert.UserReq2DO(req) userDO := convert.UserReq2DO(req)
userDO.Deleted = false
userDO.UpdateTime = time.Now()
userDO.NickName = req.NickName
userDO.Avatar = req.Avatar
//密码加密 //密码加密
h := sha256.Sum256([]byte(req.Password)) h := sha256.Sum256([]byte(req.Password))
@ -46,7 +51,7 @@ func (l login) Register(ctx context.Context, req dto.RegisterReq) error {
return repository.GUser.Create(ctx, &userDO) return repository.GUser.Create(ctx, &userDO)
} }
func (l login) Login(ctx context.Context, req dto.LoginReq) (*dto.UserVO, error) { func (l user) Login(ctx context.Context, req dto.LoginReq) (*dto.UserVO, error) {
start := time.Now().UnixMilli() start := time.Now().UnixMilli()
//1. 查询用户 //1. 查询用户
@ -70,20 +75,22 @@ func (l login) Login(ctx context.Context, req dto.LoginReq) (*dto.UserVO, error)
//3. 组装数据 //3. 组装数据
result := convert.UserDO2VO(*acUser) result := convert.UserDO2VO(*acUser)
result.Platform = req.Platform
//生成token todo 使用jwt //生成token
token := uuid.New().String() token := strings.ReplaceAll(uuid.New().String(), "-", "")
token = strings.ReplaceAll(token, "-", "") claims := dto.LoginTokenVo{Sn: result.Sn, Token: token}
result.Token = token //设置超时时间
claims.SetExtByPlatform(req.Platform)
result.Token = xjwt.GenJwtToken(claims)
diff := time.Now().UnixMilli() - start diff := time.Now().UnixMilli() - start
logs.NewLog("").Infof("Login cost: %d", diff) logs.NewLog("").Infof("Login cost: %d", diff)
return &result, nil return &result, nil
} }
func (l login) SendResetPwdCode(ctx context.Context, account string) error { func (l user) SendResetPwdCode(ctx context.Context, emailP string) error {
//1. 验证账号 //1. 验证账号
acUser := repository.GUser.FindByAccount(ctx, account) acUser := repository.GUser.FindByEmail(ctx, emailP)
if acUser == nil || acUser.Deleted == true { if acUser == nil || acUser.Deleted == true {
return errors.New("user not exist") return errors.New("user not exist")
} }
@ -92,7 +99,7 @@ func (l login) SendResetPwdCode(ctx context.Context, account string) error {
code := utils.GetPseudoRandomCode(6) code := utils.GetPseudoRandomCode(6)
//3. save into redis //3. save into redis
if err := redis.Set(constant.G_RESET_PWD_CODE+account, code, time.Duration(30)*time.Minute); err != nil { if err := redis.Set(constant.G_RESET_PWD_CODE+emailP, code, time.Duration(30)*time.Minute); err != nil {
return errors.New("cache illegal") return errors.New("cache illegal")
} }
@ -102,7 +109,7 @@ func (l login) SendResetPwdCode(ctx context.Context, account string) error {
} }
//4. send code //4. send code
if err := email.SendEmailVerifyCodeByEmail(code, account); err != nil { if err := email.SendEmailVerifyCodeByEmail(code, emailP); err != nil {
return errors.New("send email code illegal") return errors.New("send email code illegal")
} }
@ -110,7 +117,7 @@ func (l login) SendResetPwdCode(ctx context.Context, account string) error {
} }
// ResetPwd 重置密码 // ResetPwd 重置密码
func (l login) ResetPwd(ctx context.Context, req dto.ResetPwdReq) error { func (l user) ResetPwd(ctx context.Context, req dto.ResetPwdReq) error {
//1. 验证code //1. 验证code
redisStr, err := redis.Get(constant.G_RESET_PWD_CODE + req.Account) redisStr, err := redis.Get(constant.G_RESET_PWD_CODE + req.Account)
if err != nil && redisStr != req.Code { if err != nil && redisStr != req.Code {
@ -127,29 +134,29 @@ func (l login) ResetPwd(ctx context.Context, req dto.ResetPwdReq) error {
h := sha256.Sum256([]byte(req.Password)) h := sha256.Sum256([]byte(req.Password))
passHash := hex.EncodeToString(h[:]) passHash := hex.EncodeToString(h[:])
acUser.Pwd = &passHash acUser.Pwd = &passHash
repository.GUser.UpdateById(ctx, acUser) if updateErr := repository.GUser.UpdateById(ctx, acUser); updateErr != nil {
logs.NewLog("").Errorf("update user password req:%+v error: %+v", req, updateErr)
//生成token return errors.New("system error")
token := uuid.New().String() }
token = strings.ReplaceAll(token, "-", "")
return nil return nil
} }
func (l login) GetLoginResult(ctx context.Context, c *echo.Context) *dto.UserVO { func (l user) GetLoginResult(ctx context.Context, c *echo.Context) (*dto.UserVO, error) {
accessToken := utils.GetAccessToken(c) accessToken := utils.GetAccessToken(c)
if accessToken == "" { loginVo, err := xjwt.ParseJwtToken(accessToken)
return nil if err != nil || loginVo.Sn == "" {
return nil, errors.New("token illegal")
} }
//redis //查询数据库
redisStr, err := redis.Get(constant.THIRD_LOGIN_TOKEN + accessToken) userDO := repository.GUser.FindBySn(ctx, loginVo.Sn)
if err != nil && redisStr != "" { if err != nil || userDO == nil {
var loginInfo dto.UserLoginToken return nil, errors.New("user not exist")
if redisErr := json.Unmarshal([]byte(redisStr), &loginInfo); redisErr == nil {
return nil
}
} }
return nil result := convert.UserDO2VO(*userDO)
result.Token = accessToken
result.Platform = loginVo.Platform
return utils.ToPtr(result), nil
} }

View File

@ -12,3 +12,7 @@ func Test1(t *testing.T) {
passHash := hex.EncodeToString(h[:]) passHash := hex.EncodeToString(h[:])
fmt.Print(passHash) fmt.Print(passHash)
} }
func TestLogin(t *testing.T) {
}