change db config
parent
e0dccc3464
commit
795f83aff5
@ -0,0 +1,71 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"go_mqtt/mydb"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type Useryf struct {
|
||||
Username string `gorm:"primaryKey;size:255"`
|
||||
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)
|
||||
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
|
||||
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
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue