64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
|
package repository
|
||
|
|
|
||
|
|
import (
|
||
|
|
"carrotskin/internal/model"
|
||
|
|
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// clientRepository ClientRepository的实现
|
||
|
|
type clientRepository struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewClientRepository 创建ClientRepository实例
|
||
|
|
func NewClientRepository(db *gorm.DB) ClientRepository {
|
||
|
|
return &clientRepository{db: db}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) Create(client *model.Client) error {
|
||
|
|
return r.db.Create(client).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) FindByClientToken(clientToken string) (*model.Client, error) {
|
||
|
|
var client model.Client
|
||
|
|
err := r.db.Where("client_token = ?", clientToken).First(&client).Error
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &client, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) FindByUUID(uuid string) (*model.Client, error) {
|
||
|
|
var client model.Client
|
||
|
|
err := r.db.Where("uuid = ?", uuid).First(&client).Error
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &client, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) FindByUserID(userID int64) ([]*model.Client, error) {
|
||
|
|
var clients []*model.Client
|
||
|
|
err := r.db.Where("user_id = ?", userID).Find(&clients).Error
|
||
|
|
return clients, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) Update(client *model.Client) error {
|
||
|
|
return r.db.Save(client).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) IncrementVersion(clientUUID string) error {
|
||
|
|
return r.db.Model(&model.Client{}).
|
||
|
|
Where("uuid = ?", clientUUID).
|
||
|
|
Update("version", gorm.Expr("version + 1")).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) DeleteByClientToken(clientToken string) error {
|
||
|
|
return r.db.Where("client_token = ?", clientToken).Delete(&model.Client{}).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientRepository) DeleteByUserID(userID int64) error {
|
||
|
|
return r.db.Where("user_id = ?", userID).Delete(&model.Client{}).Error
|
||
|
|
}
|