feat:完善信息

This commit is contained in:
henry
2021-09-29 16:25:56 +08:00
parent b876eab301
commit c556053feb
16 changed files with 382 additions and 84 deletions

View File

@ -54,7 +54,7 @@ func (c *Account) Login(account, password, captchaKey, captchaValue, equipment,
return nil, err return nil, err
} else if !isExist { } else if !isExist {
return nil, errors.New("当前不属于任何租户,不可登录") return nil, errors.New("当前不属于任何租户,不可登录")
} else if mSysUserTenant.Status != model2.SysUserTenantStatusForEnable { } else if mSysUserTenant.Status != model2.AccountStatusForEnable {
return nil, errors.New("当前账号已禁用,不可登录,请联系管理员!") return nil, errors.New("当前账号已禁用,不可登录,请联系管理员!")
} }
if mSysUserTenant.TenantID <= 0 { if mSysUserTenant.TenantID <= 0 {

View File

@ -51,6 +51,26 @@ func (m *Images) AnalysisSlice(domain string) []string {
return images return images
} }
// AccountStatus 账号状态
type AccountStatus struct {
Status AccountStatusKind `gorm:"column:status;type:tinyint(1);default:1;comment:状态(1启用2禁用)" json:"-"`
}
// AccountStatusKind 状态
type AccountStatusKind int
const (
// AccountStatusForEnable 启用
AccountStatusForEnable AccountStatusKind = iota + 1
// AccountStatusForDisable 禁用
AccountStatusForDisable
)
// Format 格式化
func (m *AccountStatus) Format() string {
return utils.AnyToJSON(m)
}
// Position 坐标信息 // Position 坐标信息
type Position struct { type Position struct {
Longitude float64 `json:"longitude"` // 经度 Longitude float64 `json:"longitude"` // 经度
@ -61,14 +81,3 @@ type Position struct {
func (m *Position) Format() string { func (m *Position) Format() string {
return utils.AnyToJSON(m) return utils.AnyToJSON(m)
} }
// Tags 标签
type Tags struct {
Key string `json:"key"`
Value string `json:"value"`
}
// Format 格式化
func (m *Tags) Format() string {
return utils.AnyToJSON(m)
}

View File

@ -7,7 +7,7 @@ type SysUserTenant struct {
Department string `gorm:"column:department;type:varchar(100);default:null;comment:部门信息" json:"department"` Department string `gorm:"column:department;type:varchar(100);default:null;comment:部门信息" json:"department"`
Role string `gorm:"column:role;type:varchar(100);default:null;comment:角色信息" json:"role"` Role string `gorm:"column:role;type:varchar(100);default:null;comment:角色信息" json:"role"`
Identity SysUserTenantIdentity `gorm:"column:identity;type:tinyint(1);default:0;comment:用户身份1管理员2用户" json:"-"` Identity SysUserTenantIdentity `gorm:"column:identity;type:tinyint(1);default:0;comment:用户身份1管理员2用户" json:"-"`
Status SysUserTenantStatus `gorm:"column:status;type:tinyint(1);default:1;comment:状态(1启用2禁用)" json:"-"` AccountStatus
ModelDeleted ModelDeleted
ModelAt ModelAt
} }
@ -21,16 +21,6 @@ const (
SysUserTenantIdentityForSystemUser SysUserTenantIdentityForSystemUser
) )
// SysUserTenantStatus 状态
type SysUserTenantStatus int
const (
// SysUserTenantStatusForEnable 启用
SysUserTenantStatusForEnable SysUserTenantStatus = iota + 1
// SysUserTenantStatusForDisable 禁用
SysUserTenantStatusForDisable
)
func (m *SysUserTenant) TableName() string { func (m *SysUserTenant) TableName() string {
return m.NewTableName("sys_user_tenant") return m.NewTableName("sys_user_tenant")
} }

View File

@ -1,13 +1,26 @@
package model package model
import (
"SciencesServer/utils"
"gorm.io/gorm"
"time"
)
type TenantUser struct { type TenantUser struct {
Model Model
ModelTenant UUID uint64 `gorm:"column:uuid;uniqueIndex:idx_tenant_user_uuid;type:int;default:0;comment:用户唯一UUID" json:"-"`
UUID uint64 `gorm:"column:uuid;uniqueIndex:idx_sys_user_uuid;type:int;default:0;comment:用户唯一UUID" json:"-"` Avatar string `gorm:"column:avatar;type:varchar(255);default:null;comment:头像" json:"avatar"`
Avatar string `gorm:"column:avatar;type:varchar(255);default:null;comment:头像" json:"avatar"` Name string `gorm:"column:name;type:varchar(20);default:null;comment:真实姓名" json:"name"`
Name string `gorm:"column:name;type:varchar(20);default:null;comment:真实姓名" json:"name"` Mobile string `gorm:"column:mobile;index:idx_tenant_user_mobile;type:varchar(15);default:null;comment:联系方式" json:"mobile"`
Mobile string `gorm:"column:mobile;index:idx_sys_user_mobile;type:varchar(15);default:null;comment:联系方式" json:"mobile"` Email string `gorm:"column:email;type:varchar(50);default:null;comment:邮箱" json:"email"`
Email string `gorm:"column:email;type:varchar(50);default:null;comment:邮箱" json:"email"` Identity int `gorm:"column:identity;type:int(8);default:0;comment:身份信息" json:"-"`
Password string `gorm:"column:password;type:varchar(100);default:null;comment:密码" json:"-"`
Salt string `gorm:"column:salt;type:varchar(10);default:null;comment:盐值" json:"-"`
Province uint64 `gorm:"column:province;type:int;default:0;comment:所在省" json:"province"`
City uint64 `gorm:"column:city;type:int;default:0;comment:所在市" json:"city"`
District uint64 `gorm:"column:district;type:int;default:0;comment:所在区/县" json:"district"`
Address string `gorm:"column:address;type:varchar(255);default:null;comment:详细地址" json:"address"`
AccountStatus
ModelDeleted ModelDeleted
ModelAt ModelAt
} }
@ -16,6 +29,20 @@ func (m *TenantUser) TableName() string {
return m.NewTableName("tenant_user") return m.NewTableName("tenant_user")
} }
func (m *TenantUser) BeforeCreate(db *gorm.DB) error {
m.NewPassword()
snowflake, _ := utils.NewSnowflake(1)
m.UUID = uint64(snowflake.GetID())
m.Status = AccountStatusForEnable
m.CreatedAt = time.Now()
return nil
}
func (m *TenantUser) NewPassword() {
m.Salt = utils.GetRandomString(8)
m.Password = utils.HashString([]byte(utils.Md5String(m.Password, m.Salt)))
}
func NewTenantUser() *TenantUser { func NewTenantUser() *TenantUser {
return &TenantUser{} return &TenantUser{}
} }

View File

@ -3,6 +3,7 @@ package api
import ( import (
"SciencesServer/app/common/api" "SciencesServer/app/common/api"
"SciencesServer/app/enterprise/controller/account" "SciencesServer/app/enterprise/controller/account"
"SciencesServer/app/service"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
@ -10,10 +11,17 @@ type Account struct{}
type ( type (
accountLoginForm struct { accountLoginForm struct {
Mode int `json:"mode" form:"mode" binding:"required"` Mode int `json:"mode" form:"mode" binding:"required"`
Mobile string `json:"mobile" form:"mobile"`
Captcha string `json:"captcha" form:"captcha"`
Password string `json:"password" form:"password"`
} }
accountRegisterForm struct { accountRegisterForm struct {
Name string `json:"name" form:"name" binding:"required"`
Mobile string `json:"mobile" form:"mobile" binding:"required"`
Captcha string `json:"captcha" form:"captcha" binding:"required"`
Password string `json:"password" form:"password" binding:"required"`
RepeatPass string `json:"repeat_pass" form:"repeat_pass" binding:"required"`
} }
) )
@ -24,22 +32,45 @@ func (a *Account) Login(c *gin.Context) {
api.APIFailure(err.(error))(c) api.APIFailure(err.(error))(c)
return return
} }
data, err := account.NewLogin().Login()(account.LoginMode(form.Mode), nil) data, err := account.NewLogin()().Launch(account.LoginMode(form.Mode), &account.LoginRequest{
Captcha: struct {
Mobile string
Captcha string
}{Mobile: form.Mobile, Captcha: form.Captcha},
Password: struct {
Mobile string
Password string
}{Mobile: form.Mobile, Password: form.Password},
})
api.APIResponse(err, data) api.APIResponse(err, data)
} }
func (a *Account) Register() { func (a *Account) Register(c *gin.Context) {
form := new(accountRegisterForm)
if err := api.Bind(form)(c); err != nil {
api.APIFailure(err.(error))(c)
return
}
data, err := account.NewRegister()().Launch(&account.RegisterRequest{
Name: form.Name, Mobile: form.Mobile, Captcha: form.Captcha,
Password: form.Password, RepeatPass: form.RepeatPass,
})
api.APIResponse(err, data)
} }
func (c *Account) BindName() { func (a *Account) BindMobile() {
account.NewOther()().BindMobile()
} }
func (c *Account) BindMobile() { func (a *Account) Logout(c *gin.Context) {
handle := api.GetSession()(c)
}
func (a *Account) Logout() {
session := new(service.SessionEnterprise)
if handle != nil {
session = handle.(*service.SessionEnterprise)
}
err := account.NewLogout()(session).Launch()
api.APIResponse(err)(c)
} }

