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

package logic

import (
	"app.yhyue.com/moapp/jybase/common"
	"context"
	"database/sql"
	IC "jyBXSubscribe/rpc/init"
	"jyBXSubscribe/rpc/model"
	"strings"

	"jyBXSubscribe/rpc/internal/svc"
	"jyBXSubscribe/rpc/type/bxsubscribe"

	"github.com/zeromicro/go-zero/core/logx"
)

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

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

// 信息分发
func (l *MsgDistributorLogic) MsgDistributor(in *bxsubscribe.MsgDistributorReq) (*bxsubscribe.StatusResp, error) {
	userEnt := model.EntInfo(common.IntAll(in.EntId), common.IntAll(in.EntUserId))
	if !(userEnt.Role_admin_system || userEnt.Role_admin_department) {
		return &bxsubscribe.StatusResp{ErrorCode: -1, ErrorMsg: "无权限"}, nil
	}
	//根据信息查询区域列表
	var regin, infoIds []string
	var pushCas []*model.PushCa
	for _, id := range strings.Split(in.MessageId, ",") {
		if id != "" {
			infoIds = append(infoIds, id)
			pushCas = append(pushCas, &model.PushCa{InfoId: id})
		}
	}
	if len(infoIds) == 0 || in.Staffs == "" {
		return &bxsubscribe.StatusResp{ErrorCode: -1, ErrorMsg: "缺少参数"}, nil
	}
	infoList := model.NewSubscribePush().GetInfoByIds(IC.MgoBidding, IC.DB.Mongo.Bidding.Collection, IC.DB.Mongo.Bidding.CollectionBack, pushCas)
	for _, info := range infoList {
		if info.Area != "" && info.Area != "全国" {
			regin = append(regin, info.Area)
		}
	}
	//区域人员校验
	userArr := model.Distributor(regin, common.IntAll(in.EntId), common.IntAll(in.EntUserId))
	for _, uid := range strings.Split(in.Staffs, ",") {
		check := false
		for _, user := range userArr {
			if user.Id == common.IntAll(uid) {
				check = true
				break
			}
		}
		if !check {
			return &bxsubscribe.StatusResp{ErrorCode: -1, ErrorMsg: "选择人员异常"}, nil
		}
	}
	//查询分发信息内容
	msgRes := IC.BaseServiceMysql.SelectBySql("SELECT p.* FROM pushentniche p,(SELECT infoid,MIN(id) as id FROM pushentniche WHERE infoid in ('?') GROUP BY infoid) b WHERE p.id=b.id", strings.Join(infoIds, "','"))
	if msgRes == nil || len(*msgRes) == 0 {
		return &bxsubscribe.StatusResp{ErrorCode: -1, ErrorMsg: "获取分发信息内容异常"}, nil
	}
	//分发数据库修改
	ok := IC.BaseServiceMysql.ExecTx("分发数据库修改", func(tx *sql.Tx) bool {
		insertRow := []string{"entid", "deptid", "infoid", "matchkeys", "date", "type", "product", "matchways", "matchitems", "disid", "source", "userid", "isvisit", "visittime"}
		msgItems := []string{"entid", "deptid", "infoid", "matchkeys", "date", "type", "product", "matchways", "matchitems", "disid"}
		for _, m := range *msgRes {
			var msgValues, insertValue []interface{}
			for _, key := range msgItems {
				msgValues = append(msgValues, m[key])
			}
			for _, uid := range userArr {
				insertValue = append(insertValue, msgValues)
				insertValue = append(insertValue, []interface{}{3, uid, nil, nil})
			}
			if affected, _ := IC.BaseServiceMysql.InsertBatchByTx(tx, "pushentniche", insertRow, insertValue); affected == 0 {
				return false
			}
		}
		return true
	})
	if !ok {
		return &bxsubscribe.StatusResp{ErrorCode: -1, ErrorMsg: "数据分发异常"}, nil
	}
	return &bxsubscribe.StatusResp{Status: 1}, nil
}