project_management/app/model/appUser.go

package model

import (
	"context"
	"project_management/app/e"
	"project_management/global"
	"project_management/unity"
	"project_management/utils"
	"strings"
	"time"
)

type AppUser struct {
	//gorm.Model
	utils.BaseModel
	Username      string `gorm:"type:varchar(50);index:username_name,unique" json:"username" min:"3" max:"50"` // 用户名
	Phone         string `gorm:"type:varchar(50);" json:"phone" min:"11" max:"11"`                             // 手机号
	Nickname      string `gorm:"type:varchar(50);" json:"nickname"`                                            // 昵称
	Password      string `gorm:"type:varchar(50);" json:"password"`                                            // 密码
	LastLoginTime string `gorm:"type:datetime;" json:"last_login_time"`                                        // 最后登录时间
	State         int    `gorm:"type:int;" json:"state"`                                                       // 状态 0:正常 1:禁用
	AppID         string `gorm:"type:varchar(50);" json:"app_id" `                                             // 应用id
	RegistMethod  int    `gorm:"type:int;" json:"regist_method"`                                               // 注册方式 1 短信登录 2 微信登录 3 系统添加
}

type AppUserRegist struct {
	Username     string `gorm:"type:varchar(50);index:username_name,unique" json:"username" validate:"required" min:"3" max:"50"` // 用户名
	Phone        string `gorm:"type:varchar(50);" json:"phone" validate:"required" min:"11" max:"11"`                             // 手机号
	Nickname     string `gorm:"type:varchar(50);" json:"nickname"`                                                                // 昵称
	Password     string `gorm:"type:varchar(50);" json:"password" validate:"required" min:"6" max:"20"`                           // 密码
	AppID        string `gorm:"type:varchar(50);not null;unique" json:"app_id" validate:"required"`                               // 应用id
	Code         string `json:"code" validate:"required" min:"6" max:"6"`                                                         // 验证码
	RegistMethod int    `gorm:"type:int;" json:"regist_method"`                                                                   // 注册方式 1 短信登录 2 微信登录 3 系统添加
}

func (a AppUser) DeleteAppUserByID(id int, tableName string) e.Rescode {
	//TODO implement me
	tx := global.DBLink.Table(tableName).Where("id=?", id).Delete(&a)
	if tx.Error != nil {
		return e.DELETEFAIL
	}
	return e.SUCCESS
}
func (a AppUser) UpdateAppUser(appUser AppUser, tabaleName string) e.Rescode {
	//TODO implement me
	tx := global.DBLink.Table(tabaleName).Updates(appUser)
	if tx.Error != nil {
		return e.UPDATEFAIL
	}
	return e.SUCCESS
}

func (a AppUser) GetAppUserList(params unity.QueryPageParams, tableName string, queryCond string) (result []AppUser, total int64, err error) {
	query := global.DBLink.Table(tableName)
	if params.Query != "%%" {
		query = query.Where(queryCond, params.Query)
	}
	if params.State != "" {
		query = query.Where("state=?", params.State)
	}
	if err = query.Count(&total).Error; err != nil {
		return nil, 0, err
	}
	offset := (params.Page - 1) * params.Size
	if err = query.Offset(offset).Limit(params.Size).Order(params.Desc).Find(&result).Error; err != nil {
		return nil, 0, err
	}
	return result, total, nil
}

// RegistAppUser 注册用户
func (a AppUser) RegistAppUser(appUser AppUserRegist) e.Rescode {
	//TODO implement me
	tx := global.DBLink.Select("phone").Where("phone = ?", appUser.Phone).First(&a)
	ctx := context.Background()
	result, err := global.Rdb.Get(ctx, appUser.Phone).Result()
	if err != nil {
		return e.TheVerificationCodeWasNotSent
	} else if result != appUser.Code {
		return e.CodeIsError
	}
	if tx.RowsAffected == 0 {
		a.AppID = appUser.AppID
		a.State = 0
		a.Phone = appUser.Phone
		a.Username = appUser.Username
		a.Nickname = "游客" + unity.RandomId()
		md5 := utils.MD5(appUser.Password)
		a.Password = md5
		a.LastLoginTime = time.Now().Format("2006-01-02 15:04:05")
		a.RegistMethod = appUser.RegistMethod
		tableName := "appUser_" + appUser.AppID
		//验证表是否已经创建，没有则创建
		table := global.DBLink.Migrator().HasTable(tableName)
		if !table {
			global.DBLink.Table(tableName).AutoMigrate(&AppUser{})
		}
		db := global.DBLink.Table(tableName).Create(&a)
		if db.Error != nil {
			return e.RegistrationFailed
		}
		return e.SUCCESS
	}
	return e.AlreadyExists
}

// AddAppUser 添加用户
func (a AppUser) AddAppUser(appUser AppUser) e.Rescode {
	//TODO implement me
	//验证表是否已经创建，没有则创建
	tableName := "appUser_" + appUser.AppID
	table := global.DBLink.Migrator().HasTable(tableName)
	if !table {
		global.DBLink.Table(tableName).AutoMigrate(&AppUser{})
	}
	a.AppID = appUser.AppID
	a.State = 0
	a.Nickname = appUser.Nickname
	a.Username = appUser.Username
	a.Password = utils.MD5(appUser.Password)
	a.LastLoginTime = time.Now().Format("2006-01-02 15:04:05")
	a.RegistMethod = 3
	if err := global.DBLink.Table(tableName).Create(&a).Error; err != nil {
		if strings.Contains(err.Error(), "Duplicate entry") {
			return e.TheUserAlreadyExists
		}
		return e.AddAppUserFail
	}
	return e.SUCCESS
}

// GetAllUserCount 获取用户总数
func GetAllUserCount(tableName string) (e.Rescode, int64) {
	var count int64
	tx := global.DBLink.Table(tableName).Count(&count)
	if tx.Error != nil {
		return e.ERROR, 0
	}
	return e.SUCCESS, count
}