View File

@ -0,0 +1,14 @@
package api
import (
"SciencesServer/app/common/api"
"SciencesServer/app/enterprise/controller"
"github.com/gin-gonic/gin"
)
type Config struct{}
func (a *Config) Identity(c *gin.Context) {
data := controller.NewConfig().Identity()
api.APISuccess(data)
}

View File

@ -0,0 +1,3 @@
package api
type User struct{}

View File

@ -0,0 +1,41 @@
package account
import (
"SciencesServer/app/enterprise/model"
"SciencesServer/app/service"
"SciencesServer/config"
"SciencesServer/utils"
)
type Account struct{}
type LoginCallback func(user *model.TenantUser) *LoginResponse
type (
LoginResponse struct {
Token string `json:"token"`
EffectTime int `json:"effect_time"`
}
)
func (c *Account) Login() LoginCallback {
return func(mTenantUser *model.TenantUser) *LoginResponse {
token := utils.JWTEncrypt(config.SettingInfo.TokenEffectTime, map[string]interface{}{
config.TokenForUID: mTenantUser.UUID,
})
session := service.NewSessionEnterprise()
session.Token = token
session.UID = mTenantUser.UUID
session.Name = mTenantUser.Name
session.Mobile = mTenantUser.Mobile
session.Identity = mTenantUser.Identity
service.Publish(config.EventForRedisHashProduce, config.RedisKeyForAccount, mTenantUser.UUIDToString(), session)
return &LoginResponse{Token: token, EffectTime: config.SettingInfo.TokenEffectTime}
}
}
func NewAccount() *Account {
return &Account{}
}

