diff --git a/model/autogenerated_order.go b/model/autogenerated_order.go index 9b2f9f4..e27a68a 100644 --- a/model/autogenerated_order.go +++ b/model/autogenerated_order.go @@ -2027,6 +2027,10 @@ func (qs OrderCardQuerySet) StoreIdEq(storeId uint32) OrderCardQuerySet { return qs.w(qs.db.Where("store_id = ?", storeId)) } +func (qs OrderCardQuerySet) StoreIdOrRevertStoreIdEq(storeId uint64) OrderCardQuerySet { + return qs.w(qs.db.Where("store_id = ? or revert_store_id = ?", storeId, storeId)) +} + // StoreIdGt is an autogenerated method // nolint: dupl func (qs OrderCardQuerySet) StoreIdGt(storeId uint32) OrderCardQuerySet { diff --git a/model/game_card.go b/model/game_card.go index 58532e8..5ba6ff9 100644 --- a/model/game_card.go +++ b/model/game_card.go @@ -919,10 +919,15 @@ func (m *CooperativeRentCardOrderListReq) List() ([]Order, int, error) { if m.PageSize == 0 { m.PageSize = 10 } - assistant := GetUserByUid(m.AssistantUid) - if m.SerialNumber == "" { - m.StoreId = assistant.StoreId + + if m.StoreId == 0 { + return []Order{}, 0, nil } + + assistant := GetUserByUid(m.AssistantUid) + //if m.SerialNumber == "" && m.StoreId == 0{ + // m.StoreId = assistant.StoreId + //} if assistant.UserType != 2 { return orders, 0, errors.New("not assistant") } @@ -978,7 +983,16 @@ func (m *CooperativeRentCardOrderListReq) List() ([]Order, int, error) { qs = qs.CreatedAtLte(m.EndTime) } if m.StoreId != 0 { - qs = qs.StoreIdEq(m.StoreId) + err := NewOrderCardQuerySet(DB).StoreIdOrRevertStoreIdEq(m.StoreId).OrderDescByID().OrderAscByCardStatus().All(&orderCards) + if err != nil { + logger.Error("err:", err) + return orders, 0, err + } + orderIds := make([]uint32, 0) + for i, _ := range orderCards { + orderIds = append(orderIds, orderCards[i].OrderId) + } + qs = qs.IDIn(orderIds...) } count, err := qs.Count() if err != nil {