Add emailRegister check (#308)

* fix bug

* Your commit message

* fix bug register email

* fix bug email resetpassword

* fix bug mysql sleep

* fix bug login.go

* fix bug login.go

* fix bug login.go

* fix bug login.go

* fix bug login.go

* fix bug login.go

* fix bug pb

* fix bug pb

* update to adjust more type

internal/rpc/chat/login.go

@@ -321,6 +321,7 @@ func (o *chatSvr) RegisterUser(ctx context.Context, req *chat.RegisterUserReq) (
 			return nil, err
 		}
 	}
+	var registerType int32
 	if req.User.PhoneNumber != "" {
 		if req.User.AreaCode[0] != '+' {
 			req.User.AreaCode = "+" + req.User.AreaCode
@@ -337,7 +338,9 @@ func (o *chatSvr) RegisterUser(ctx context.Context, req *chat.RegisterUserReq) (
 		} else if !o.Database.IsNotFound(err) {
 			return nil, err
 		}
+		registerType = constant.PhoneRegister
 	}
+
 	if req.User.Account != "" {
 		_, err := o.Database.TakeAttributeByAccount(ctx, req.User.Account)
 		if err == nil {
@@ -346,8 +349,10 @@ func (o *chatSvr) RegisterUser(ctx context.Context, req *chat.RegisterUserReq) (
 			return nil, err
 		}
 	}
+
 	if req.User.Email != "" {
 		_, err := o.Database.TakeAttributeByEmail(ctx, req.User.Email)
+		registerType = constant.EmailRegister
 		if err == nil {
 			return nil, eerrs.ErrEmailAlreadyRegister.Wrap()
 		} else if !o.Database.IsNotFound(err) {
@@ -385,6 +390,7 @@ func (o *chatSvr) RegisterUser(ctx context.Context, req *chat.RegisterUserReq) (
 		AllowVibration: constant.DefaultAllowVibration,
 		AllowBeep:      constant.DefaultAllowBeep,
 		AllowAddFriend: constant.DefaultAllowAddFriend,
+		RegisterType:   registerType,
 	}
 	if err := o.Database.RegisterUser(ctx, register, account, attribute); err != nil {
 		return nil, err

internal/rpc/chat/utils.go

@@ -57,6 +57,7 @@ func DbToPbUserFullInfo(attribute *chat.Attribute) *common.UserFullInfo {
 		AllowBeep:        attribute.AllowBeep,
 		AllowVibration:   attribute.AllowVibration,
 		GlobalRecvMsgOpt: attribute.GlobalRecvMsgOpt,
+		RegisterType:     attribute.RegisterType,
 	}
 }
 

pkg/common/constant/constant.go

@@ -184,3 +184,8 @@ const (
 const DefaultPlatform = 1
 
 const CtxApiToken = "api-token"
+
+const (
+	EmailRegister = 1
+	PhoneRegister = 2
+)

pkg/common/db/table/chat/attribute.go

@@ -37,6 +37,7 @@ type Attribute struct {
 	AllowBeep        int32     `gorm:"column:allow_beep;default:1"`
 	AllowAddFriend   int32     `gorm:"column:allow_add_friend;default:1"`
 	GlobalRecvMsgOpt int32     `gorm:"column:global_recv_msg_opt;default:0"`
+	RegisterType     int32     `gorm:"column:register_type"`
 }
 
 func (Attribute) TableName() string {