View File

@ -4,7 +4,6 @@ import (
model2 "SciencesServer/app/common/model" model2 "SciencesServer/app/common/model"
"SciencesServer/app/enterprise/model" "SciencesServer/app/enterprise/model"
"SciencesServer/app/handle" "SciencesServer/app/handle"
"SciencesServer/config"
"SciencesServer/utils" "SciencesServer/utils"
"errors" "errors"
) )
@ -12,48 +11,46 @@ import (
type Login struct{} type Login struct{}
type ( type (
LoginHandle func(LoginMode, *LoginRequest) (*LoginResponse, error) LoginHandle func() *Login
) )
type ( type (
LoginRequest struct { LoginRequest struct {
Captcha struct { Captcha struct {
Mobile string `json:"mobile"` Mobile string
Captcha string `json:"captcha"` Captcha string
} }
Password struct { Password struct {
Mobile string `json:"mobile"` Mobile string
Password string `json:"password"` Password string
} }
Platform struct { Platform struct {
OpenID string `json:"open_id"` OpenID string
} }
} }
LoginResponse struct {
Token string `json:"token"`
EffectTime int `json:"effect_time"`
}
) )
// LoginMode 登陆模式 // LoginMode 登陆模式
type LoginMode int type LoginMode int
const ( const (
LoginModeForCaptcha LoginMode = iota + 1e2 + 1 // 验证码登陆 LoginModeForSmsCaptcha LoginMode = iota + 1e2 + 1 // 短信验证码登陆
LoginModeForPassword // 密码登陆 LoginModeForPassword // 密码登陆
LoginModeForWechat // 微信登陆 LoginModeForWechat // 微信登陆
LoginModeForQQ // QQ登陆 LoginModeForQQ // QQ登陆
) )
var loginHandle = map[LoginMode]func(*LoginRequest) (*model.TenantUser, error){ var loginHandle = map[LoginMode]func(*LoginRequest) (*model.TenantUser, error){
LoginModeForCaptcha: loginForCaptcha, LoginModeForPassword: loginForPassword, LoginModeForSmsCaptcha: loginForSmsCaptcha, LoginModeForPassword: loginForPassword,
} }
// loginForCaptcha 验证码登陆 // loginForSmsCaptcha 短信验证码登陆
func loginForCaptcha(req *LoginRequest) (*model.TenantUser, error) { func loginForSmsCaptcha(req *LoginRequest) (*model.TenantUser, error) {
if !utils.ValidateMobile(req.Captcha.Mobile) {
return nil, errors.New("手机号码格式异常")
}
pass, err := handle.NewCaptcha().Validate(&handle.CaptchaSms{ pass, err := handle.NewCaptcha().Validate(&handle.CaptchaSms{
Mobile: req.Captcha.Mobile, Mobile: req.Captcha.Mobile, Captcha: req.Captcha.Captcha,
Captcha: req.Captcha.Captcha,
}) })
if err != nil { if err != nil {
return nil, err return nil, err
@ -64,20 +61,26 @@ func loginForCaptcha(req *LoginRequest) (*model.TenantUser, error) {
mTenantUsr := model.NewTenantUser() mTenantUsr := model.NewTenantUser()
if isExist, err = model2.FirstField(mTenantUsr.TenantUser, []string{"id", "name", "status"}, if isExist, err = model2.FirstField(mTenantUsr.TenantUser, []string{"id", "uuid", "name", "mobile", "status"},
model2.NewWhere("mobile", req.Captcha.Mobile)); err != nil { model2.NewWhere("mobile", req.Captcha.Mobile)); err != nil {
return nil, err return nil, err
} else if isExist { } else if isExist {
return nil, errors.New("当前手机号码未注册") return mTenantUsr, nil
} }
mTenantUsr.Name = req.Captcha.Mobile
mTenantUsr.Password = utils.GetRandomString(12)
return mTenantUsr, nil return mTenantUsr, nil
} }
// loginForPassword 密码登陆 // loginForPassword 密码登陆
func loginForPassword(req *LoginRequest) (*model.TenantUser, error) { func loginForPassword(req *LoginRequest) (*model.TenantUser, error) {
if !utils.ValidateMobile(req.Password.Mobile) {
return nil, errors.New("手机号码格式异常")
}
mTenantUsr := model.NewTenantUser() mTenantUsr := model.NewTenantUser()
isExist, err := model2.FirstField(mTenantUsr.TenantUser, []string{"id", "name", "status"}, isExist, err := model2.FirstField(mTenantUsr.TenantUser, []string{"id", "uuid", "name", "mobile",
"password", "salt", "status"},
model2.NewWhere("mobile", req.Password.Mobile)) model2.NewWhere("mobile", req.Password.Mobile))
if err != nil { if err != nil {
@ -95,27 +98,46 @@ func loginForPlatform(req *LoginRequest) error {
return nil return nil
} }
func (c *Login) Login() LoginHandle { func (c *Login) Launch(mode LoginMode, req *LoginRequest) (*LoginResponse, error) {
return func(mode LoginMode, req *LoginRequest) (*LoginResponse, error) { handle, has := loginHandle[mode]
handle, has := loginHandle[mode]
if !has { if !has {
return nil, errors.New("未知的登陆模式") return nil, errors.New("未知的登陆模式")
} }
user, err := handle(req) user, err := handle(req)
if err != nil { if err != nil {
return nil, err return nil, err
} }
token := utils.JWTEncrypt(config.SettingInfo.TokenEffectTime, map[string]interface{}{ if user.Status != model2.AccountStatusForEnable {
config.TokenForUID: user.UUID, return nil, errors.New("该账号已禁止登陆,请联系管理员")
}) }
return &LoginResponse{ return NewAccount().Login()(user), err
Token: token, EffectTime: config.SettingInfo.TokenEffectTime, }
}, nil
func (c *Login) BindName(token, name string) {
// 解析token
tokenInfo := utils.JWTDecrypt(token)
if tokenInfo == nil {
//return errors.New("")
} }
} }
func NewLogin() *Login { func (c *Login) BindMobile(token, mobile, captcha string) (*LoginResponse, error) {
return &Login{} pass, err := handle.NewCaptcha().Validate(&handle.CaptchaSms{
Mobile: mobile, Captcha: captcha,
})
if err != nil {
return nil, err
} else if !pass {
return nil, errors.New("验证码错误或已过期")
}
return nil, err
}
func NewLogin() LoginHandle {
return func() *Login {
return &Login{}
}
} }

View File

@ -0,0 +1,22 @@
package account
import (
"SciencesServer/app/service"
"SciencesServer/config"
"SciencesServer/utils"
)
type Logout struct{ *service.SessionEnterprise }
type LogoutHandle func(*service.SessionEnterprise) *Logout
func (c *Logout) Launch() error {
service.Publish(config.EventForRedisHashDestroy, config.RedisKeyForAccount, utils.UintToString(c.UID))
return nil
}
func NewLogout() LogoutHandle {
return func(enterprise *service.SessionEnterprise) *Logout {
return &Logout{enterprise}
}
}

View File

@ -0,0 +1,15 @@
package account
type Other struct{}
type OtherHandle func() *Other
func (c *Other) BindMobile() {
}
func NewOther() OtherHandle {
return func() *Other {
return &Other{}
}
}

View File

@ -0,0 +1,68 @@
package account
import (
model2 "SciencesServer/app/common/model"
"SciencesServer/app/enterprise/model"
"SciencesServer/app/handle"
"errors"
)
type Register struct{}
type RegisterHandle func() *Register
type (
RegisterRequest struct {
Name, Mobile, Captcha, Password, RepeatPass string
}
)
func (c *RegisterRequest) checkPassword() bool {
return c.Password == c.RepeatPass
}
func (c *RegisterRequest) checkUserExist(mTenantUser *model.TenantUser) (bool, error) {
var count int64
if err := model2.Count(mTenantUser.TenantUser, &count, model2.NewWhere("mobile", c.Mobile)); err != nil {
return false, err
}
return count > 0, nil
}
func (c *RegisterRequest) checkCaptcha() (bool, error) {
return handle.NewCaptcha().Validate(&handle.CaptchaSms{Captcha: c.Captcha, Mobile: c.Mobile})
}
func (c *Register) Launch(req *RegisterRequest) (*LoginResponse, error) {
if req.checkPassword() {
return nil, errors.New("两次密码不一致")
}
mTenantUser := model.NewTenantUser()
pass, err := req.checkUserExist(mTenantUser)
if err != nil {
return nil, err
} else if pass {
return nil, errors.New("当前手机号码已注册")
}
if pass, err = req.checkCaptcha(); err != nil {
return nil, err
} else if !pass {
return nil, errors.New("验证码错误或已过期")
}
mTenantUser.Name = req.Name
mTenantUser.Mobile = req.Mobile
mTenantUser.Password = req.Password
if err = model2.Create(mTenantUser.TenantUser); err != nil {
return nil, err
}
return NewAccount().Login()(mTenantUser), err
}
func NewRegister() RegisterHandle {
return func() *Register {
return &Register{}
}
}

View File

@ -0,0 +1,13 @@
package controller
import "SciencesServer/config"
type Config struct{}
func (c *Config) Identity() map[uint]string {
return config.TenantUserIdentityData
}
func NewConfig() *Config {
return &Config{}
}

View File

@ -1,17 +1,20 @@
package model package model
import "SciencesServer/app/common/model" import (
"SciencesServer/app/common/model"
"SciencesServer/utils"
)
type TenantUser struct { type TenantUser struct {
*model.TenantUser *model.TenantUser
} }
func (m *TenantUser) UUIDToString() string { func (m *TenantUser) UUIDToString() string {
return "" return utils.UintToString(m.UUID)
} }
func (m *TenantUser) ValidatePassword(password string) bool { func (m *TenantUser) ValidatePassword(password string) bool {
return true return utils.HashCompare([]byte(m.Password), []byte(utils.Md5String(password, m.Salt)))
} }
func NewTenantUser() *TenantUser { func NewTenantUser() *TenantUser {

View File

@ -26,3 +26,24 @@ func (this *Session) UnmarshalBinary(data []byte) error {
func NewSession() *Session { func NewSession() *Session {
return &Session{} return &Session{}
} }
// SessionEnterprise 企业用户
type SessionEnterprise struct {
UID uint64 `json:"uid"` // 唯一标识ID
Token string `json:"token"` // token
Name string `json:"name"` // 名称
Mobile string `json:"mobile"` // 手机号码
Identity int `json:"identity"` // 身份信息
}
func (this *SessionEnterprise) MarshalBinary() ([]byte, error) {
return json.Marshal(this)
}
func (this *SessionEnterprise) UnmarshalBinary(data []byte) error {
return utils.FromJSONBytes(data, this)
}
func NewSessionEnterprise() *SessionEnterprise {
return &SessionEnterprise{}
}

19
config/tenant.go Normal file
View File

@ -0,0 +1,19 @@
package config
// TenantUserIdentity 租户用户身份信息
type TenantUserIdentity uint
const (
TenantUserIdentityForCompany uint = 1 // 1
TenantUserIdentityForExpert uint = 1 << 1 // 2
TenantUserIdentityForResearch uint = 1 << 2 // 4
TenantUserIdentityForLaboratory uint = 1 << 3 // 8
TenantUserIdentityForAgent uint = 1 << 4 // 16
)
// TenantUserIdentityData 用户身份信息
var TenantUserIdentityData = map[uint]string{
TenantUserIdentityForCompany: "企业", TenantUserIdentityForExpert: "专家",
TenantUserIdentityForResearch: "研究机构", TenantUserIdentityForLaboratory: "实验室",
TenantUserIdentityForAgent: "即可经纪人",
}