144 lines
4.2 KiB
Go
144 lines
4.2 KiB
Go
package partners
|
|
|
|
import (
|
|
"context"
|
|
"furtuna-be/internal/common/logger"
|
|
"furtuna-be/internal/entity"
|
|
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type PartnerRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewPartnerRepository(db *gorm.DB) *PartnerRepository {
|
|
return &PartnerRepository{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (b *PartnerRepository) Create(ctx context.Context, Partner *entity.PartnerDB) (*entity.PartnerDB, error) {
|
|
err := b.db.Create(Partner).Error
|
|
if err != nil {
|
|
logger.ContextLogger(ctx).Error("error when create Partner", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
return Partner, nil
|
|
}
|
|
|
|
func (b *PartnerRepository) CreateWithTx(ctx context.Context, tx *gorm.DB, Partner *entity.PartnerDB) (*entity.PartnerDB, error) {
|
|
err := tx.Create(Partner).Error
|
|
if err != nil {
|
|
logger.ContextLogger(ctx).Error("error when create Partner", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
return Partner, nil
|
|
}
|
|
|
|
func (b *PartnerRepository) Update(ctx context.Context, Partner *entity.PartnerDB) (*entity.PartnerDB, error) {
|
|
if err := b.db.Save(Partner).Error; err != nil {
|
|
logger.ContextLogger(ctx).Error("error when update Partner", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
return Partner, nil
|
|
}
|
|
|
|
func (b *PartnerRepository) UpdateWithTx(ctx context.Context, tx *gorm.DB, Partner *entity.PartnerDB) (*entity.PartnerDB, error) {
|
|
if err := tx.Save(Partner).Error; err != nil {
|
|
logger.ContextLogger(ctx).Error("error when update Partner", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
return Partner, nil
|
|
}
|
|
|
|
func (b *PartnerRepository) GetByID(ctx context.Context, id int64) (*entity.PartnerDB, error) {
|
|
Partner := new(entity.PartnerDBSearch)
|
|
|
|
query := b.db.Table("partners p").
|
|
Select("p.*, w.balance as wallet_balance, u.name as admin_name, u.email as admin_email, u.phone_number as admin_phone_number").
|
|
Joins("LEFT JOIN wallets w ON w.partner_id = p.id").
|
|
Joins("LEFT JOIN users u ON p.admin_user_id = u.id").
|
|
Where("p.id = ? AND p.deleted_at IS NULL", id)
|
|
|
|
if err := query.First(&Partner).Error; err != nil {
|
|
logger.ContextLogger(ctx).Error("error when get by id Partner", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
Partner.Partner.Balance = Partner.WalletBalance
|
|
Partner.Partner.AdminName = Partner.AdminName
|
|
Partner.Partner.AdminPhoneNumber = Partner.AdminPhoneNumber
|
|
Partner.Partner.AdminEmail = Partner.AdminEmail
|
|
|
|
return &entity.PartnerDB{Partner.Partner}, nil
|
|
}
|
|
|
|
func (b *PartnerRepository) GetAll(ctx context.Context, req entity.PartnerSearch) (entity.PartnerList, int, error) {
|
|
var partners []*entity.PartnerDBSearch
|
|
var total int64
|
|
|
|
query := b.db.Table("partners p").
|
|
Select("p.*, w.balance as wallet_balance, u.name as admin_name, u.email as admin_email, u.phone_number as admin_phone_number").
|
|
Joins("LEFT JOIN wallets w ON w.partner_id = p.id").
|
|
Joins("LEFT JOIN users u ON p.admin_user_id = u.id").
|
|
Where("p.deleted_at IS NULL")
|
|
|
|
if req.Search != "" {
|
|
query = query.Where("p.name ILIKE ?", "%"+req.Search+"%")
|
|
}
|
|
|
|
if req.Name != "" {
|
|
query = query.Where("p.name ILIKE ?", "%"+req.Name+"%")
|
|
}
|
|
|
|
if req.PartnerID != nil {
|
|
query = query.Where("p.id = ?", req.PartnerID)
|
|
}
|
|
|
|
if req.Limit > 0 {
|
|
query = query.Limit(req.Limit)
|
|
}
|
|
|
|
if req.Offset > 0 {
|
|
query = query.Offset(req.Offset)
|
|
}
|
|
|
|
// Find partners with joined wallet balances
|
|
if err := query.Find(&partners).Error; err != nil {
|
|
logger.ContextLogger(ctx).Error("error when getting all partners", zap.Error(err))
|
|
return nil, 0, err
|
|
}
|
|
|
|
// Counting total records
|
|
if err := b.db.Table("partners p").
|
|
Joins("LEFT JOIN wallets w ON w.partner_id = p.id").
|
|
Where("p.deleted_at IS NULL").
|
|
Count(&total).Error; err != nil {
|
|
logger.ContextLogger(ctx).Error("error when counting partners", zap.Error(err))
|
|
return nil, 0, err
|
|
}
|
|
|
|
partnersSearchResult := entity.PartnerList{}
|
|
|
|
for _, partner := range partners {
|
|
partner.Balance = partner.WalletBalance
|
|
partner.Partner.AdminName = partner.AdminName
|
|
partner.Partner.AdminPhoneNumber = partner.AdminPhoneNumber
|
|
partner.Partner.AdminEmail = partner.AdminEmail
|
|
partnersSearchResult = append(partnersSearchResult, &entity.PartnerDB{partner.Partner})
|
|
}
|
|
|
|
return partnersSearchResult, int(total), nil
|
|
|
|
}
|
|
func (b *PartnerRepository) Delete(ctx context.Context, id int64) error {
|
|
Partner := new(entity.PartnerDB)
|
|
Partner.ID = id
|
|
if err := b.db.Delete(Partner).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|