go_mqtt/models/Useryf.go

package models

import (
	"errors"
	"fmt"
	"go_mqtt/mydb"
	"gorm.io/gorm"
)

type Useryf struct {
	Username string `gorm:"primaryKey;size:100"`
	Password string `gorm:"size:255"`
	Date     string `gorm:"size:32"`
	Nfc      string `gorm:"size:255"`
	Name     string `gorm:"size:255"`
}

func init() {
	fmt.Println("Useryf init()")
	mydb.DB.AutoMigrate(&Useryf{})
}

// TableName 会将 User 的表名重写为 `user`
func (Useryf) TableName() string {
	return "useryf"
}

func SaveUseryf(user *Useryf) {
	result := mydb.DB.Create(&user)
	if result.Error != nil {
		fmt.Println("Failed to create User:", result.Error)
	} else {
		fmt.Println("User created successfully!")
	}
}

func UpdateUseryf(item *Useryf) {
	fmt.Println("用户存在,更新操作")
	var user Useryf
	//mydb.DB.First(&user, item.Username)
	mydb.DB.Where("username = ?", item.Username).First(&user)
	user.Password = item.Password
	user.Date = item.Date
	user.Name = item.Name
	user.Nfc = item.Nfc
	mydb.DB.Save(&user)
}

func GetUseryf(item *Useryf) bool {
	var user Useryf
	//mydb.DB.First(&user, item.Username)

	//result := mydb.DB.First(&user, item.Username).Error
	result := mydb.DB.Where("username = ?", item.Username).First(&user).Error
	fmt.Println("-----user---->>")
	fmt.Println(user)

	if errors.Is(result, gorm.ErrRecordNotFound) {
		// 记录不存在
		fmt.Println("用户不存在")
	} else if result == nil {
		// 记录存在
		fmt.Println("用户存在")
		return true
	} else {
		// 其他错误
		fmt.Println("查询出错")
	}
	return false
}