24 lines
483 B
Go
24 lines
483 B
Go
package services
|
|
|
|
import (
|
|
"CatsOfMastodonBotGo/internal/models"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UserService struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewUserService(db *gorm.DB) *UserService {
|
|
return &UserService{db: db}
|
|
}
|
|
|
|
func (us *UserService) CreateUser(user models.ComUser) int {
|
|
return int(us.db.Create(&user).RowsAffected)
|
|
}
|
|
|
|
func (us *UserService) GetUserByUsername(username string) models.ComUser {
|
|
var user models.ComUser
|
|
us.db.Where("username = ?", username).First(&user)
|
|
return user
|
|
} |