83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"enaklo-pos-be/internal/common/mycontext"
|
|
"enaklo-pos-be/internal/entity"
|
|
"enaklo-pos-be/internal/repository/models"
|
|
"github.com/pkg/errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ProductRepository interface {
|
|
GetProductsByIDs(ctx mycontext.Context, ids []int64, partnerID int64) ([]*entity.Product, error)
|
|
GetProductDetails(ctx mycontext.Context, productIDs []int64, partnerID int64) (*entity.ProductDetails, error)
|
|
}
|
|
|
|
type productRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewproductRepository(db *gorm.DB) *productRepository {
|
|
return &productRepository{db: db}
|
|
}
|
|
|
|
func (r *productRepository) GetProductsByIDs(ctx mycontext.Context, ids []int64, partnerID int64) ([]*entity.Product, error) {
|
|
if len(ids) == 0 {
|
|
return []*entity.Product{}, nil
|
|
}
|
|
|
|
var productsDB []models.ProductDB
|
|
|
|
if err := r.db.Where("id IN ? AND partner_id = ?", ids, partnerID).Find(&productsDB).Error; err != nil {
|
|
return nil, errors.Wrap(err, "failed to find products")
|
|
}
|
|
|
|
products := make([]*entity.Product, 0, len(productsDB))
|
|
for i := range productsDB {
|
|
product := r.toDomainProductModel(&productsDB[i])
|
|
products = append(products, product)
|
|
}
|
|
|
|
return products, nil
|
|
}
|
|
|
|
func (r *productRepository) GetProductDetails(ctx mycontext.Context, productIDs []int64, partnerID int64) (*entity.ProductDetails, error) {
|
|
if len(productIDs) == 0 {
|
|
return &entity.ProductDetails{
|
|
Products: make(map[int64]*entity.Product),
|
|
}, nil
|
|
}
|
|
|
|
var productsDB []models.ProductDB
|
|
|
|
if err := r.db.Where("id IN ? AND partner_id = ?", productIDs, partnerID).Find(&productsDB).Error; err != nil {
|
|
return nil, errors.Wrap(err, "failed to find products")
|
|
}
|
|
|
|
productMap := make(map[int64]*entity.Product, len(productsDB))
|
|
|
|
for i := range productsDB {
|
|
product := r.toDomainProductModel(&productsDB[i])
|
|
productMap[product.ID] = product
|
|
}
|
|
|
|
return &entity.ProductDetails{
|
|
Products: productMap,
|
|
PartnerID: partnerID,
|
|
}, nil
|
|
}
|
|
|
|
func (r *productRepository) toDomainProductModel(dbModel *models.ProductDB) *entity.Product {
|
|
return &entity.Product{
|
|
ID: dbModel.ID,
|
|
PartnerID: dbModel.PartnerID,
|
|
Name: dbModel.Name,
|
|
Description: dbModel.Description,
|
|
Price: dbModel.Price,
|
|
Type: dbModel.Type,
|
|
Status: dbModel.Status,
|
|
CreatedAt: dbModel.CreatedAt,
|
|
UpdatedAt: dbModel.UpdatedAt,
|
|
}
|
|
}
|