mirror of
https://github.com/krau/SaveAny-Bot.git
synced 2026-05-10 17:52:44 +08:00
23 lines
456 B
Go
23 lines
456 B
Go
package dao
|
|
|
|
import (
|
|
"github.com/krau/SaveAny-Bot/model"
|
|
)
|
|
|
|
func CreateUser(userID int64) error {
|
|
if _, err := GetUserByUserID(userID); err == nil {
|
|
return nil
|
|
}
|
|
return db.Create(&model.User{UserID: userID}).Error
|
|
}
|
|
|
|
func GetUserByUserID(userID int64) (*model.User, error) {
|
|
var user model.User
|
|
err := db.Where("user_id = ?", userID).First(&user).Error
|
|
return &user, err
|
|
}
|
|
|
|
func UpdateUser(user *model.User) error {
|
|
return db.Save(user).Error
|
|
}
|