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

package logic

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

	"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, staffs []string
	staffs = strings.Split(in.Staffs, ",")
	var pushCas []*model.PushCa
	for _, eid := range strings.Split(in.MessageId, ",") {
		if eid != "" {
			array := encrypt.DecodeArticleId2ByCheck(eid)
			if len(array) == 1 && array[0] != "" {
				infoIds = append(infoIds, array[0])
				pushCas = append(pushCas, &model.PushCa{InfoId: array[0]})
			}
		}
	}
	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 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(fmt.Sprintf("SELECT p.* FROM pushentniche p,(SELECT infoid,MIN(id) as id FROM pushentniche WHERE infoid in ('%s') and entid =? GROUP BY infoid) b WHERE p.id=b.id", strings.Join(infoIds, "','")), in.EntId)
	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", "type", "product", "matchways", "matchitems", "disid", "source", "date", "userid", "isvisit", "visittime"}
		msgItems := []string{"entid", "deptid", "infoid", "matchkeys", "type", "product", "matchways", "matchitems", "disid"}
		dateNew := time.Now().Unix()
		for _, m := range *msgRes {
			var msgValues, insertValue []interface{}
			for _, key := range msgItems {
				msgValues = append(msgValues, m[key])
			}
			for _, uid := range staffs {
				insertValue = append(insertValue, msgValues...)
				insertValue = append(insertValue, []interface{}{3, dateNew, uid, nil, nil}...)
			}
			affected, _ := IC.BaseServiceMysql.InsertBatchByTx(tx, "pushentniche", insertRow, insertValue)
			if affected == 0 {
				return false
			}
		}
		return true
	})
	if !ok {
		return &bxsubscribe.StatusResp{ErrorCode: -1, ErrorMsg: "数据分发异常"}, nil
	}
	return &bxsubscribe.StatusResp{Status: 1}, nil
}