jyMicroservices/jyBXSubscribe/rpc/internal/logic/bypushhistorylogic.go

package logic

import (
	"context"
	"github.com/zeromicro/go-zero/core/logx"
	IC "jyBXSubscribe/rpc/init"
	"jyBXSubscribe/rpc/internal/svc"
	"jyBXSubscribe/rpc/model"
	"jyBXSubscribe/rpc/type/bxsubscribe"
	"jyBXSubscribe/rpc/util"
	"strings"
	"time"
)

type ByPushHistoryLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewByPushHistoryLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ByPushHistoryLogic {
	return &ByPushHistoryLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

// 推送页面筛选导出
func (l *ByPushHistoryLogic) ByPushHistory(in *bxsubscribe.SubscribeInfosReq) (*bxsubscribe.ByPushHistoryResp, error) {
	// todo: add your logic here and delete this line
	vipType := in.UserType
	spqp := &model.SubPushQueryParam{
		Mgo_bidding:      IC.MgoBidding,                      //mongo
		Bidding:          IC.DB.Mongo.Bidding.Collection,     //招标信息 表
		Bidding_back:     IC.DB.Mongo.Bidding.CollectionBack, //招标信息备份数据 表名
		UserId:           in.UserId,                          //用户id
		PageNum:          1,                                  //当前页码
		PageSize:         20000,                              //每页多少条数据
		SelectTime:       in.SelectTime,                      //时间跨度
		Area:             in.Area,                            //省份
		City:             in.City,                            //城市
		Export:           true,
		Buyerclass:       in.BuyerClass, //采购单位类型
		Subtype:          in.Subtype,    //信息类型
		Subscopeclass:    in.Industry,   //行业
		Key:              in.KeyWords,   //关键词
		Price:            in.Price,      //价格区间
		FileExists:       in.FileExists, //是否有附件
		EntId:            in.EntId,      //商机管理企业id
		EntUserId:        in.EntUserId,  //商机管理用户id
		DeptId:           in.DeptId,     //商机管理部门id
		NewUserId:        in.NewUserId,
		BaseServiceMysql: IC.BaseServiceMysql,
		IsEnt:            in.IsEnt,
	}
	if selectIds := strings.TrimSpace(in.SelectIds); selectIds != "" {
		//encodeArr := strings.Split(selectIds, ",")
		//idArr := make([]string, 0, len(encodeArr))
		//for _, encodeId := range encodeArr {
		//	if tmp := util.DecodeId(encodeId); tmp != "" {
		//		idArr = append(idArr, tmp)
		//	}
		//}
		spqp.SelectInfoIds = strings.Split(selectIds, ",")
	}
	_, _, list := model.NewSubscribePush(vipType).Datas(spqp)
	if list == nil || len(list) == 0 {
		return &bxsubscribe.ByPushHistoryResp{
			ErrorCode: 1,
			ErrorMsg:  "未查询到数据",
		}, nil
	}
	ids := make([]string, 0, len(list))
	keyWords := make([]string, 0, len(list))
	for _, pushData := range list {
		if pushData.XId != "" {
			if xid := util.DecodeId(pushData.XId); len(xid) > 0 {
				ids = append(ids, xid)
				keyWords = append(keyWords, strings.Join(pushData.MatchKeys, ","))
			}
		}
	}
	saveData := map[string]interface{}{
		"s_userid":     in.UserId,
		"comeinfrom":   "pushHistory",
		"comeintime":   time.Now().Unix(),
		"selectIds":    ids,
		"pushKeyWords": keyWords,
	}

	_id := IC.Mgo.Save("export_search", saveData)
	if _id == "" {
		return &bxsubscribe.ByPushHistoryResp{
			ErrorCode: 1,
			ErrorMsg:  "创建导出异常",
		}, nil
	}

	return &bxsubscribe.ByPushHistoryResp{
		Data: util.EncodeId(_id),
	}, nil
}