Add callback and update user role

internal/repository/payment/payment.go

@@ -38,6 +38,14 @@ func (r *PaymentRepository) Update(ctx context.Context, payment *entity.Payment)
 	return payment, nil
 }
 
+func (r *PaymentRepository) UpdateWithTx(ctx context.Context, tx *gorm.DB, payment *entity.Payment) (*entity.Payment, error) {
+	if err := tx.WithContext(ctx).Save(payment).Error; err != nil {
+		logger.ContextLogger(ctx).Error("error when updating payment", zap.Error(err))
+		return nil, err
+	}
+	return payment, nil
+}
+
 // FindByID retrieves a payment record by its ID
 func (r *PaymentRepository) FindByID(ctx context.Context, id uuid.UUID) (*entity.Payment, error) {
 	payment := new(entity.Payment)
@@ -62,9 +70,9 @@ func (r *PaymentRepository) FindByOrderAndPartnerID(ctx context.Context, orderID
 }
 
 // FindByReferenceID retrieves a payment record by its reference ID
-func (r *PaymentRepository) FindByReferenceID(ctx context.Context, referenceID string) (*entity.Payment, error) {
+func (r *PaymentRepository) FindByReferenceID(ctx context.Context, db *gorm.DB, referenceID string) (*entity.Payment, error) {
 	payment := new(entity.Payment)
-	if err := r.db.WithContext(ctx).Where("reference_id = ?", referenceID).First(payment).Error; err != nil {
+	if err := db.WithContext(ctx).Where("reference_id = ?", referenceID).First(payment).Error; err != nil {
 		logger.ContextLogger(ctx).Error("error when finding payment by reference ID", zap.Error(err))
 		return nil, err
 	}