From da57f021912f800aa504fdede2cc4ab8238a8348 Mon Sep 17 00:00:00 2001 From: liuhaotian Date: Tue, 25 Jan 2022 17:03:14 +0800 Subject: [PATCH] feat : login --- .../api/murder/internal/handler/scripts_h.go | 9 ++- .../api/murder/internal/handler/user_h.go | 50 ++++++++++++ .../api/murder/internal/model/business.go | 11 +++ app/brahma/api/murder/internal/model/dm.go | 13 +++ .../api/murder/internal/model/journey.go | 4 + app/brahma/api/murder/internal/model/token.go | 30 +++++++ app/brahma/api/murder/internal/model/user.go | 42 ++++++++++ .../api/murder/internal/service/token.go | 26 ++++++ .../api/murder/internal/service/user.go | 81 +++++++++++++++++++ app/brahma/api/murder/main.go | 5 ++ .../token/internal/server/bussiness.go | 1 + .../service/token/internal/server/dm.go | 8 ++ 12 files changed, 277 insertions(+), 3 deletions(-) create mode 100644 app/brahma/api/murder/internal/handler/user_h.go create mode 100644 app/brahma/api/murder/internal/model/business.go create mode 100644 app/brahma/api/murder/internal/model/dm.go create mode 100644 app/brahma/api/murder/internal/model/journey.go create mode 100644 app/brahma/api/murder/internal/model/token.go create mode 100644 app/brahma/api/murder/internal/model/user.go create mode 100644 app/brahma/api/murder/internal/service/token.go create mode 100644 app/brahma/api/murder/internal/service/user.go create mode 100644 app/brahma/service/token/internal/server/bussiness.go create mode 100644 app/brahma/service/token/internal/server/dm.go diff --git a/app/brahma/api/murder/internal/handler/scripts_h.go b/app/brahma/api/murder/internal/handler/scripts_h.go index 04aa75f..95f70e3 100644 --- a/app/brahma/api/murder/internal/handler/scripts_h.go +++ b/app/brahma/api/murder/internal/handler/scripts_h.go @@ -9,7 +9,7 @@ import ( "git.icechen.cn/monorepo/backend/pkg/api" "git.icechen.cn/monorepo/backend/pkg/proto/brahma/murders" "git.icechen.cn/monorepo/backend/pkg/rpc" - "github.com/go-playground/validator/v10" + validator "github.com/go-playground/validator/v10" "github.com/gofiber/fiber/v2" ctxLogger "github.com/luizsuper/ctxLoggers" "github.com/pkg/errors" @@ -21,8 +21,13 @@ import ( var ( pageIsNotInt = api.WarpFError(errors.New("page 参数错误")) limitIsNotInt = api.WarpFError(errors.New("limit 参数错误")) + validate *validator.Validate ) +func init() { + validate = validator.New() +} + func GetScriptsH(ctx *fiber.Ctx) error { page := ctx.Query("page", "1") limit := ctx.Query("size", "10") @@ -110,8 +115,6 @@ func GetScriptApi(ctx *fiber.Ctx) error { return pageIsNotInt } - validate := validator.New() - err = validate.Var(p, "gte=0") if err != nil { ctxLogger.Error(ctx.UserContext(), limitIsNotInt.Error(), zap.String("limit", limit)) diff --git a/app/brahma/api/murder/internal/handler/user_h.go b/app/brahma/api/murder/internal/handler/user_h.go new file mode 100644 index 0000000..7bdef7d --- /dev/null +++ b/app/brahma/api/murder/internal/handler/user_h.go @@ -0,0 +1,50 @@ +package handler + +import ( + "git.icechen.cn/monorepo/backend/app/brahma/api/murder/internal/service" + "git.icechen.cn/monorepo/backend/pkg/api" + "github.com/gofiber/fiber/v2" + ctxLogger "github.com/luizsuper/ctxLoggers" + "github.com/pkg/errors" + "go.uber.org/zap" + "strconv" +) + +var ( + idIsNotInt = api.WarpFError(errors.New("类型不是int")) + illegalInt = api.WarpFError(errors.New("非法类型")) + codeIsNull = api.WarpFError(errors.New("传入code为空")) +) + +func Login(ctx *fiber.Ctx) error { + code := ctx.Query("code", "") + + if code == "" { + ctxLogger.Error(ctx, codeIsNull.Error()) + return codeIsNull + } + + appType := ctx.Query("type", "") + + l, err := strconv.Atoi(appType) + + if err != nil { + ctxLogger.Error(ctx, idIsNotInt.Error(), zap.String("appType is ", appType)) + return idIsNotInt + } + //目前暂定的appId + err = validate.Var(l, "oneof=1 2 3") + if err != nil { + ctxLogger.Error(ctx, illegalInt.Error(), zap.String("appType is ", appType)) + return illegalInt + } + + err = service.Login(ctx.UserContext(), code, appType) + + if err != nil { + ctxLogger.Error(ctx, "login err", zap.String("reason", err.Error())) + return err + } + + return ctx.JSON(success(nil)) +} diff --git a/app/brahma/api/murder/internal/model/business.go b/app/brahma/api/murder/internal/model/business.go new file mode 100644 index 0000000..fa30a0c --- /dev/null +++ b/app/brahma/api/murder/internal/model/business.go @@ -0,0 +1,11 @@ +package model + +type Business struct { + address string + point float64 + tags string + pictures []string + Uuid string + Phone string + Scripts *[]Scripts `json:"tags,omitempty" gorm:"many2many:business_scripts;foreignKey:Uuid;joinForeignKey:CategoryUid;References:Uuid;JoinReferences:TagUid" valid:"no_empty"` +} diff --git a/app/brahma/api/murder/internal/model/dm.go b/app/brahma/api/murder/internal/model/dm.go new file mode 100644 index 0000000..9d4b9b3 --- /dev/null +++ b/app/brahma/api/murder/internal/model/dm.go @@ -0,0 +1,13 @@ +package model + +//todo:建立一个公共的model + +type Dm struct { + address string + point float64 + tags string + pictures []string + Uuid string + Phone string + Scripts *[]Scripts `json:"tags,omitempty" gorm:"many2many:business_scripts;foreignKey:Uuid;joinForeignKey:CategoryUid;References:Uuid;JoinReferences:TagUid" valid:"no_empty"` +} diff --git a/app/brahma/api/murder/internal/model/journey.go b/app/brahma/api/murder/internal/model/journey.go new file mode 100644 index 0000000..1a9e338 --- /dev/null +++ b/app/brahma/api/murder/internal/model/journey.go @@ -0,0 +1,4 @@ +package model + +type model struct { +} diff --git a/app/brahma/api/murder/internal/model/token.go b/app/brahma/api/murder/internal/model/token.go new file mode 100644 index 0000000..016ae24 --- /dev/null +++ b/app/brahma/api/murder/internal/model/token.go @@ -0,0 +1,30 @@ +package model + +import ( + "context" + "git.icechen.cn/monorepo/backend/pkg/orm" + "time" +) + +type Token struct { + Tokenid string `json:"tokenid" gorm:"column:tokenid"` // token 唯一id + OpenID string `json:"openId" gorm:"column:openId"` + TypeID string `json:"typeId" gorm:"column:typeId"` + ExpireTime time.Time `json:"expire_time" gorm:"column:expire_time"` + Level int `json:"level" gorm:"column:level"` +} + +func (m *Token) TableName() string { + return "token" +} + +func NewToken(ctx context.Context, token *Token) error { + db, err := orm.GetContextDB(ctx) + if err != nil { + return err + } + if err = db.Create(token).Error; err != nil { + return err + } + return nil +} diff --git a/app/brahma/api/murder/internal/model/user.go b/app/brahma/api/murder/internal/model/user.go new file mode 100644 index 0000000..e1f14cf --- /dev/null +++ b/app/brahma/api/murder/internal/model/user.go @@ -0,0 +1,42 @@ +package model + +import ( + "context" + "git.icechen.cn/monorepo/backend/pkg/orm" +) + +type User struct { + NickName string `json:"nick_name" gorm:"column:nick_name"` // 用户昵称 + AvatarUrl string `json:"avatar_url" gorm:"column:avatar_url"` // 用户头像图片的 URL。URL 最后一个数值代表正方形头像大小（有 0、46、64、96、132 数值可选，0 代表 640x640 的正方形头像，46 表示 46x46 的正方形头像，剩余数值以此类推。默认132），用户没有头像时该项为空。若用户更换头像，原有头像 URL 将失效。 + Openid string `json:"openid" gorm:"column:openid"` // 用户唯一标识 + SessionKey string `json:"session_key" gorm:"column:session_key"` // 会话密钥 + Typeid string `json:"typeid" gorm:"column:typeid"` // 某个小程序的唯一id，配置文件里面的key +} + +func (m *User) TableName() string { + return "user" +} + +func GetUserInfo(ctx context.Context, typeId, openId string) (*User, error) { + db, err := orm.GetContextDB(ctx) + if err != nil { + return nil, err + } + i := new(User) + affected := db.First(i, "typeid = ? AND openid = ?", typeId, openId).RowsAffected + if affected == 1 { + return i, nil + } + return nil, nil +} + +func NewUser(ctx context.Context, user *User) error { + db, err := orm.GetContextDB(ctx) + if err != nil { + return err + } + if err = db.Create(user).Error; err != nil { + return err + } + return nil +} diff --git a/app/brahma/api/murder/internal/service/token.go b/app/brahma/api/murder/internal/service/token.go new file mode 100644 index 0000000..78b7a1f --- /dev/null +++ b/app/brahma/api/murder/internal/service/token.go @@ -0,0 +1,26 @@ +package service + +import ( + "context" + "git.icechen.cn/monorepo/backend/app/brahma/api/murder/internal/model" + "time" +) + +func generateToken(ctx context.Context, user *model.User) error { + m := new(model.Token) + m.OpenID = user.Openid + m.TypeID = user.Typeid + m.Level = 1 + + if user.AvatarUrl == "" { + m.Level = 0 + } + + m.ExpireTime = time.Now().Add(8 * time.Hour) + + err := model.NewToken(ctx, m) + if err != nil { + return err + } + return nil +} diff --git a/app/brahma/api/murder/internal/service/user.go b/app/brahma/api/murder/internal/service/user.go new file mode 100644 index 0000000..4d14937 --- /dev/null +++ b/app/brahma/api/murder/internal/service/user.go @@ -0,0 +1,81 @@ +package service + +import ( + "context" + "fmt" + "git.icechen.cn/monorepo/backend/app/brahma/api/murder/internal/model" + "git.icechen.cn/monorepo/backend/pkg/config" + "git.icechen.cn/monorepo/backend/pkg/wx" + ctxLogger "github.com/luizsuper/ctxLoggers" + "math/rand" + "strconv" + "time" +) + +var appMap = make(map[string]Applet) + +func init() { + a := new(AppletConfig) + err := config.GetConfig(a) + if err != nil { + ctxLogger.Error(nil, err.Error()) + panic(err) + } + for _, v := range a.Apps { + appMap[v.Type] = v + } +} + +type AppletConfig struct { + Apps []Applet `json:"applet"` +} + +type Applet struct { + Type string `json:"type"` + Appid string `json:"appid"` + Secret string `json:"secret"` +} + +func Login(ctx context.Context, code, appType string) error { + session, err := wx.Code2Session(appMap[appType].Appid, appMap[appType].Secret, code) + if err != nil { + return err + } + + i, err := model.GetUserInfo(ctx, appType, session.OpenId) + if err != nil { + return err + } + + if i == nil { + i = new(model.User) + i.Openid = session.OpenId + err = Register(ctx, i, true) + if err != nil { + return err + } + } + + err = generateToken(ctx, i) + if err != nil { + return err + } + + return nil +} + +func Register(ctx context.Context, user *model.User, isNew bool) error { + if isNew { + rand.Seed(time.Now().UnixNano()) + intn := rand.Intn(2 ^ 1024) + itoa := strconv.Itoa(intn) + user.NickName = fmt.Sprintf("游客%v", itoa) + } + + err := model.NewUser(ctx, user) + if err != nil { + return err + } + + return nil +} diff --git a/app/brahma/api/murder/main.go b/app/brahma/api/murder/main.go index ec37619..08342fd 100755 --- a/app/brahma/api/murder/main.go +++ b/app/brahma/api/murder/main.go @@ -20,6 +20,7 @@ const ( category = "/category" scripts = "/scripts" script = "/script" + userInfo = "/userInfo" ) var ( @@ -114,6 +115,10 @@ func routerInit() { scriptGroup.Get("", func(ctx *fiber.Ctx) error { return handler.GetScriptApi(ctx) }) + userGroup := app.Group(userInfo) + userGroup.Get("login", func(ctx *fiber.Ctx) error { + return handler.Login(ctx) + }) } diff --git a/app/brahma/service/token/internal/server/bussiness.go b/app/brahma/service/token/internal/server/bussiness.go new file mode 100644 index 0000000..05a7892 --- /dev/null +++ b/app/brahma/service/token/internal/server/bussiness.go @@ -0,0 +1 @@ +package servesr diff --git a/app/brahma/service/token/internal/server/dm.go b/app/brahma/service/token/internal/server/dm.go new file mode 100644 index 0000000..8ea9348 --- /dev/null +++ b/app/brahma/service/token/internal/server/dm.go @@ -0,0 +1,8 @@ +package servesr + +type Dm struct { + Name string + Age int +} + +//多对多的表