Add callback and update user role
This commit is contained in:
@@ -28,14 +28,23 @@ func (r *WalletRepository) Create(ctx context.Context, tx *gorm.DB, wallet *enti
|
||||
return wallet, nil
|
||||
}
|
||||
|
||||
func (r *WalletRepository) Update(ctx context.Context, wallet *entity.Wallet) (*entity.Wallet, error) {
|
||||
if err := r.db.Save(wallet).Error; err != nil {
|
||||
func (r *WalletRepository) Update(ctx context.Context, db *gorm.DB, wallet *entity.Wallet) (*entity.Wallet, error) {
|
||||
if err := db.Save(wallet).Error; err != nil {
|
||||
logger.ContextLogger(ctx).Error("error when updating wallet", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
return wallet, nil
|
||||
}
|
||||
|
||||
func (r *WalletRepository) GetByPartnerID(ctx context.Context, db *gorm.DB, partnerID int64) (*entity.Wallet, error) {
|
||||
wallet := new(entity.Wallet)
|
||||
if err := db.WithContext(ctx).Where("partner_id = ?", partnerID).First(wallet).Error; err != nil {
|
||||
logger.ContextLogger(ctx).Error("error when finding wallet by partner ID", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
return wallet, nil
|
||||
}
|
||||
|
||||
func (r *WalletRepository) GetByID(ctx context.Context, id int64) (*entity.Wallet, error) {
|
||||
wallet := new(entity.Wallet)
|
||||
if err := r.db.First(wallet, id).Error; err != nil {
|
||||
|
||||
Reference in New Issue
Block a user