DataService/jy_publishing/task.go

package main

import (
	"app.yhyue.com/BP/servicerd/proto"
	"context"
	"encoding/json"
	"fmt"
	"google.golang.org/grpc"
	"gopkg.in/mgo.v2/bson"
	pb "jy_publishing/proto"
	"net"
	"strconv"
	"strings"
	"time"
	"utils"
	"utils/log"
	"utils/mongodb"
)

var (
	JyUrl      = "https://www.jianyu360.cn/article/content/%s.html"
	InfoFields = []string{"title", "project_code", "province", "city", "industry", "buyer", "budget", "winner", "amount",
		"detail", "attch", "contract_person", "contract_phone", "attach", "discern_attach"}
)

var SaveFields = map[string]string{
	"title":           "title",
	"project_code":    "projectcode",
	"province":        "area",
	"city":            "city",
	"buyer":           "buyer",
	"budget":          "budget",
	"winner":          "s_winner",
	"amount":          "bidamount",
	"detail":          "detail",
	"contract_person": "buyerperson",
	"contract_phone":  "buyertel",
	"discern_attach":  "attach_text",
	//"attch": "",
	//"industry": "",
	//"contract_overt": "",
}

// @Description 信息处理（信息发布和附件识别）
// 1、敏感词处理，2、信息发布，3、信息删除
// @Author J 2022/4/9 11:47 AM
func taskInfo(obj interface{}) {
	info, _ := obj.(map[string]interface{})
	if util.ObjToString(info["action"]) == "1" {
		// 敏感词处理
		Sensitive(info)

	} else if util.ObjToString(info["action"]) == "2" {
		// 数据处理
		InfoPub(info)
	} else if util.ObjToString(info["action"]) == "3" {
		id := util.ObjToString(info["id"])
		DelMethod(id)
	}
}

// @Description 敏感词处理(title, content, attachment)
// @Author J 2022/4/11 9:36 AM
func Sensitive(info map[string]interface{}) {
	tmp := info["appendInfo"].(map[string]interface{})
	tArr := WordsIdentify(util.ObjToString(tmp["title"]))
	dArr := WordsIdentify(util.ObjToString(tmp["detail"]))
	if tmp["attach"] != nil {
		attsMap := tmp["attach"].(map[string]interface{})
		other := map[string]interface{}{
			"id":      tmp["id"],
			"action":  tmp["action"],
			"msgType": tmp["msgType"],
			"title":   strings.Join(tArr, ","),
			"detail":  strings.Join(dArr, ","),
		}
		otherJson, _ := json.Marshal(other)
		var attsArr []*pb.Request
		for _, m := range attsMap {
			m1 := m.(map[string]interface{})
			attsArr = append(attsArr, &pb.Request{
				FileUrl:     util.ObjToString(m1["fid"]),
				FileName:    util.ObjToString(m1["filename"]),
				FileType:    util.ObjToString(m1["ftype"]),
				ReturnType:  2,
				ExtractType: 2,
			})
		}
		msginfo := &pb.FileRequest{
			Message: attsArr,
			Other:   string(otherJson),
		}
		_ = MProducer.Publish(msginfo)
	} else {
		// 没有附件
		req := &pb.SensitiveRequest{
			Id:      util.ObjToString(info["id"]),
			MsgType: util.ObjToString(info["msgtype"]),
			Title:   tArr,
			Detail:  dArr,
		}
		JyRpcSensitive(req)
	}

	//atts := tmp["attachment"].(map[string]interface{})
	//resultAtts := make(map[string]interface{})
	//resultAttach := make(map[string]interface{})
	//for k, v := range atts {
	//	attach := make(map[string]interface{}) // attach_text字段
	//	resp, err := AttsMethod(v.(map[string]interface{}))
	//	if err != nil {
	//		return nil
	//	}
	//	for i, r := range resp.Result {
	//		if w := WordsIdentify(r.TextContent); w != nil {
	//			resultAtts[k] = w
	//		}
	//		attach[strconv.Itoa(i)] = map[string]interface{}{"file_name": r.FileName, "attach_url": r.TextUrl}
	//	}
	//	resultAttach[k] = attach
	//}
	//resultMap["attach_text"] = resultAttach
	//resultMap["attachment"] = resultAtts
}

