Michael Chus
20309d1f0e
Renamed module path git.mchus.pro/mchus/quoteforge → git.mchus.pro/mchus/priceforge, renamed package quoteforge → priceforge, moved binary from cmd/qfs to cmd/pfs.
125 lines
3.1 KiB
Go
125 lines
3.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"time"
|
|
|
|
"git.mchus.pro/mchus/priceforge/internal/models"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type PriceRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewPriceRepository(db *gorm.DB) *PriceRepository {
|
|
return &PriceRepository{db: db}
|
|
}
|
|
|
|
type PricePoint struct {
|
|
Price float64
|
|
Date time.Time
|
|
}
|
|
|
|
// GetPriceHistory returns price history from lot_log for a component
|
|
func (r *PriceRepository) GetPriceHistory(lotName string, periodDays int) ([]PricePoint, error) {
|
|
var points []PricePoint
|
|
since := time.Now().AddDate(0, 0, -periodDays)
|
|
|
|
err := r.db.Model(&models.LotLog{}).
|
|
Select("price, date").
|
|
Where("lot = ? AND date >= ?", lotName, since).
|
|
Order("date DESC").
|
|
Scan(&points).Error
|
|
|
|
return points, err
|
|
}
|
|
|
|
// GetLatestPrice returns the most recent price for a component
|
|
func (r *PriceRepository) GetLatestPrice(lotName string) (*PricePoint, error) {
|
|
var point PricePoint
|
|
err := r.db.Model(&models.LotLog{}).
|
|
Select("price, date").
|
|
Where("lot = ?", lotName).
|
|
Order("date DESC").
|
|
First(&point).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &point, nil
|
|
}
|
|
|
|
// GetPriceOverride returns active override for a component
|
|
func (r *PriceRepository) GetPriceOverride(lotName string) (*models.PriceOverride, error) {
|
|
var override models.PriceOverride
|
|
today := time.Now().Truncate(24 * time.Hour)
|
|
|
|
err := r.db.
|
|
Where("lot_name = ?", lotName).
|
|
Where("valid_from <= ?", today).
|
|
Where("valid_until IS NULL OR valid_until >= ?", today).
|
|
Order("valid_from DESC").
|
|
First(&override).Error
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &override, nil
|
|
}
|
|
|
|
// CreatePriceOverride creates a new price override
|
|
func (r *PriceRepository) CreatePriceOverride(override *models.PriceOverride) error {
|
|
return r.db.Create(override).Error
|
|
}
|
|
|
|
// GetPriceOverrides returns all overrides for a component
|
|
func (r *PriceRepository) GetPriceOverrides(lotName string) ([]models.PriceOverride, error) {
|
|
var overrides []models.PriceOverride
|
|
err := r.db.
|
|
Where("lot_name = ?", lotName).
|
|
Order("valid_from DESC").
|
|
Find(&overrides).Error
|
|
return overrides, err
|
|
}
|
|
|
|
// DeletePriceOverride deletes an override
|
|
func (r *PriceRepository) DeletePriceOverride(id uint) error {
|
|
return r.db.Delete(&models.PriceOverride{}, id).Error
|
|
}
|
|
|
|
// GetQuoteCount returns the number of quotes in lot_log for a period
|
|
func (r *PriceRepository) GetQuoteCount(lotName string, periodDays int) (int64, error) {
|
|
var count int64
|
|
since := time.Now().AddDate(0, 0, -periodDays)
|
|
|
|
err := r.db.Model(&models.LotLog{}).
|
|
Where("lot = ? AND date >= ?", lotName, since).
|
|
Count(&count).Error
|
|
|
|
return count, err
|
|
}
|
|
|
|
// GetQuoteCounts returns quote counts for multiple lot names
|
|
func (r *PriceRepository) GetQuoteCounts(lotNames []string) (map[string]int64, error) {
|
|
type Result struct {
|
|
Lot string
|
|
Count int64
|
|
}
|
|
var results []Result
|
|
|
|
err := r.db.Model(&models.LotLog{}).
|
|
Select("lot, COUNT(*) as count").
|
|
Where("lot IN ?", lotNames).
|
|
Group("lot").
|
|
Scan(&results).Error
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
counts := make(map[string]int64)
|
|
for _, r := range results {
|
|
counts[r.Lot] = r.Count
|
|
}
|
|
return counts, nil
|
|
}
|