三十六、Gin注册功能-检查账号是否存在

一、初始化

1、在cms.go中添加数据库连接方法

复制代码
func connDB(app *CmsApp) {
	mysqlDB, err := gorm.Open(mysql.Open("root:rootroot@tcp(localhost:3306)/?charset=utf8mb4&parseTime=True&loc=Local"))
	if err != nil {
		panic(err)
	}
	db, err := mysqlDB.DB()
	if err != nil {
		panic(err)
	}
	//最大连接数
	db.SetMaxOpenConns(4)
	//最大空闲连接,一般为最大连接数/2
	db.SetMaxIdleConns(2)
	mysqlDB = mysqlDB.Debug()
	app.db = mysqlDB
}

2、在cms.go中进行初始化数据库连接方法

复制代码
type CmsApp struct {
	db *gorm.DB
}

func NewCmsApp() *CmsApp {
	app := &CmsApp{}
	connDB(app)
	return app
}

二、完成account表的字段映射model

复制代码
type Account struct {
	ID        int64     `gorm:"column:id;primaryKey;autoIncrement:true"`
	UserId    string    `gorm:"column:user_id"`
	Password  string    `gorm:"column:password"`
	Nickname  string    `gorm:"column:nickname"`
	CreatedAt time.Time `gorm:"column:created_at"`
	UpdatedAt time.Time `gorm:"column:updated_at"`
}

func (a Account) TableName() string {
    //在数据表中省略了库名,写在了这里因为这样更灵活,方便做分库分表
	table := "cms_account.account"
	return table
}

三、完成account的dao方法

复制代码
package dao

import (
	"ConnetMain/internal/model"
	"gorm.io/gorm"
)
//声明为gorm的db
type AccountDao struct {
	db *gorm.DB
}
//调用时引入初始化CmsApp.db
func NewAccountDao(db *gorm.DB) *AccountDao {
	return &AccountDao{db: db}
}
//具体实现方法,查询是否存在这个user_id
func (a *AccountDao) IsExist(userID string) (bool, error) {
	var account model.Account
	err := a.db.Where("user_id=?", userID).First(&account).Error
    //返回查询不到错误返回false
	if err == gorm.ErrRecordNotFound {
		return false, err
	}
    //返回错误不为空
	if err != nil {
		return false, err
	}
	return true, nil
}

四、具体调用

在register.go中进行调用IsExist方法

复制代码
//声明dao包下NewAccountDao
accountDao := dao.NewAccountDao(c.db)
//调用IsExist方法
	isExist, err := accountDao.IsExist(req.UserID)
//发送错误则终止程序
	if err != nil {
		ctx.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}
//返回true则代表能吃查询到相关数据则表示账号已经存在
	if isExist {
		ctx.JSON(http.StatusBadRequest, gin.H{"err": "账号已存在"})
	}

五、实验

相关推荐
无小道2 分钟前
c++--typedef和#define的用法及区别
c语言·开发语言·汇编·c++
考虑考虑11 分钟前
Redis8新增特性
redis·后端·程序员
labixiong16 分钟前
全方位理解跨源资源共享-CORS
前端·后端
AntBlack19 分钟前
闲谈 :AI编程效率反而降低了 ,大家AI 编程的正确姿势到底是什么?
前端·后端·ai编程
SoniaChen3331 分钟前
Rust基础-part2-变量和可变类型
开发语言·后端·rust
Victor35638 分钟前
MySQL(131)如何解决MySQL CPU使用率过高问题?
后端
深栈解码41 分钟前
第二章:Class文件解剖:字节码的二进制密码
java·后端
error_cn43 分钟前
finger命令输出格式解析
后端
Victor35644 分钟前
MySQL(132)如何调整MySQL内存使用?
后端
武子康1 小时前
大数据-38 Redis 分布式缓存 详细介绍 缓存、读写、旁路、穿透模式
大数据·redis·后端