package logic import ( "context" "lone-services/pkg/modelbase" "lone-services/pkg/utils" "lone-services/rpc/store/pb" "lone-services/services/store/internal/dao" "lone-services/services/store/internal/model" "lone-services/services/store/internal/svc" "strconv" "github.com/zeromicro/go-zero/core/logx" ) type DepositIgnoreItemsLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewDepositIgnoreItemsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DepositIgnoreItemsLogic { return &DepositIgnoreItemsLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *DepositIgnoreItemsLogic) DepositIgnoreItems(in *store.DepositVerifyRequest) (*store.Response, error) { adminInfo := utils.GetUserFromCtx(l.ctx) if adminInfo.ID < utils.NumberOne { return l.fail(utils.ErrorNoLoginInfo) } var info []dao.DepositInfo w := modelbase.Params{ Eq: map[string]string{"status": strconv.Itoa(utils.StatusNotPass)}, Page: int(in.Page), Size: int(in.Size), Order: "id desc", } storeMap := make(map[int64]string) items, err := model.DepositModel{}.Init().Page(w, &info) if err != nil { l.Logger.Error(err) return l.fail(utils.Fail) } if len(storeMap) < utils.NumberOne { var storeIds []string for _, values := range info { storeIds = append(storeIds, strconv.Itoa(int(values.StoreId))) } ww := modelbase.Params{In: map[string][]string{"id": storeIds}} var storeItems []dao.StoreNames err := model.StoreModel{}.Init().Items(ww, &storeItems) if err != nil { l.Logger.Error(err) return l.fail(utils.Fail) } for _, item := range storeItems { storeMap[item.Id] = item.Name } } for key, val := range info { if name, ok := storeMap[val.StoreId]; ok { info[key].StoreName = name } } items.Items = info return l.ok(items) }