96 lines
3.0 KiB
Go
96 lines
3.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/Caknoooo/go-gin-clean-starter/modules/sequence/repository"
|
|
pkgdto "github.com/Caknoooo/go-gin-clean-starter/pkg/dto"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type SequenceService interface {
|
|
GenerateNumber(ctx context.Context, clientId string, config pkgdto.SequenceConfig) (string, error)
|
|
GenerateNumberWithPeriod(ctx context.Context, clientId string, config pkgdto.SequenceConfig, suffix string) (string, error)
|
|
GenerateNumberWithSuffix(ctx context.Context, clientId string, config pkgdto.SequenceConfig, suffix string) (string, error)
|
|
GetNextSequence(ctx context.Context, clientId string, config pkgdto.SequenceConfig) (int, error)
|
|
}
|
|
|
|
type sequenceService struct {
|
|
db *gorm.DB
|
|
sequenceRepo repository.SequenceRepository
|
|
}
|
|
|
|
func NewSequenceService(sequenceRepo repository.SequenceRepository, db *gorm.DB) SequenceService {
|
|
return &sequenceService{
|
|
sequenceRepo: sequenceRepo,
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
// GenerateNumberWithSuffix implements SequenceService.
|
|
func (s *sequenceService) GenerateNumberWithSuffix(ctx context.Context, clientId string, config pkgdto.SequenceConfig, suffix string) (string, error) {
|
|
period := config.Period
|
|
seq, err := s.sequenceRepo.GetOrCreateSequence(ctx, s.db, clientId, config.EntityType, period)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
seq.CurrentSeq++
|
|
if err := s.sequenceRepo.UpdateSequence(ctx, s.db, seq); err != nil {
|
|
return "", err
|
|
}
|
|
refNum := fmt.Sprintf("%s-%s-%04d", config.Prefix, suffix, seq.CurrentSeq)
|
|
return refNum, nil
|
|
}
|
|
|
|
func (s *sequenceService) GenerateNumber(ctx context.Context, clientId string, config pkgdto.SequenceConfig) (string, error) {
|
|
period := config.Period
|
|
if period == "" {
|
|
period = "-"
|
|
}
|
|
seq, err := s.sequenceRepo.GetOrCreateSequence(ctx, s.db, clientId, config.EntityType, period)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
seq.CurrentSeq++
|
|
if err := s.sequenceRepo.UpdateSequence(ctx, s.db, seq); err != nil {
|
|
return "", err
|
|
}
|
|
refNum := fmt.Sprintf("%s-%04d", config.Prefix, seq.CurrentSeq)
|
|
return refNum, nil
|
|
}
|
|
|
|
func (s *sequenceService) GenerateNumberWithPeriod(ctx context.Context, clientId string, config pkgdto.SequenceConfig, suffix string) (string, error) {
|
|
if config.Period == "" {
|
|
config.Period = time.Now().Format("0601")
|
|
}
|
|
period := config.Period
|
|
seq, err := s.sequenceRepo.GetOrCreateSequence(ctx, s.db, clientId, config.EntityType, period)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
seq.CurrentSeq++
|
|
if err := s.sequenceRepo.UpdateSequence(ctx, s.db, seq); err != nil {
|
|
return "", err
|
|
}
|
|
refNum := fmt.Sprintf("%s-%s-%s-%04d", config.Prefix, suffix, period, seq.CurrentSeq)
|
|
return refNum, nil
|
|
}
|
|
|
|
func (s *sequenceService) GetNextSequence(ctx context.Context, clientId string, config pkgdto.SequenceConfig) (int, error) {
|
|
period := config.Period
|
|
if period == "" {
|
|
period = "-"
|
|
}
|
|
seq, err := s.sequenceRepo.GetOrCreateSequence(ctx, s.db, clientId, config.EntityType, period)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
seq.CurrentSeq++
|
|
if err := s.sequenceRepo.UpdateSequence(ctx, s.db, seq); err != nil {
|
|
return 0, err
|
|
}
|
|
return seq.CurrentSeq, nil
|
|
}
|