texas-poker-bk/internal/dao/user_dao.go

package dao

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
	"net/http"
	"texas-poker-bk/internal/model/entity"
)

var UserDao *User

func initUserDao() {
	UserDao = &User{DB: Dao.Sqlite}
	err := Dao.Sqlite.AutoMigrate(&entity.User{})
	if err != nil {
		fmt.Println("user table", err)
	}
}

type User struct {
	DB *gorm.DB
}

func FindUserByName(ctx *gin.Context) {
	username := ctx.Param("username")
	user := UserDao.FindUserByName(username)
	if user == nil {
		ctx.JSON(http.StatusOK, gin.H{
			"msg": "not found",
		})
		return
	}
	ctx.JSON(http.StatusOK, gin.H{
		"msg":  "ok",
		"data": user,
	})
}

func (u *User) FindUserByName(username string) *entity.User {
	user := &entity.User{}
	tx := u.DB.Model(user).Where("username=?", username).Limit(1).Scan(user)
	// 查询出结果时 tx.RowsAffected 固定=1
	if tx.RowsAffected == 0 {
		return nil
	}
	return user
}