81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
package db
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
type User struct {
|
|
IsDelete bool `gorm:"default:false" json:"IsDelete"`
|
|
ID uint `gorm:"primarykey" json:"id,omitempty"`
|
|
Name string `gorm:"unique;not null" json:"name,omitempty"`
|
|
Token string `gorm:"unique;not null" json:"token,omitempty"`
|
|
CreatedAt time.Time `json:"createdAt,omitempty"`
|
|
UpdatedAt time.Time `json:"updatedAt,omitempty"`
|
|
TotalTokens int `json:"total_tokens,omitempty"`
|
|
// DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
|
|
}
|
|
|
|
func CreateUser(u *User) error {
|
|
result := db.Create(u)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 添加用户
|
|
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, token string) error {
|
|
user := &User{Token: token}
|
|
result := db.Model(&User{}).Where("id = ?", id).Updates(user)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetUserByID(id uint) (*User, error) {
|
|
var user User
|
|
result := db.Where("id = ?", id).First(&user)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func GetUserByName(name string) (*User, error) {
|
|
var user User
|
|
result := db.Where(&User{Name: name}).First(&user)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func GetAllUsers() ([]*User, error) {
|
|
var users []*User
|
|
result := db.Find(&users)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return users, nil
|
|
}
|