jiangshuai
2023-09-20 6becb6b281b34e17197a25af86acd35628e88811
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
package models
 
import (
    "fmt"
    "github.com/shopspring/decimal"
    "gorm.io/gorm"
    "wms/constvar"
    "wms/pkg/mysqlx"
)
 
type (
    // Material 物料
    Material struct {
        BaseModelString
        Name string `gorm:"unique;type:varchar(191);not null;comment:物料名称" json:"name"` //物料名称
        //MaterialType constvar.ProductType  `gorm:"index;type:int(11);comment:物料类型(数字)" json:"materialType"`
        Model constvar.MaterialMode `gorm:"type:varchar(191);not null;comment:物料类型(字符串)" json:"model"` //物料类型(字符串)
        //Explain           string                  `gorm:"type:varchar(512);comment:编号说明" json:"explain"`
        //CodeStandardID    string                  `gorm:"type:varchar(191);comment:编码规范ID" json:"codeStandardID"`
        //Specs             string                  `gorm:"type:varchar(191);comment:物料规格" json:"specs"`
        //Type              string                  `gorm:"type:varchar(191);comment:物料型号" json:"type"`
        MinInventory decimal.Decimal `gorm:"type:decimal(35,18);comment:最小库存" json:"minInventory"` //最小库存
        MaxInventory decimal.Decimal `gorm:"type:decimal(35,18);comment:最大库存" json:"maxInventory"` //最大库存
        Amount       decimal.Decimal `gorm:"type:decimal(35,18);comment:数量" json:"amount"`         //数量
        //LockAmount        decimal.Decimal         `gorm:"type:decimal(35,18);default:0;comment:锁定数量" json:"lockAmount"`
        Unit string `gorm:"type:varchar(100);comment:单位" json:"unit"` //单位
        //Note              string                  `gorm:"type:varchar(1024);comment:备注" json:"note"`
        TemplateID string `gorm:"type:varchar(191);comment:模板ID" json:"templateID"` //模板ID
        //FSource           string                  `gorm:"type:varchar(191);comment:生产车间" json:"-"`
        //Status            constvar.MaterialStatus `gorm:"type:int(11);comment:状态" json:"status"`
        Supplier      string          `gorm:"type:varchar(191);comment:供应商" json:"supplier"`         //供应商
        PurchasePrice decimal.Decimal `gorm:"type:decimal(35,18);comment:采购价格" json:"purchasePrice"` //采购价格
        //PurchaseAheadDay  int                     `gorm:"type:int(11);comment:采购提前期(天)" json:"purchaseAheadDay"`
        //ProduceAheadDay   int                     `gorm:"type:int(11);comment:制造提前期(天)" json:"produceAheadDay"`
        MinPurchaseAmount decimal.Decimal `gorm:"type:decimal(35,18);comment:最小采购量" json:"minPurchaseAmount"` //最小采购量
        //PurchaseType      constvar.PurchaseType `gorm:"type:int(11);comment:采购类型" json:"purchaseType"`
        IsSale    bool            `gorm:"type:tinyint(1);comment:是否销售" json:"isSale"`        //是否销售
        SalePrice decimal.Decimal `gorm:"type:decimal(35,18);comment:销售单价" json:"salePrice"` //销售单价
        AutoIncr  uint            `gorm:"type:int(11);comment:自增ID;default:0;" json:"-"`
        //wms添加字段
        ProductType             constvar.ProductType       `gorm:"type:int(11);comment:产品类型" json:"productType"`                    //产品类型
        InvoicingStrategy       constvar.InvoicingStrategy `gorm:"type:int(11);comment:开票策略" json:"invoicingStrategy"`              //开票策略
        OrderCreation           constvar.OrderCreation     `gorm:"type:int(11);comment:订单创建" json:"orderCreation"`                  //订单创建
        CustomerTaxes           decimal.Decimal            `gorm:"type:decimal(20,2);comment:客户税" json:"customerTaxes"`             //客户税百分比
        Cost                    decimal.Decimal            `gorm:"type:decimal(20,2);comment:成本" json:"cost"`                       //成本
        CategoryId              int                        `gorm:"type:int(11);comment:产品类别id" json:"categoryId"`                   //产品类别id
        CategoryName            string                     `gorm:"type:varchar(255);comment:产品类别名称" json:"categoryName"`            //产品类别名称
        InternalReference       string                     `gorm:"type:varchar(255);comment:内部参考" json:"internalReference"`         //内部参考
        Barcode                 string                     `gorm:"type:varchar(255);comment:条码" json:"barcode"`                     //条码
        ProductTagId            int                        `gorm:"type:int(11);comment:产品标签id" json:"productTagId"`                 //产品标签id
        ProductTagName          string                     `gorm:"type:varchar(255);comment:产品标签名称" json:"productTagName"`          //产品标签名称
        CompanyId               int                        `gorm:"type:int(11);comment:公司id" json:"companyId"`                      //公司id
        CompanyName             string                     `gorm:"type:varchar(255);comment:公司名称" json:"companyName"`               //公司名称
        InternalNotes           string                     `gorm:"type:varchar(512);comment:内部说明" json:"internalNotes"`             //内部说明
        SelectProduct           string                     `gorm:"type:varchar(255);comment:可选产品id" json:"selectProduct"`           //可选产品id
        SellExplain             string                     `gorm:"type:varchar(512);comment:销售说明" json:"sellExplain"`               //销售说明
        CanBePurchased          bool                       `gorm:"type:int(11);comment:是否可采购" json:"canBePurchased"`                //是否可采购
        CurrencyName            string                     `gorm:"type:varchar(255);comment:币种名称" json:"currencyName"`              //币种名称
        DeliveryAdvanceTime     int                        `gorm:"type:int(11);comment:交货提前时间(天)" json:"deliveryAdvanceTime"`       //交货提前时间(天)
        ControlStrategy         constvar.InvoicingStrategy `gorm:"type:int(11);comment:控制策略" json:"controlStrategy"`                //控制策略
        BuyExplain              string                     `gorm:"type:varchar(512);comment:采购说明" json:"buyExplain"`                //采购说明
        Principal               string                     `gorm:"type:varchar(255);comment:负责人" json:"principal"`                  //负责人
        Weight                  decimal.Decimal            `gorm:"type:decimal(20,2);comment:重量" json:"weight"`                     //重量
        Volume                  decimal.Decimal            `gorm:"type:decimal(20,2);comment:体积" json:"volume"`                     //体积
        HSCode                  string                     `gorm:"type:varchar(255);comment:HS编码" json:"HSCode"`                    //HS编码
        OriginCountryId         int                        `gorm:"type:int(11);comment:原产地id" json:"originCountryId"`               //原产地id
        OriginCountryName       string                     `gorm:"type:varchar(255);comment:原产地名称" json:"originCountryName"`        //原产地名称
        InStorageExplain        string                     `gorm:"type:varchar(512);comment:入库说明" json:"inStorageExplain"`          //入库说明
        OutStorageExplain       string                     `gorm:"type:varchar(512);comment:出库说明" json:"outStorageExplain"`         //出库说明
        InternalTransferExplain string                     `gorm:"type:varchar(512);comment:内部调拨说明" json:"internalTransferExplain"` //内部调拨说明
    }
 
    MaterialSearch struct {
        Material
        //MaterialTypes   []constvar.MaterialType
        Keyword string
        //SetTemplateType constvar.SetTemplateType
        Order    string
        PageNum  int
        PageSize int
        Ids      []string
        Orm      *gorm.DB
    }
 
    IdAndName struct {
        Id   string `json:"id"`
        Name string `json:"name"`
    }
)
 
