75 lines
2.2 KiB
Go
75 lines
2.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/Caknoooo/go-gin-clean-starter/database/entities"
|
|
"github.com/Caknoooo/go-gin-clean-starter/modules/aisle/query"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type AisleRepository interface {
|
|
Create(ctx context.Context, tx *gorm.DB, aisle entities.MAisleEntity) (entities.MAisleEntity, error)
|
|
GetById(ctx context.Context, tx *gorm.DB, aisleId string) (entities.MAisleEntity, error)
|
|
GetAll(ctx context.Context, filter query.AisleFilter) ([]entities.MAisleEntity, int64, error)
|
|
Update(ctx context.Context, tx *gorm.DB, aisle entities.MAisleEntity) (entities.MAisleEntity, error)
|
|
Delete(ctx context.Context, tx *gorm.DB, aisleId string) error
|
|
}
|
|
|
|
type aisleRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewAisleRepository(db *gorm.DB) AisleRepository {
|
|
return &aisleRepository{db: db}
|
|
}
|
|
|
|
func (r *aisleRepository) Create(ctx context.Context, tx *gorm.DB, aisle entities.MAisleEntity) (entities.MAisleEntity, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
if err := tx.WithContext(ctx).Create(&aisle).Error; err != nil {
|
|
return aisle, err
|
|
}
|
|
return aisle, nil
|
|
}
|
|
|
|
func (r *aisleRepository) GetById(ctx context.Context, tx *gorm.DB, aisleId string) (entities.MAisleEntity, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
var aisle entities.MAisleEntity
|
|
if err := tx.WithContext(ctx).Preload("Zona").Preload("DimUom").Preload("WeightUom").Preload("Client").First(&aisle, "id = ?", aisleId).Error; err != nil {
|
|
return aisle, err
|
|
}
|
|
return aisle, nil
|
|
}
|
|
|
|
func (r *aisleRepository) GetAll(ctx context.Context, filter query.AisleFilter) ([]entities.MAisleEntity, int64, error) {
|
|
var aisles []entities.MAisleEntity
|
|
var total int64
|
|
db := query.ApplyAisleFilters(r.db, filter)
|
|
db.Find(&aisles).Count(&total)
|
|
return aisles, total, nil
|
|
}
|
|
|
|
func (r *aisleRepository) Update(ctx context.Context, tx *gorm.DB, aisle entities.MAisleEntity) (entities.MAisleEntity, error) {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
if err := tx.WithContext(ctx).Save(&aisle).Error; err != nil {
|
|
return aisle, err
|
|
}
|
|
return aisle, nil
|
|
}
|
|
|
|
func (r *aisleRepository) Delete(ctx context.Context, tx *gorm.DB, aisleId string) error {
|
|
if tx == nil {
|
|
tx = r.db
|
|
}
|
|
if err := tx.WithContext(ctx).Delete(&entities.MAisleEntity{}, "id = ?", aisleId).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|