From 84346bdafad764c8964256829dc7dc564fe95769 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期一, 01 七月 2024 22:30:50 +0800 Subject: [PATCH] 月度统计出入库按类型汇总报表定时任务和手动跑任务接口 --- models/warehouse_month_stats.go | 29 ++++++++++++++++++++++++++++- 1 files changed, 28 insertions(+), 1 deletions(-) diff --git a/models/warehouse_month_stats.go b/models/warehouse_month_stats.go index 72f494c..658fab5 100644 --- a/models/warehouse_month_stats.go +++ b/models/warehouse_month_stats.go @@ -48,8 +48,12 @@ } ) +func (slf *WarehouseStatsItems) TableName() string { + return "wms_warehouse_month_stats_items" +} + func (slf *WarehouseMonthStats) TableName() string { - return "wms_month_stats" + return "wms_warehouse_month_stats" } func NewWarehouseMonthStatsSearch() *WarehouseMonthStatsSearch { @@ -96,6 +100,21 @@ return slf } +func (slf *WarehouseMonthStatsSearch) SetWarehouseId(id int) *WarehouseMonthStatsSearch { + slf.WarehouseId = id + return slf +} + +func (slf *WarehouseMonthStatsSearch) Save(record *WarehouseMonthStats) error { + var db = slf.build() + + if err := db.Omit("CreatedAt").Save(record).Error; err != nil { + return fmt.Errorf("save err: %v, record: %+v", err, record) + } + + return nil +} + func (slf *WarehouseMonthStatsSearch) build() *gorm.DB { var db = slf.Orm.Model(&WarehouseMonthStats{}) @@ -120,6 +139,14 @@ db = db.Select(slf.Fields) } + if slf.WarehouseId != 0 { + db = db.Where("warehouse_id = ?", slf.WarehouseId) + } + + if slf.Preload { + db = db.Preload("InputItems").Preload("OutputItems") + } + return db } -- Gitblit v1.8.0