78 lines
2.3 KiB
Go
78 lines
2.3 KiB
Go
|
package model
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ InvInventoryLogModel = (*customInvInventoryLogModel)(nil)
|
||
|
|
|
||
|
|
type (
|
||
|
|
InvInventoryLogModel interface {
|
||
|
|
invInventoryLogModel
|
||
|
|
FindList(ctx context.Context, tenantId string, page, pageSize int64, productId, refType string, changeType int64, startDate, endDate string) ([]*InvInventoryLog, int64, error)
|
||
|
|
FindByRefId(ctx context.Context, refId string) ([]*InvInventoryLog, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
customInvInventoryLogModel struct {
|
||
|
|
*defaultInvInventoryLogModel
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewInvInventoryLogModel(conn sqlx.SqlConn) InvInventoryLogModel {
|
||
|
|
return &customInvInventoryLogModel{
|
||
|
|
defaultInvInventoryLogModel: newInvInventoryLogModel(conn),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *customInvInventoryLogModel) FindList(ctx context.Context, tenantId string, page, pageSize int64, productId, refType string, changeType int64, startDate, endDate string) ([]*InvInventoryLog, int64, error) {
|
||
|
|
where := "WHERE tenant_id = ?"
|
||
|
|
args := []interface{}{tenantId}
|
||
|
|
|
||
|
|
if productId != "" {
|
||
|
|
where += " AND product_id = ?"
|
||
|
|
args = append(args, productId)
|
||
|
|
}
|
||
|
|
if refType != "" {
|
||
|
|
where += " AND ref_type = ?"
|
||
|
|
args = append(args, refType)
|
||
|
|
}
|
||
|
|
if changeType > 0 {
|
||
|
|
where += " AND change_type = ?"
|
||
|
|
args = append(args, changeType)
|
||
|
|
}
|
||
|
|
if startDate != "" {
|
||
|
|
where += " AND log_date >= ?"
|
||
|
|
args = append(args, startDate)
|
||
|
|
}
|
||
|
|
if endDate != "" {
|
||
|
|
where += " AND log_date <= ?"
|
||
|
|
args = append(args, endDate)
|
||
|
|
}
|
||
|
|
|
||
|
|
var total int64
|
||
|
|
countQuery := fmt.Sprintf("SELECT COUNT(*) FROM %s %s", m.table, where)
|
||
|
|
if err := m.conn.QueryRowCtx(ctx, &total, countQuery, args...); err != nil {
|
||
|
|
return nil, 0, err
|
||
|
|
}
|
||
|
|
|
||
|
|
var list []*InvInventoryLog
|
||
|
|
query := fmt.Sprintf("SELECT %s FROM %s %s ORDER BY created_at DESC LIMIT ? OFFSET ?", invInventoryLogRows, m.table, where)
|
||
|
|
args = append(args, pageSize, (page-1)*pageSize)
|
||
|
|
if err := m.conn.QueryRowsCtx(ctx, &list, query, args...); err != nil {
|
||
|
|
return nil, 0, err
|
||
|
|
}
|
||
|
|
return list, total, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *customInvInventoryLogModel) FindByRefId(ctx context.Context, refId string) ([]*InvInventoryLog, error) {
|
||
|
|
var list []*InvInventoryLog
|
||
|
|
query := fmt.Sprintf("SELECT %s FROM %s WHERE ref_id = ? ORDER BY created_at DESC", invInventoryLogRows, m.table)
|
||
|
|
if err := m.conn.QueryRowsCtx(ctx, &list, query, refId); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return list, nil
|
||
|
|
}
|