func (slf Material) TableName() string {
    return "material"
}
 
func NewMaterialSearch() *MaterialSearch {
    return &MaterialSearch{Orm: mysqlx.GetDB()}
}
 
func (slf *MaterialSearch) SetOrm(tx *gorm.DB) *MaterialSearch {
    slf.Orm = tx
    return slf
}
 
func (slf *MaterialSearch) SetPage(page, size int) *MaterialSearch {
    slf.PageNum, slf.PageSize = page, size
    return slf
}
 
func (slf *MaterialSearch) SetOrder(order string) *MaterialSearch {
    slf.Order = order
    return slf
}
 
func (slf *MaterialSearch) SetID(id string) *MaterialSearch {
    slf.ID = id
    return slf
}
 
func (slf *MaterialSearch) SetIsSale(isSale bool) *MaterialSearch {
    slf.IsSale = isSale
    return slf
}
 
func (slf *MaterialSearch) SetIDs(ids []string) *MaterialSearch {
    slf.Ids = ids
    return slf
}
 
func (slf *MaterialSearch) SetName(name string) *MaterialSearch {
    slf.Name = name
    return slf
}
 
//func (slf *MaterialSearch) SetMaterialType(materialType constvar.MaterialType) *MaterialSearch {
//    slf.MaterialType = materialType
//    return slf
//}
//
//func (slf *MaterialSearch) SetMaterialTypes(materialTypes []constvar.MaterialType) *MaterialSearch {
//    slf.MaterialTypes = materialTypes
//    return slf
//}
 
