Add User Partner and Product

internal/repository/wallet/wallet.go

@@ -0,0 +1,65 @@
+package repository
+
+import (
+	"context"
+	"furtuna-be/internal/common/logger"
+	"furtuna-be/internal/entity"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+type WalletRepository struct {
+	db *gorm.DB
+}
+
+func NewWalletRepository(db *gorm.DB) *WalletRepository {
+	return &WalletRepository{
+		db: db,
+	}
+}
+
+func (r *WalletRepository) Create(ctx context.Context, tx *gorm.DB, wallet *entity.Wallet) (*entity.Wallet, error) {
+	err := tx.Create(wallet).Error
+	if err != nil {
+		logger.ContextLogger(ctx).Error("error when creating wallet", zap.Error(err))
+		return nil, err
+	}
+	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 {
+		logger.ContextLogger(ctx).Error("error when updating wallet", 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 {
+		logger.ContextLogger(ctx).Error("error when getting wallet by id", zap.Error(err))
+		return nil, err
+	}
+	return wallet, nil
+}
+
+func (r *WalletRepository) GetAll(ctx context.Context) ([]*entity.Wallet, error) {
+	var wallets []*entity.Wallet
+	if err := r.db.Find(&wallets).Error; err != nil {
+		logger.ContextLogger(ctx).Error("error when getting all wallets", zap.Error(err))
+		return nil, err
+	}
+	return wallets, nil
+}
+
+func (r *WalletRepository) Delete(ctx context.Context, id int64) error {
+	wallet := new(entity.Wallet)
+	wallet.ID = id
+	if err := r.db.Delete(wallet).Error; err != nil {
+		logger.ContextLogger(ctx).Error("error when deleting wallet", zap.Error(err))
+		return err
+	}
+	return nil
+}