// @Description 敏感词识别
// @Author J 2022/4/12 1:33 PM
func WordsIdentify(str string) []string {
	if str == "" {
		return nil
	}
	ret := Ms.Discern(str, 2)
	if len(ret) > 0 {
		var words []string
		for _, r := range ret {
			words = append(words, r.MatchRule)
		}
		return words
	}
	return []string{}
}

// @Description 附件调用gRpc接口处理
// @Author J 2022/4/12 10:02 AM
// Deprecated
func AttsMethod(att map[string]interface{}) (*pb.FileResponse, error) {
	reqs := &pb.FileRequest{
		Message: []*pb.Request{{
			FileName: "",
			FileType: "",
			FileUrl:  ""}},
		Other: "",
	}
	// 1.调用gRPC接口
	conn, err := grpc.Dial(ClientAddr, grpc.WithInsecure())
	if err != nil {
		return nil, err
	}
	var client proto.ServiceClient
	client = proto.NewServiceClient(conn)
	repl, err := client.Apply(context.Background(), &proto.ApplyReqData{Name: "extract_service", Balance: 0})
	if err != nil {
		return nil, err
	}
	//2.业务调用
	addr := fmt.Sprintf("%s:%d", repl.Ip, repl.Port)
	conn_b, err := grpc.Dial(addr, grpc.WithInsecure())
	if err != nil {
		return nil, err
	}
	defer func(conn_b *grpc.ClientConn) {
		_ = conn_b.Close()
	}(conn_b)
	pc := pb.NewFileExtractClient(conn_b)
	rep, err := pc.FileExtract(context.Background(), reqs)
	if err != nil {
		return nil, err
	}
	return rep, nil
}

// @Description 信息发布
// @Author J 2022/4/12 1:57 PM
func InfoPub(info map[string]interface{}) {
	tmp := info["appendInfo"].(map[string]interface{})
	saveMap := make(map[string]interface{})
	jyMap := make(map[string]interface{})
	for _, f := range InfoFields {
		if tmp[f] == nil {
			continue
		}
		if f == "budget" || f == "amount" {
			saveMap[SaveFields[f]] = util.Float64All(tmp[f])
			jyMap[f] = util.Float64All(tmp[f])
		} else if f == "industry" {
			// topscopeclass/subcopeclass
			//if s := util.ObjToString(tmp[f]); s != "" {
			//
			//	for _, s2 := range strings.Split(s, ",") {
			//		arr := strings.Split(s2, "_")
			//		// todo
			//	}
			//}
		} else if f == "winner" {
			if s := util.ObjToString(tmp[f]); s != "" {
				s = strings.ReplaceAll(s, "，", ",") //中文变英文
				saveMap[SaveFields[f]] = s
				saveMap[f] = s
				jyMap[f] = s
				jyMap[SaveFields[f]] = s
			}
		} else if f == "attach" {
			s := util.ObjToString(tmp[f])
			atts := map[string]interface{}{}
			if err := json.Unmarshal([]byte(s), &atts); err != nil {
				log.Error("data Unmarshal Failed：", log.Field("error", err))
			}
			saveMap[SaveFields[f]] = atts
			//for _, a := range atts {
			//	a1 := a.(map[string]interface{})
			//	a1["filename"] = util.ObjToString(a1["fileName"])
			//	delete(a1, "fileName")
			//	a1[""] = a[""]
			//}
		} else if f == "discern_attach" {
			if s := util.ObjToString(tmp[f]); s != "" {
				atts := map[string]interface{}{}
				if err := json.Unmarshal([]byte(s), &atts); err != nil {
					log.Error("data Unmarshal Failed：", log.Field("error", err))
				}
				saveMap[SaveFields[f]] = atts
			}

		} else {
			if s := util.ObjToString(tmp[f]); s != "" {
				saveMap[SaveFields[f]] = s
				if SaveFields[f] == "buyer" || SaveFields[f] == "buyerperson" || SaveFields[f] == "buyertel" ||
					SaveFields[f] == "area" || SaveFields[f] == "city" {
					jyMap[SaveFields[f]] = s
				}
			}
		}

	}
	now := time.Now()
	saveMap["comeintime"] = now.Unix()
	saveMap["publishtime"] = now.Unix()
	saveMap["contenthtml"] = tmp["detail"]
	saveMap["site"] = "剑鱼信息发布平台"
	saveMap["channel"] = "公告"
	saveMap["spidercode"] = "a_jyxxfbpt_gg"
	saveMap["extracttype"] = 0
	saveMap["areaval"] = 0
	saveMap["infoformat"] = 1
	saveMap["dataging"] = 0
	saveMap["buyerhint"] = util.IntAll(tmp["contact_overt"])
	_id := bson.NewObjectIdWithTime(now)
	saveMap["href"] = fmt.Sprintf(JyUrl, util.CommonEncodeArticle("content", mongodb.BsonIdToSId(_id)))
	saveMap["competehref"] = "#"
	saveMap["jyfb_data"] = jyMap
	saveMap["jyfbid"] = tmp["_id"]

	MgoBid.SaveByOriID(BidColl, saveMap)

}

