39 lines
719 B
Go
39 lines
719 B
Go
package db
|
|
|
|
import "gorm.io/gorm"
|
|
|
|
type User struct {
|
|
gorm.Model
|
|
Name string `gorm:"not null"`
|
|
Token string `gorm:"unique;not null"`
|
|
}
|
|
|
|
// 添加用户
|
|
func AddUser(name, token string) error {
|
|
user := &User{Name: name, Token: token}
|
|
result := db.Create(&user)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 删除用户
|
|
func DeleteUser(id uint) error {
|
|
result := db.Delete(&User{}, id)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 修改用户
|
|
func UpdateUser(id uint, name, token string) error {
|
|
user := &User{Name: name, Token: token}
|
|
result := db.Model(&User{}).Where("id = ?", id).Updates(user)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|