mirror of
https://gitlab.com/mbugroup/lti-api.git
synced 2026-05-20 13:31:56 +00:00
fix: adjusment module depletion, chickin, recording refactor
This commit is contained in:
Hafizh A. Y
@@ -0,0 +1,381 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
commonSvc "gitlab.com/mbugroup/lti-api.git/internal/common/service"
|
||||
"gitlab.com/mbugroup/lti-api.git/internal/config"
|
||||
"gitlab.com/mbugroup/lti-api.git/internal/database"
|
||||
entity "gitlab.com/mbugroup/lti-api.git/internal/entities"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type productWarehouseScopeRow struct {
|
||||
ProductWarehouseID uint `gorm:"column:product_warehouse_id"`
|
||||
ProductID uint `gorm:"column:product_id"`
|
||||
WarehouseID uint `gorm:"column:warehouse_id"`
|
||||
ProjectFlockKandangID *uint `gorm:"column:project_flock_kandang_id"`
|
||||
}
|
||||
|
||||
type reflowTarget struct {
|
||||
ProductWarehouseID uint
|
||||
ProductID uint
|
||||
WarehouseID uint
|
||||
ProjectFlockKandangID *uint
|
||||
FlagGroupCode string
|
||||
}
|
||||
|
||||
func main() {
|
||||
var (
|
||||
projectFlockKandangID uint
|
||||
apply bool
|
||||
asOfRaw string
|
||||
includeShared bool
|
||||
)
|
||||
|
||||
flag.UintVar(&projectFlockKandangID, "project-flock-kandang-id", 0, "Project flock kandang ID (required)")
|
||||
flag.BoolVar(&apply, "apply", false, "Apply reflow. If false, run as dry-run")
|
||||
flag.StringVar(&asOfRaw, "as-of", "", "Optional AsOf boundary. Format: RFC3339 or YYYY-MM-DD")
|
||||
flag.BoolVar(&includeShared, "include-shared", true, "Include product warehouses referenced by transactions in this PFK scope (including shared/non-bound product warehouses)")
|
||||
flag.Parse()
|
||||
|
||||
if projectFlockKandangID == 0 {
|
||||
log.Fatal("--project-flock-kandang-id is required")
|
||||
}
|
||||
|
||||
asOf, err := parseAsOf(asOfRaw)
|
||||
if err != nil {
|
||||
log.Fatalf("invalid --as-of: %v", err)
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
db := database.Connect(config.DBHost, config.DBName)
|
||||
fifoStockV2Svc := commonSvc.NewFifoStockV2Service(db, nil)
|
||||
|
||||
exists, err := projectFlockKandangExists(ctx, db, projectFlockKandangID)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to check project flock kandang: %v", err)
|
||||
}
|
||||
if !exists {
|
||||
log.Fatalf("project_flock_kandang_id %d not found", projectFlockKandangID)
|
||||
}
|
||||
|
||||
scopedPWs, err := loadScopedProductWarehouses(ctx, db, projectFlockKandangID, includeShared)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to load scoped product warehouses: %v", err)
|
||||
}
|
||||
if len(scopedPWs) == 0 {
|
||||
fmt.Printf("Mode: %s\n", modeLabel(apply))
|
||||
fmt.Printf("Scope: project_flock_kandang_id=%d\n", projectFlockKandangID)
|
||||
fmt.Println("No product warehouse found in scope")
|
||||
return
|
||||
}
|
||||
|
||||
targets := make([]reflowTarget, 0, len(scopedPWs))
|
||||
skippedPW := 0
|
||||
failedResolve := 0
|
||||
|
||||
for _, pw := range scopedPWs {
|
||||
flagGroups, err := resolveFlagGroupsByProductWarehouse(ctx, db, pw.ProductWarehouseID)
|
||||
if err != nil {
|
||||
fmt.Printf("FAIL pw=%d error=resolve flag groups: %v\n", pw.ProductWarehouseID, err)
|
||||
failedResolve++
|
||||
continue
|
||||
}
|
||||
if len(flagGroups) == 0 {
|
||||
fmt.Printf("SKIP pw=%d reason=no active fifo v2 route by product flag\n", pw.ProductWarehouseID)
|
||||
skippedPW++
|
||||
continue
|
||||
}
|
||||
for _, group := range flagGroups {
|
||||
targets = append(targets, reflowTarget{
|
||||
ProductWarehouseID: pw.ProductWarehouseID,
|
||||
ProductID: pw.ProductID,
|
||||
WarehouseID: pw.WarehouseID,
|
||||
ProjectFlockKandangID: pw.ProjectFlockKandangID,
|
||||
FlagGroupCode: group,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
sort.Slice(targets, func(i, j int) bool {
|
||||
if targets[i].ProductWarehouseID == targets[j].ProductWarehouseID {
|
||||
return targets[i].FlagGroupCode < targets[j].FlagGroupCode
|
||||
}
|
||||
return targets[i].ProductWarehouseID < targets[j].ProductWarehouseID
|
||||
})
|
||||
|
||||
fmt.Printf("Mode: %s\n", modeLabel(apply))
|
||||
fmt.Printf("Scope: project_flock_kandang_id=%d include_shared=%t\n", projectFlockKandangID, includeShared)
|
||||
if asOf != nil {
|
||||
fmt.Printf("AsOf: %s\n", asOf.UTC().Format(time.RFC3339))
|
||||
} else {
|
||||
fmt.Println("AsOf: <nil> (full timeline)")
|
||||
}
|
||||
fmt.Printf("Product warehouses in scope: %d\n", len(scopedPWs))
|
||||
fmt.Printf("Planned reflow targets: %d\n\n", len(targets))
|
||||
|
||||
for _, target := range targets {
|
||||
fmt.Printf(
|
||||
"PLAN pw=%d product=%d warehouse=%d pw_pfk=%s flag_group=%s\n",
|
||||
target.ProductWarehouseID,
|
||||
target.ProductID,
|
||||
target.WarehouseID,
|
||||
displayOptionalUint(target.ProjectFlockKandangID),
|
||||
target.FlagGroupCode,
|
||||
)
|
||||
}
|
||||
|
||||
if !apply {
|
||||
fmt.Println()
|
||||
fmt.Printf("Summary: planned=%d skipped_pw=%d failed_resolve=%d applied=0 failed_apply=0\n", len(targets), skippedPW, failedResolve)
|
||||
if failedResolve > 0 {
|
||||
os.Exit(1)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
successApply := 0
|
||||
failedApply := 0
|
||||
for idx, target := range targets {
|
||||
req := commonSvc.FifoStockV2ReflowRequest{
|
||||
FlagGroupCode: target.FlagGroupCode,
|
||||
ProductWarehouseID: target.ProductWarehouseID,
|
||||
AsOf: asOf,
|
||||
IdempotencyKey: fmt.Sprintf(
|
||||
"manual-pfk-reflow-%d-%d-%s-%d-%d",
|
||||
projectFlockKandangID,
|
||||
target.ProductWarehouseID,
|
||||
strings.ToUpper(strings.TrimSpace(target.FlagGroupCode)),
|
||||
time.Now().UnixNano(),
|
||||
idx,
|
||||
),
|
||||
}
|
||||
|
||||
res, err := fifoStockV2Svc.Reflow(ctx, req)
|
||||
if err != nil {
|
||||
fmt.Printf("FAIL pw=%d flag_group=%s error=%v\n", target.ProductWarehouseID, target.FlagGroupCode, err)
|
||||
failedApply++
|
||||
continue
|
||||
}
|
||||
|
||||
fmt.Printf(
|
||||
"DONE pw=%d flag_group=%s rollback=%.3f allocate=%.3f pending=%.3f processed_usable=%d\n",
|
||||
target.ProductWarehouseID,
|
||||
target.FlagGroupCode,
|
||||
res.Rollback.ReleasedQty,
|
||||
res.Allocate.AllocatedQty,
|
||||
res.Allocate.PendingQty,
|
||||
res.ProcessedUsables,
|
||||
)
|
||||
successApply++
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
fmt.Printf(
|
||||
"Summary: planned=%d skipped_pw=%d failed_resolve=%d applied=%d failed_apply=%d\n",
|
||||
len(targets),
|
||||
skippedPW,
|
||||
failedResolve,
|
||||
successApply,
|
||||
failedApply,
|
||||
)
|
||||
if failedResolve > 0 || failedApply > 0 {
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func parseAsOf(raw string) (*time.Time, error) {
|
||||
raw = strings.TrimSpace(raw)
|
||||
if raw == "" {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
layouts := []string{
|
||||
time.RFC3339Nano,
|
||||
time.RFC3339,
|
||||
"2006-01-02 15:04:05",
|
||||
"2006-01-02",
|
||||
}
|
||||
|
||||
for _, layout := range layouts {
|
||||
parsed, err := time.Parse(layout, raw)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
if layout == "2006-01-02" {
|
||||
endOfDay := time.Date(parsed.Year(), parsed.Month(), parsed.Day(), 23, 59, 59, int(time.Second-time.Nanosecond), time.UTC)
|
||||
return &endOfDay, nil
|
||||
}
|
||||
asOf := parsed.UTC()
|
||||
return &asOf, nil
|
||||
}
|
||||
|
||||
return nil, fmt.Errorf("unsupported format %q", raw)
|
||||
}
|
||||
|
||||
func modeLabel(apply bool) string {
|
||||
if apply {
|
||||
return "APPLY"
|
||||
}
|
||||
return "DRY-RUN"
|
||||
}
|
||||
|
||||
func displayOptionalUint(v *uint) string {
|
||||
if v == nil {
|
||||
return "NULL"
|
||||
}
|
||||
return fmt.Sprintf("%d", *v)
|
||||
}
|
||||
|
||||
func projectFlockKandangExists(ctx context.Context, db *gorm.DB, projectFlockKandangID uint) (bool, error) {
|
||||
var count int64
|
||||
err := db.WithContext(ctx).
|
||||
Table("project_flock_kandangs").
|
||||
Where("id = ?", projectFlockKandangID).
|
||||
Count(&count).Error
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
return count > 0, nil
|
||||
}
|
||||
|
||||
func loadScopedProductWarehouses(ctx context.Context, db *gorm.DB, projectFlockKandangID uint, includeShared bool) ([]productWarehouseScopeRow, error) {
|
||||
if !includeShared {
|
||||
var rows []productWarehouseScopeRow
|
||||
err := db.WithContext(ctx).
|
||||
Table("product_warehouses").
|
||||
Select("id AS product_warehouse_id, product_id, warehouse_id, project_flock_kandang_id").
|
||||
Where("project_flock_kandang_id = ?", projectFlockKandangID).
|
||||
Order("id ASC").
|
||||
Scan(&rows).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return rows, nil
|
||||
}
|
||||
|
||||
query := `
|
||||
WITH scoped_pw AS (
|
||||
SELECT pw.id AS product_warehouse_id
|
||||
FROM product_warehouses pw
|
||||
WHERE pw.project_flock_kandang_id = ?
|
||||
|
||||
UNION
|
||||
SELECT pc.product_warehouse_id
|
||||
FROM project_chickins pc
|
||||
WHERE pc.project_flock_kandang_id = ?
|
||||
AND pc.deleted_at IS NULL
|
||||
|
||||
UNION
|
||||
SELECT rs.product_warehouse_id
|
||||
FROM recordings r
|
||||
JOIN recording_stocks rs ON rs.recording_id = r.id
|
||||
WHERE r.project_flock_kandangs_id = ?
|
||||
AND r.deleted_at IS NULL
|
||||
|
||||
UNION
|
||||
SELECT rd.product_warehouse_id
|
||||
FROM recordings r
|
||||
JOIN recording_depletions rd ON rd.recording_id = r.id
|
||||
WHERE r.project_flock_kandangs_id = ?
|
||||
AND r.deleted_at IS NULL
|
||||
|
||||
UNION
|
||||
SELECT rd.source_product_warehouse_id
|
||||
FROM recordings r
|
||||
JOIN recording_depletions rd ON rd.recording_id = r.id
|
||||
WHERE r.project_flock_kandangs_id = ?
|
||||
AND r.deleted_at IS NULL
|
||||
AND rd.source_product_warehouse_id IS NOT NULL
|
||||
|
||||
UNION
|
||||
SELECT re.product_warehouse_id
|
||||
FROM recordings r
|
||||
JOIN recording_eggs re ON re.recording_id = r.id
|
||||
WHERE r.project_flock_kandangs_id = ?
|
||||
AND r.deleted_at IS NULL
|
||||
|
||||
UNION
|
||||
SELECT lts.product_warehouse_id
|
||||
FROM laying_transfer_sources lts
|
||||
WHERE lts.source_project_flock_kandang_id = ?
|
||||
AND lts.deleted_at IS NULL
|
||||
AND lts.product_warehouse_id IS NOT NULL
|
||||
|
||||
UNION
|
||||
SELECT ltt.product_warehouse_id
|
||||
FROM laying_transfer_targets ltt
|
||||
WHERE ltt.target_project_flock_kandang_id = ?
|
||||
AND ltt.deleted_at IS NULL
|
||||
AND ltt.product_warehouse_id IS NOT NULL
|
||||
|
||||
UNION
|
||||
SELECT pi.product_warehouse_id
|
||||
FROM purchase_items pi
|
||||
WHERE pi.project_flock_kandang_id = ?
|
||||
AND pi.product_warehouse_id IS NOT NULL
|
||||
)
|
||||
SELECT DISTINCT
|
||||
pw.id AS product_warehouse_id,
|
||||
pw.product_id,
|
||||
pw.warehouse_id,
|
||||
pw.project_flock_kandang_id
|
||||
FROM scoped_pw s
|
||||
JOIN product_warehouses pw ON pw.id = s.product_warehouse_id
|
||||
ORDER BY pw.id ASC
|
||||
`
|
||||
|
||||
var rows []productWarehouseScopeRow
|
||||
err := db.WithContext(ctx).
|
||||
Raw(
|
||||
query,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
projectFlockKandangID,
|
||||
).
|
||||
Scan(&rows).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return rows, nil
|
||||
}
|
||||
|
||||
func resolveFlagGroupsByProductWarehouse(ctx context.Context, db *gorm.DB, productWarehouseID uint) ([]string, error) {
|
||||
var groups []string
|
||||
err := db.WithContext(ctx).
|
||||
Table("fifo_stock_v2_route_rules rr").
|
||||
Select("DISTINCT rr.flag_group_code").
|
||||
Joins("JOIN fifo_stock_v2_flag_groups fg ON fg.code = rr.flag_group_code AND fg.is_active = TRUE").
|
||||
Where("rr.is_active = TRUE").
|
||||
Where(`
|
||||
EXISTS (
|
||||
SELECT 1
|
||||
FROM product_warehouses pw
|
||||
JOIN flags f ON f.flagable_id = pw.product_id
|
||||
JOIN fifo_stock_v2_flag_members fm ON fm.flag_name = f.name AND fm.is_active = TRUE
|
||||
WHERE pw.id = ?
|
||||
AND f.flagable_type = ?
|
||||
AND fm.flag_group_code = rr.flag_group_code
|
||||
)
|
||||
`, productWarehouseID, entity.FlagableTypeProduct).
|
||||
Order("rr.flag_group_code ASC").
|
||||
Scan(&groups).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return groups, nil
|
||||
}
|
||||
Reference in New Issue
Block a user