mirror of
https://github.com/serty2005/rmser.git
synced 2026-02-04 19:02:33 -06:00
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package drafts
|
|
|
|
import (
|
|
"rmser/internal/domain/drafts"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/shopspring/decimal"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type pgRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewRepository(db *gorm.DB) drafts.Repository {
|
|
return &pgRepository{db: db}
|
|
}
|
|
|
|
func (r *pgRepository) Create(draft *drafts.DraftInvoice) error {
|
|
return r.db.Create(draft).Error
|
|
}
|
|
|
|
func (r *pgRepository) GetByID(id uuid.UUID) (*drafts.DraftInvoice, error) {
|
|
var draft drafts.DraftInvoice
|
|
err := r.db.
|
|
Preload("Items", func(db *gorm.DB) *gorm.DB {
|
|
return db.Order("draft_invoice_items.raw_name ASC")
|
|
}).
|
|
Preload("Items.Product").
|
|
Preload("Items.Product.MainUnit").
|
|
Preload("Items.Container").
|
|
Where("id = ?", id).
|
|
First(&draft).Error
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &draft, nil
|
|
}
|
|
|
|
func (r *pgRepository) Update(draft *drafts.DraftInvoice) error {
|
|
// Обновляем поля шапки + привязки к серверу
|
|
return r.db.Model(draft).Updates(map[string]interface{}{
|
|
"status": draft.Status,
|
|
"document_number": draft.DocumentNumber,
|
|
"date_incoming": draft.DateIncoming,
|
|
"supplier_id": draft.SupplierID,
|
|
"store_id": draft.StoreID,
|
|
"comment": draft.Comment,
|
|
"rms_invoice_id": draft.RMSInvoiceID,
|
|
"rms_server_id": draft.RMSServerID, // Вдруг поменялся, хотя не должен
|
|
"updated_at": gorm.Expr("NOW()"),
|
|
}).Error
|
|
}
|
|
|
|
func (r *pgRepository) CreateItems(items []drafts.DraftInvoiceItem) error {
|
|
if len(items) == 0 {
|
|
return nil
|
|
}
|
|
return r.db.CreateInBatches(items, 100).Error
|
|
}
|
|
|
|
func (r *pgRepository) UpdateItem(itemID uuid.UUID, productID *uuid.UUID, containerID *uuid.UUID, qty, price decimal.Decimal) error {
|
|
sum := qty.Mul(price)
|
|
isMatched := productID != nil
|
|
|
|
return r.db.Model(&drafts.DraftInvoiceItem{}).
|
|
Where("id = ?", itemID).
|
|
Updates(map[string]interface{}{
|
|
"product_id": productID,
|
|
"container_id": containerID,
|
|
"quantity": qty,
|
|
"price": price,
|
|
"sum": sum,
|
|
"is_matched": isMatched,
|
|
}).Error
|
|
}
|
|
|
|
func (r *pgRepository) Delete(id uuid.UUID) error {
|
|
return r.db.Delete(&drafts.DraftInvoice{}, id).Error
|
|
}
|
|
|
|
// GetActive фильтрует по UserID
|
|
func (r *pgRepository) GetActive(userID uuid.UUID) ([]drafts.DraftInvoice, error) {
|
|
var list []drafts.DraftInvoice
|
|
|
|
activeStatuses := []string{
|
|
drafts.StatusProcessing,
|
|
drafts.StatusReadyToVerify,
|
|
drafts.StatusError,
|
|
drafts.StatusCanceled,
|
|
}
|
|
|
|
err := r.db.
|
|
Preload("Items").
|
|
Preload("Store").
|
|
Where("user_id = ? AND status IN ?", userID, activeStatuses). // <-- FILTER
|
|
Order("created_at DESC").
|
|
Find(&list).Error
|
|
|
|
return list, err
|
|
}
|