// @Description 附件处理完成队列
// @Author J 2022/4/13 3:29 PM
func taskAtts(obj interface{}) {
	result := obj.(*pb.FileResponse)
	util.Info(result.String())
	otherMap := make(map[string]interface{})
	if err := json.Unmarshal([]byte(result.Other), &otherMap); err != nil {
		log.Error("data Unmarshal Failed：", log.Field("error", err))
	}
	atts := make(map[string]interface{})
	atts_text := make(map[string]interface{})
	for i, r := range result.Result {
		at := make(map[string]interface{})
		text := make(map[string]interface{})
		at["state"] = r.ErrorState
		if r.ErrorState == "200" {
			at["sensitive"] = WordsIdentify(r.TextContent)
			text["file_name"] = r.FileName
			text["attach_url"] = r.TextUrl
		}
		atts[strconv.Itoa(i)] = at
		if len(text) > 0 {
			atts_text[strconv.Itoa(i)] = text
		}
	}
	attsJson, _ := json.Marshal(atts)
	attsTextJson, _ := json.Marshal(atts_text)
	// 直接调用剑鱼接口
	req := &pb.SensitiveRequest{
		Id:          util.ObjToString(otherMap["id"]),
		MsgType:     util.ObjToString(otherMap["msgType"]),
		Action:      util.ObjToString(otherMap["action"]),
		Title:       strings.Split(util.ObjToString(otherMap["title"]), ","),
		Detail:      strings.Split(util.ObjToString(otherMap["detail"]), ","),
		Attachments: string(attsJson),
		AttachTxt:   string(attsTextJson),
	}
	JyRpcSensitive(req)
}

// @Description 敏感词识别完成调用剑鱼接口
// @Author J 2022/4/13 11:16 AM
func JyRpcSensitive(req *pb.SensitiveRequest) {
	conn := JyRpcClient.Conn()
	defer conn.Close()
	jyIntf := pb.NewJyInterfaceClient(conn)
	resp, err := jyIntf.SensitiveMethod(context.Background(), req)
	if err != nil {
		log.Error(err.Error())
	}
	log.Info(resp.String())
}

// @Description 信息删除(es、bidding、extract、project)
// @Author J 2022/4/8 4:37 PM:00
func DelMethod(res string) {
	q := map[string]interface{}{"_id": mongodb.StringTOBsonId(res)}
	b := MgoBid.Del(BidColl, q)
	if !b {
		log.Error(" bidding del fail...")
	}
	b = MgoExt.Del(ExtColl, q)
	if !b {
		log.Error(" extract del fail...")
	}
	Es.DelById(Index, Itype, res)
	Es.DelById(IndexAll, Itype, res)
	project := Sysconfig["project"].(map[string]interface{})
	by, _ := json.Marshal(map[string]interface{}{
		"infoid": res,
		"stype":  "deleteInfo",
	})
	addr := &net.UDPAddr{
		IP:   net.ParseIP(project["addr"].(string)),
		Port: util.IntAll(project["port"]),
	}
	log.Debug(string(by))
	_ = UdpClient.WriteUdp(by, util.OP_TYPE_DATA, addr)
}