三十六、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)//最大空闲连接，一般为最大连接数/2db.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 daoimport ("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.Accounterr := a.db.Where("user_id=?", userID).First(&account).Error//返回查询不到错误返回falseif 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": "账号已存在"})}

五、实验