mirror of
https://gitlab.com/mbugroup/lti-api.git
synced 2026-05-25 15:55:44 +00:00
init depresiasi
This commit is contained in:
@@ -0,0 +1,103 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
const farmDepreciationSnapshotTable = "farm_depreciation_snapshots"
|
||||
|
||||
func NormalizeDateOnlyUTC(value time.Time) time.Time {
|
||||
if value.IsZero() {
|
||||
return value
|
||||
}
|
||||
v := value.UTC()
|
||||
return time.Date(v.Year(), v.Month(), v.Day(), 0, 0, 0, 0, time.UTC)
|
||||
}
|
||||
|
||||
func MinNonZeroDateOnlyUTC(values ...time.Time) time.Time {
|
||||
var out time.Time
|
||||
for _, value := range values {
|
||||
if value.IsZero() {
|
||||
continue
|
||||
}
|
||||
normalized := NormalizeDateOnlyUTC(value)
|
||||
if out.IsZero() || normalized.Before(out) {
|
||||
out = normalized
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func InvalidateFarmDepreciationSnapshotsFromDate(ctx context.Context, db *gorm.DB, farmIDs []uint, fromDate time.Time) error {
|
||||
if db == nil {
|
||||
return nil
|
||||
}
|
||||
if fromDate.IsZero() {
|
||||
return nil
|
||||
}
|
||||
|
||||
fromDate = NormalizeDateOnlyUTC(fromDate)
|
||||
query := db.WithContext(ctx).
|
||||
Table(farmDepreciationSnapshotTable).
|
||||
Where("period_date >= ?", fromDate)
|
||||
if len(farmIDs) > 0 {
|
||||
query = query.Where("project_flock_id IN ?", farmIDs)
|
||||
}
|
||||
return query.Delete(nil).Error
|
||||
}
|
||||
|
||||
func ResolveProjectFlockIDsByProjectFlockKandangIDs(ctx context.Context, db *gorm.DB, pfkIDs []uint) ([]uint, error) {
|
||||
if db == nil || len(pfkIDs) == 0 {
|
||||
return []uint{}, nil
|
||||
}
|
||||
|
||||
var projectFlockIDs []uint
|
||||
if err := db.WithContext(ctx).
|
||||
Table("project_flock_kandangs").
|
||||
Distinct("project_flock_id").
|
||||
Where("id IN ?", pfkIDs).
|
||||
Pluck("project_flock_id", &projectFlockIDs).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return projectFlockIDs, nil
|
||||
}
|
||||
|
||||
func ResolveProjectFlockIDsByExpenseID(ctx context.Context, db *gorm.DB, expenseID uint) ([]uint, error) {
|
||||
if db == nil || expenseID == 0 {
|
||||
return []uint{}, nil
|
||||
}
|
||||
|
||||
query := `
|
||||
WITH direct_farms AS (
|
||||
SELECT DISTINCT pfk.project_flock_id
|
||||
FROM expense_nonstocks ens
|
||||
JOIN project_flock_kandangs pfk ON pfk.id = ens.project_flock_kandang_id
|
||||
WHERE ens.expense_id = @expense_id
|
||||
),
|
||||
json_farms AS (
|
||||
SELECT DISTINCT (jsonb_array_elements_text(e.project_flock_id::jsonb))::bigint AS project_flock_id
|
||||
FROM expenses e
|
||||
WHERE e.id = @expense_id
|
||||
AND e.project_flock_id IS NOT NULL
|
||||
)
|
||||
SELECT DISTINCT project_flock_id
|
||||
FROM (
|
||||
SELECT project_flock_id FROM direct_farms
|
||||
UNION ALL
|
||||
SELECT project_flock_id FROM json_farms
|
||||
) x
|
||||
`
|
||||
|
||||
var ids []uint
|
||||
if err := db.WithContext(ctx).Raw(query, map[string]any{
|
||||
"expense_id": expenseID,
|
||||
}).Scan(&ids).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return ids, nil
|
||||
}
|
||||
Reference in New Issue
Block a user