func (slf *MaterialSearch) SetKeyword(keyword string) *MaterialSearch {
    slf.Keyword = keyword
    return slf
}
 
func (slf *MaterialSearch) SetTemplateID(id string) *MaterialSearch {
    slf.TemplateID = id
    return slf
}
 
//
//func (slf *MaterialSearch) SetSetTemplateType(setType constvar.SetTemplateType) *MaterialSearch {
//    slf.SetTemplateType = setType
//    return slf
//}
 
//func (slf *MaterialSearch) SetStatus(status constvar.MaterialStatus) *MaterialSearch {
//    slf.Status = status
//    return slf
//}
 
func (slf *MaterialSearch) build() *gorm.DB {
    var db = slf.Orm.Table(slf.TableName())
 
    if slf.ID != "" {
        db = db.Where("id = ?", slf.ID)
    }
 
    if slf.Name != "" {
        db = db.Where("name = ?", slf.Name)
    }
 
    if slf.TemplateID != "" {
        db = db.Where("template_id = ?", slf.TemplateID)
    }
 
    //if slf.SetTemplateType > 0 {
    //    if slf.SetTemplateType == constvar.SetTemplateTypeWithTemplate { // 模板ID非空
    //        db = db.Where("template_id <> ?", "")
    //    } else if slf.SetTemplateType == constvar.SetTemplateTypeWithNoTemplate {
    //        db = db.Where("template_id = ?", "")
    //    }
    //}
 
    //if slf.MaterialType > 0 {
    //    db = db.Where("material_type = ?", slf.MaterialType)
    //}
 
    //if len(slf.MaterialTypes) > 0 {
    //    var query string
    //    for i, materialType := range slf.MaterialTypes {
    //        if i == 0 {
    //            query += fmt.Sprintf("(material_type = %d", materialType)
    //        } else {
    //            query += fmt.Sprintf(" or material_type = %d", materialType)
    //        }
    //        if i == len(slf.MaterialTypes)-1 {
    //            query += ")"
    //        }
    //    }
    //    db = db.Where(query)
    //}
 
    if slf.Keyword != "" {
        db = db.Where("name LIKE ? or id LIKE ? ", "%"+slf.Keyword+"%", "%"+slf.Keyword+"%")
    }
 
    if slf.Order != "" {
        db = db.Order(slf.Order)
    }
 
    //if slf.Status != 0 {
    //    db = db.Where("status = ?", slf.Status)
    //}
    if len(slf.Ids) > 0 {
        db = db.Where("id in ?", slf.Ids)
    }
    if slf.IsSale {
        db = db.Where("is_sale = ?", 1)
    }
 
    return db
}
 
// Create 单条插入
func (slf *MaterialSearch) Create(record *Material) error {
    var db = slf.build()
 
    if err := db.Create(record).Error; err != nil {
        return fmt.Errorf("create err: %v, record: %+v", err, record)
    }
 
    return nil
}
 
// CreateBatch 批量插入
func (slf *MaterialSearch) CreateBatch(records []*Material) error {
    var db = slf.build()
 
    if err := db.Create(&records).Error; err != nil {
        return fmt.Errorf("create batch err: %v, records: %+v", err, records)
    }
 
    return nil
}
 
