jyInfo/rpc/common/internal/logic/sensitivemethodlogic.go

package logic

import (
	md "app.yhyue.com/moapp/jyInfo/rpc/model"
	model "app.yhyue.com/moapp/jyInfo/rpc/model/es"
	"app.yhyue.com/moapp/jyInfo/rpc/util"
	mc "app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/redis"
	"context"
	"fmt"
	"log"
	"strings"
	"time"

	"app.yhyue.com/moapp/jyInfo/rpc/common/commoninfo"
	"app.yhyue.com/moapp/jyInfo/rpc/common/internal/svc"
	"github.com/zeromicro/go-zero/core/logx"
)

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

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

//数据组rpc回调，敏感词过滤后结果
func (l *SensitiveMethodLogic) SensitiveMethod(in *commoninfo.SensitiveRequest) (*commoninfo.SensitiveResponse, error) {
	var (
		resp commoninfo.SensitiveResponse
		//isSensitive 是否有敏感词  isPublishInfo招标信息是否需要自动发布 isPublishSup 供应信息是否需要自动发布
		isSensitive, isPublishInfo, isPublishSup, isAttach bool
	)
	log.Println("接收到敏感词回调----------------------", in)
	upData := make(map[string]interface{})
	query := make(map[string]interface{})
	attachments := make(map[string]interface{})
	if in.Attachments != "" {
		attachments = mc.StringToMap(in.Attachments)
	}
	sensitiveAttach := make(map[string]interface{})

	for k, v := range attachments {
		vs := mc.StructToMapMore(v)
		var msg []string
		if mc.InterfaceToStr(vs["state"]) != "200" {
			if mc.InterfaceToStr(vs["state"]) == "300" {
				msg = append(msg, "敏感词校验附件下载错误")
			} else if mc.InterfaceToStr(vs["state"]) == "400" {
				msg = append(msg, "敏感词校验附件解析错误")
			} else if mc.InterfaceToStr(vs["state"]) == "500" {
				msg = append(msg, "敏感词校验附件类型不支持")
			} else {
				msg = append(msg, "敏感词识别失败")
			}
		} else if s, ok := vs["sensitive"]; ok && s != nil && len(s.([]interface{})) > 0 {
			msg = mc.ObjArrToStringArr(vs["sensitive"].([]interface{}))
		}
		msgStr := strings.Join(msg, ",")
		if msgStr != "" {
			isAttach = true
			log.Println("附件敏感词:", msgStr)
		}
		sensitiveAttach[k] = msgStr
	}
	if in.Title != nil || in.Detail != nil || isAttach {
		isSensitive = true
	}
	if !isSensitive && !md.Sensitive.Information {
		isPublishInfo = true
	}

	if !isSensitive && !md.Sensitive.SupplyInfo {
		isPublishSup = true
	}

	if in.Title != nil {
		upData["sensitive_title"] = strings.Join(in.Title, ",")
	}
	if in.Detail != nil {
		upData["sensitive_detail"] = strings.Join(in.Detail, ",")
	}
	if isAttach {
		upData["sensitive_attach"] = mc.MapToJson(sensitiveAttach)
	}
	if in.AttachTxt != "" && in.MsgType != "3" {
		upData["discern_attach"] = in.AttachTxt
	}
	//当前信息是否已被删除 默认：false；删除：true
	isDel := false
	query["id"] = in.Id
	upData["status"] = 2
	switch in.MsgType {
	case "1", "2", "4", "5", "6": //招标信息
		data := md.Mysql.FindOne("information", query, "", "")
		if data != nil && len(*data) > 0 {
			//是否已被删除
			isDel = mc.IntAll((*data)["is_del"]) == -1
			//未删除 且 不用人工审核 直接入消息队列 进行发布流程
			if !isDel && isPublishInfo {
				upData["status"] = 3
				upData["review_time"] = time.Now().Format("2006-01-02 15:04:05")
				upData["review_desc"] = "自动审批通过"
				//调用发布功能
				nsq, err := util.NewNsqInfo(md.NsqConfig.Ip, md.NsqConfig.Topic, in.Id, "2", in.MsgType, false, *data)
				if err != nil || nsq.NsqPushInfo() != nil {
					resp.ErrCode = -2
					resp.ErrMsg = "发布信息：信息入nsq消息队列失败"
					//break
				}
			}
			//修改此信息敏感词过滤结果
			if !md.Mysql.Update("information", query, upData) {
				log.Println("调用信息发布成功，更新信息失败", query, upData)
				resp.ErrCode = -1
				resp.ErrMsg = resp.ErrMsg + "敏感词过滤成功，更新库数据信息失败"
			}
		}
	case "3": //供应信息
		data := md.Mysql.FindOne("supply_info", query, "", "")
		if data != nil && len(*data) > 0 {
			//是否已被删除
			isDel = mc.IntAll((*data)["is_del"]) == -1
			//信息未被删除 且 不用人工审核 直接进入es 发布信息
			if !isDel && isPublishSup {
				upData["status"] = 3
				upData["published"] = 2
				upData["review_time"] = time.Now().Format("2006-01-02 15:04:05")
				upData["publish_time"] = time.Now().Format("2006-01-02 15:04:05")
				upData["review_desc"] = "自动审批通过"
				//调用发布功能
				entNameKye := fmt.Sprintf("userEntName_%s_%s_%s", (*data)["user_id"], in.Id, in.MsgType)
				entName := redis.GetStr("other", entNameKye)
				supInfo := make(map[string]interface{})
				supInfo["id"] = (*data)["id"]
				supInfo["title"] = (*data)["title"]
				supInfo["detail"] = (*data)["detail"]
				supInfo["province"] = (*data)["province"]
				supInfo["city"] = (*data)["city"]
				supInfo["ent_id"] = (*data)["ent_id"]
				validityTime := time.Now().Unix()
				if (*data)["validity_time"] != nil && (*data)["validity_time"] != "" {
					validity, _ := time.ParseInLocation("2006-01-02 15:04:05", mc.InterfaceToStr((*data)["validity_time"]), time.Local)
					validityTime = validity.Unix()
				}
				supInfo["validity_time"] = validityTime
				if !model.SaveSupplyInfo(entName, supInfo) {
					log.Println("调用信息发布成功，更新信息失败", entName, supInfo)
					resp.ErrCode = -3
					resp.ErrMsg = "供应信息： 调用SaveSupplyInfo 信息发布失败"
				}
			}

			if !md.Mysql.Update("supply_info", query, upData) {
				log.Println("调用信息发布成功，更新信息失败", query, upData)
				resp.ErrCode = -1
				resp.ErrMsg = resp.ErrMsg + "供应信息：敏感词过滤成功，更新库信息状态失败"
			}
		}
	default:
		resp.ErrCode = -1
		resp.ErrMsg = "信息类型错误"
	}

	return &resp, nil
}