tidy code

pkg/common/db/mysql_model/im_mysql_model/group_member_model.go

@@ -1,6 +1,7 @@
 package im_mysql_model
 
 import (
+	"Open_IM/pkg/common/constant"
 	"Open_IM/pkg/common/db"
 	"time"
 )
@@ -23,6 +24,9 @@ func InsertIntoGroupMember(toInsertInfo GroupMember) error {
 		return err
 	}
 	toInsertInfo.JoinSource = time.Now()
+	if toInsertInfo.RoleLevel == 0 {
+		toInsertInfo.RoleLevel = constant.GroupOrdinaryUsers
+	}
 	err = dbConn.Table("group_member").Create(toInsertInfo).Error
 	if err != nil {
 		return err
@@ -138,7 +142,7 @@ func GetGroupOwnerInfoByGroupID(groupID string) (*GroupMember, error) {
 		return nil, err
 	}
 	for _, v := range omList {
-		if v.RoleLevel == 1 {
+		if v.RoleLevel == constant.GroupOwner {
 			return &v, nil
 		}
 	}

pkg/common/db/mysql_model/im_mysql_model/group_request_model.go

@@ -1,6 +1,7 @@
 package im_mysql_model
 
 import (
+	"Open_IM/pkg/common/constant"
 	"Open_IM/pkg/common/db"
 	"time"
 )
@@ -88,7 +89,7 @@ func GetGroupApplicationList(userID string) ([]GroupRequest, error) {
 		return nil, err
 	}
 	for _, v := range memberList {
-		if v.RoleLevel > 0 {
+		if v.RoleLevel > constant.GroupOrdinaryUsers {
 			list, err := GetGroupRequestByGroupID(v.GroupID)
 			if err != nil {
 				continue

pkg/common/db/mysql_model/im_mysql_model/user_model.go

@@ -2,6 +2,7 @@ package im_mysql_model
 
 import (
 	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/constant"
 	"Open_IM/pkg/common/db"
 	"Open_IM/pkg/utils"
 	"fmt"
@@ -15,13 +16,13 @@ func init() {
 		user, err := GetUserByUserID(v)
 		if err != nil {
 			fmt.Println("GetUserByUserID failed ", err.Error(), v, user)
-		}else{
+		} else {
 			continue
 		}
 		var appMgr User
 		appMgr.UserID = v
 		appMgr.Nickname = "AppManager" + utils.IntToString(k+1)
-		appMgr.AppMangerLevel = 2
+		appMgr.AppMangerLevel = constant.AppAdmin
 		err = UserRegister(appMgr)
 		if err != nil {
 			fmt.Println("AppManager insert error", err.Error())
@@ -36,7 +37,9 @@ func UserRegister(user User) error {
 		return err
 	}
 	user.CreateTime = time.Now()
-
+	if user.AppMangerLevel == 0 {
+		user.AppMangerLevel = constant.AppOrdinaryUsers
+	}
 	err = dbConn.Table("user").Create(&user).Error
 	if err != nil {
 		return err