func (slf *MaterialSearch) Save(record *Material) 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 *MaterialSearch) UpdateByMap(upMap map[string]interface{}) error {
    var (
        db = slf.build()
    )
 
    if err := db.Updates(upMap).Error; err != nil {
        return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap)
    }
 
    return nil
}
 
func (slf *MaterialSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error {
    var (
        db = slf.Orm.Table(slf.TableName()).Where(query, args...)
    )
 
    if err := db.Updates(upMap).Error; err != nil {
        return fmt.Errorf("update by query err: %v, query: %s, args: %+v, upMap: %+v", err, query, args, upMap)
    }
 
    return nil
}
 
func (slf *MaterialSearch) Delete() error {
    var db = slf.build()
 
    if err := db.Unscoped().Delete(&Material{}).Error; err != nil {
        return err
    }
 
    return nil
}
 
func (slf *MaterialSearch) First() (*Material, error) {
    var (
        record = new(Material)
        db     = slf.build()
    )
 
    if err := db.First(record).Error; err != nil {
        return record, err
    }
 
    return record, nil
}
 
func (slf *MaterialSearch) Find() ([]*Material, int64, error) {
    var (
        records = make([]*Material, 0)
        total   int64
        db      = slf.build()
    )
 
    if err := db.Count(&total).Error; err != nil {
        return records, total, fmt.Errorf("find count err: %v", err)
    }
    if slf.PageNum*slf.PageSize > 0 {
        db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
    }
    if err := db.Find(&records).Error; err != nil {
        return records, total, fmt.Errorf("find records err: %v", err)
    }
 
    return records, total, nil
}
 
func (slf *MaterialSearch) FindNotTotal() ([]*Material, error) {
    var (
        records = make([]*Material, 0)
        db      = slf.build()
    )
 
    if slf.PageNum*slf.PageSize > 0 {
        db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
    }
    if err := db.Find(&records).Error; err != nil {
        return records, fmt.Errorf("find records err: %v", err)
    }
 
    return records, nil
}
 
// FindByQuery 指定条件查询.
func (slf *MaterialSearch) FindByQuery(query string, args []interface{}) ([]*Material, int64, error) {
    var (
        records = make([]*Material, 0)
        total   int64
        db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
    )
 
    if err := db.Count(&total).Error; err != nil {
        return records, total, fmt.Errorf("find by query count err: %v", err)
    }
    if slf.PageNum*slf.PageSize > 0 {
        db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
    }
    if err := db.Find(&records).Error; err != nil {
        return records, total, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
    }
 
    return records, total, nil
}
 
// FindByQueryNotTotal 指定条件查询&不查询总条数.
func (slf *MaterialSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*Material, error) {
    var (
        records = make([]*Material, 0)
        db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
    )
 
    if slf.PageNum*slf.PageSize > 0 {
        db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
    }
    if err := db.Find(&records).Error; err != nil {
        return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
    }
 
    return records, nil
}
 
//func (slf *MaterialSearch) Upsert(materialList []*Material) error {
//    var buffer bytes.Buffer
//    sql := fmt.Sprintf("insert into `%s` (`id`,`created_at`,`updated_at`,`name`,`model`,`material_type`,`explain`,`code_standard_id`,`specs`,"+
//        "`type`,`min_inventory`,`max_inventory`,`amount`,`unit`,`note`,`f_source`, `purchase_type`) values", slf.TableName())
//    if _, err := buffer.WriteString(sql); err != nil {
//        return err
//    }
//
//    for i, e := range materialList {
//        buffer.WriteString(fmt.Sprintf("('%v','%v','%v','%v','%v','%v','%v','%v','%v','%v','%v','%v','%v','%v','%v','%v', '%d')", e.ID, e.CreatedAt.String(), e.UpdatedAt.String(), e.Name,
//            e.Model, e.MaterialType, e.Explain, e.CodeStandardID, e.Specs, e.Type, e.MinInventory, e.MaxInventory, e.Amount, e.Unit, e.Note, e.FSource, int(e.PurchaseType)))
//        if i == len(materialList)-1 {
//            buffer.WriteString("")
//        } else {
//            buffer.WriteString(",")
//        }
//    }
//    buffer.WriteString(fmt.Sprintf("on duplicate key update `updated_at`=values(`updated_at`), `name`=values(`name`), `specs`=values(`specs`), `type`=values(`type`), " +
//        "`min_inventory`=values(`min_inventory`), `max_inventory`=values(`max_inventory`), `amount`=values(`amount`), `unit`=values(`unit`), `note`=values(`note`), `f_source`=values(`f_source`), `purchase_type`=values(`purchase_type`);"))
//    err := slf.Orm.Exec(buffer.String()).Error
//    if err != nil {
//        logx.Errorf("Upsert sql:%v, err:%v", buffer.String(), err)
//    }
//    return err
//}
 
func (slf *MaterialSearch) GetMaterialListByIds(ids []string) ([]*Material, error) {
    var (
        records = make([]*Material, 0)
        db      = slf.Orm.Table(slf.TableName()).Where("id in (?)", ids)
    )
 
    if err := db.Find(&records).Error; err != nil {
        return records, fmt.Errorf("find records err: %v", err)
    }
 
    return records, nil
}
 
//func ExportMaterial(materials []*Material) (string, error) {
//    var fileName string
//    f := excelize.NewFile()
//
//    f.SetCellValue("Sheet1", "A1", "物料编码")
//    f.SetCellValue("Sheet1", "B1", "编码说明")
//    f.SetCellValue("Sheet1", "C1", "物料名称")
//    f.SetCellValue("Sheet1", "D1", "物料规格")
//    f.SetCellValue("Sheet1", "E1", "物料型号")
//    f.SetCellValue("Sheet1", "F1", "供应商")
//    f.SetCellValue("Sheet1", "G1", "采购价格")
//    f.SetCellValue("Sheet1", "H1", "安全库存")
//    f.SetCellValue("Sheet1", "I1", "剩余库存")
//    f.SetCellValue("Sheet1", "J1", "物料类型")
//    f.SetCellValue("Sheet1", "K1", "单位")
//    f.SetCellValue("Sheet1", "L1", "备注")
//
//    for i, m := range materials {
//        f.SetCellValue("Sheet1", "A"+strconv.Itoa(i+2), m.ID)
//        f.SetCellValue("Sheet1", "B"+strconv.Itoa(i+2), m.Explain)
//        f.SetCellValue("Sheet1", "C"+strconv.Itoa(i+2), m.Name)
//        f.SetCellValue("Sheet1", "D"+strconv.Itoa(i+2), m.Specs)
//        f.SetCellValue("Sheet1", "E"+strconv.Itoa(i+2), m.Type)
//        f.SetCellValue("Sheet1", "F"+strconv.Itoa(i+2), m.Supplier)
//        f.SetCellValue("Sheet1", "G"+strconv.Itoa(i+2), m.PurchasePrice.String())
//        f.SetCellValue("Sheet1", "H"+strconv.Itoa(i+2), m.MinInventory.String()+"-"+m.MaxInventory.String())
//        f.SetCellValue("Sheet1", "I"+strconv.Itoa(i+2), m.Amount.String())
//        f.SetCellValue("Sheet1", "J"+strconv.Itoa(i+2), m.Model)
//        f.SetCellValue("Sheet1", "K"+strconv.Itoa(i+2), m.Unit)
//        f.SetCellValue("Sheet1", "L"+strconv.Itoa(i+2), m.Note)
//    }
//
//    fileName = "material.xlsx"
//    if err := f.SaveAs(fileName); err != nil {
//        return fileName, err
//    }
//
//    return fileName, nil
//}
 
func (slf *MaterialSearch) MaxAutoIncr() (int64, error) {
    type Result struct {
        Max int64
    }
 
    var (
        result Result
        db     = slf.build()
    )
 
    err := db.Select("MAX(auto_incr) as max").Scan(&result).Error
    if err != nil {
        return result.Max, fmt.Errorf("max err: %v", err)
    }
    return result.Max, nil
}