88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"enaklo-pos-be/internal/common/logger"
|
|
"enaklo-pos-be/internal/common/mycontext"
|
|
"enaklo-pos-be/internal/entity"
|
|
"fmt"
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UndianRepo interface {
|
|
GetUndianEventByID(ctx context.Context, id int64) (*entity.UndianEventDB, error)
|
|
GetActiveUndianEvents(ctx context.Context) ([]*entity.UndianEventDB, error)
|
|
CreateUndianVouchers(ctx context.Context, vouchers []*entity.UndianVoucherDB) error
|
|
GetNextVoucherSequence(ctx mycontext.Context) (int64, error)
|
|
GetNextVoucherSequenceBatch(ctx mycontext.Context, count int) (int64, error)
|
|
}
|
|
|
|
type undianRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewUndianRepository(db *gorm.DB) *undianRepository {
|
|
return &undianRepository{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *undianRepository) GetUndianEventByID(ctx context.Context, id int64) (*entity.UndianEventDB, error) {
|
|
event := new(entity.UndianEventDB)
|
|
if err := r.db.WithContext(ctx).First(event, id).Error; err != nil {
|
|
logger.ContextLogger(ctx).Error("error when get undian event by id", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
return event, nil
|
|
}
|
|
|
|
func (r *undianRepository) GetActiveUndianEvents(ctx context.Context) ([]*entity.UndianEventDB, error) {
|
|
var events []*entity.UndianEventDB
|
|
now := time.Now()
|
|
|
|
if err := r.db.WithContext(ctx).
|
|
Where("status = 'active' AND start_date <= ? AND end_date >= ?", now, now).
|
|
Find(&events).Error; err != nil {
|
|
logger.ContextLogger(ctx).Error("error when get active undian events", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
return events, nil
|
|
}
|
|
|
|
func (r *undianRepository) CreateUndianVouchers(ctx context.Context, vouchers []*entity.UndianVoucherDB) error {
|
|
err := r.db.WithContext(ctx).Create(&vouchers).Error
|
|
if err != nil {
|
|
logger.ContextLogger(ctx).Error("error when create undian vouchers", zap.Error(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *undianRepository) GetNextVoucherSequence(ctx mycontext.Context) (int64, error) {
|
|
var sequence int64
|
|
|
|
err := r.db.WithContext(ctx).Raw("SELECT nextval('voucher_sequence')").Scan(&sequence).Error
|
|
if err != nil {
|
|
logger.ContextLogger(ctx).Error("error when get next voucher sequence", zap.Error(err))
|
|
return 0, err
|
|
}
|
|
|
|
return sequence, nil
|
|
}
|
|
|
|
func (r *undianRepository) GetNextVoucherSequenceBatch(ctx mycontext.Context, count int) (int64, error) {
|
|
var startSequence int64
|
|
|
|
query := fmt.Sprintf("SELECT nextval('voucher_sequence') + %d - %d", count-1, count-1)
|
|
err := r.db.WithContext(ctx).Raw(query).Scan(&startSequence).Error
|
|
if err != nil {
|
|
logger.ContextLogger(ctx).Error("error when get batch voucher sequence", zap.Error(err))
|
|
return 0, err
|
|
}
|
|
|
|
return startSequence, nil
|
|
} |