huanghaosheng/VLoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: change all user to account

Browse Source
This commit is contained in:
Leon
2025-12-05 14:25:39 +08:00
parent 311a27523f
commit c5331a459f
8 changed files with 95 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
internal/account/service.go
View File

@@ -7,94 +7,94 @@ import (
"golang.org/x/crypto/bcrypt"
)
type UserService struct {
userRepository *UserRepository
type AccountService struct {
accountRepository *AccountRepository
}
func NewUserService(userRepository *UserRepository) *UserService {
return &UserService{userRepository: userRepository}
func NewAccountService(accountRepository *AccountRepository) *AccountService {
return &AccountService{accountRepository: accountRepository}
}
func (us *UserService) CreateUser(user *User) error {
passwordHash, err := bcrypt.GenerateFromPassword([]byte(user.Password), bcrypt.DefaultCost)
func (as *AccountService) CreateAccount(account *Account) error {
passwordHash, err := bcrypt.GenerateFromPassword([]byte(account.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
user.Password = string(passwordHash)
if err := us.userRepository.CreateUser(user); err != nil {
account.Password = string(passwordHash)
if err := as.accountRepository.CreateAccount(account); err != nil {
return err
}
return nil
}
func (us *UserService) RenameByID(id uint, newUsername string) error {
if err := us.userRepository.RenameByID(id, newUsername); err != nil {
func (as *AccountService) RenameByID(id uint, newUsername string) error {
if err := as.accountRepository.RenameByID(id, newUsername); err != nil {
return err
}
return nil
}
func (us *UserService) ChangePassword(username, oldPassword, newPassword string) error {
user, err := us.FindByUsername(username)
func (as *AccountService) ChangePassword(username, oldPassword, newPassword string) error {
account, err := as.FindByUsername(username)
if err != nil {
return err
}
if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(oldPassword)); err != nil {
if err := bcrypt.CompareHashAndPassword([]byte(account.Password), []byte(oldPassword)); err != nil {
return err
}
passwordHash, err := bcrypt.GenerateFromPassword([]byte(newPassword), bcrypt.DefaultCost)
if err != nil {
return err
}
if err := us.userRepository.ChangePassword(user.ID, string(passwordHash)); err != nil {
if err := as.accountRepository.ChangePassword(account.ID, string(passwordHash)); err != nil {
return err
}
return nil
}
func (us *UserService) FindByID(id uint) (*User, error) {
if user, err := us.userRepository.FindByID(id); err != nil {
func (as *AccountService) FindByID(id uint) (*Account, error) {
if account, err := as.accountRepository.FindByID(id); err != nil {
return nil, err
} else {
return user, nil
return account, nil
}
}
func (us *UserService) FindByUsername(username string) (*User, error) {
if user, err := us.userRepository.FindByUsername(username); err != nil {
func (as *AccountService) FindByUsername(username string) (*Account, error) {
if account, err := as.accountRepository.FindByUsername(username); err != nil {
return nil, err
} else {
return user, nil
return account, nil
}
}
func (us *UserService) Login(username, password string) (string, error) {
user, err := us.FindByUsername(username)
func (as *AccountService) Login(username, password string) (string, error) {
account, err := as.FindByUsername(username)
if err != nil {
return "", err
}
if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password)); err != nil {
if err := bcrypt.CompareHashAndPassword([]byte(account.Password), []byte(password)); err != nil {
return "", err
}
// generate token
token, err := auth.GenerateToken(user.ID, user.Username)
token, err := auth.GenerateToken(account.ID, account.Username)
if err != nil {
return "", err
}
if err := us.userRepository.Login(user.ID, token); err != nil {
if err := as.accountRepository.Login(account.ID, token); err != nil {
return "", err
}
return token, nil
}
func (us *UserService) Logout(userID uint) error {
user, err := us.FindByID(userID)
func (as *AccountService) Logout(accountID uint) error {
account, err := as.FindByID(accountID)
if err != nil {
return err
}
if user.Token == "" {
return errors.New("user already logged out")
if account.Token == "" {
return errors.New("account already logged out")
}
return us.userRepository.Logout(user.ID, user.Token)
return as.accountRepository.Logout(account.ID, account.Token)
}