timed_tasks/zhima/main.go

package main

import (
	"encoding/json"
	"fmt"
	"log"
	"strings"
	"sync"

	"github.com/xuri/excelize/v2"
	util "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/elastic"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
)

var (
	// JianZhiLabel = make([]CreditLabel, 0)
	ZhimaMap = map[string]string{}
	MgoT     *mongodb.MongodbSim //测试环境链接
	Es       *elastic.Elastic
)

func Init() {
	f, err := excelize.OpenFile("./jianyu_zhima_0829.xlsx")
	if err != nil {
		fmt.Println(err)
		return
	}
	defer func() {
		f.Save()
		if err := f.Close(); err != nil {
			fmt.Println(err)
		}
	}()

	//2.专项债详情
	rows, err := f.GetRows("0829")
	if err != nil {
		fmt.Println(err)
		return
	}

	for i := 1; i < len(rows); i++ {
		// data := CreditLabel{
		// 	ZhimaToptype: rows[i][0],
		// 	ZhimaSubtype: rows[i][1],
		// 	ZhimaName:    rows[i][2],
		// 	ZhimaValue:   rows[i][4],
		// }
		// JianZhiLabel = append(JianZhiLabel, data)
		ZhimaMap[rows[i][2]] = rows[i][1]
	}

	//测试环境
	//MgoT = &mongodb.MongodbSim{
	//	MongodbAddr: "192.168.3.149:27102",
	//	//MongodbAddr: "192.168.3.206:27002",
	//	DbName:   "mixdata",
	//	Size:     10,
	//	UserName: "root",
	//	Password: "root",
	//	//Direct:      true,
	//}
	//MgoT.InitPool()
	//
	////测试环境es
	//Es = &elastic.Elastic{
	//	S_esurl: "http://192.168.3.149:9201",
	//	//S_esurl:  "http://172.17.4.184:19805",
	//	I_size:   10,
	//	Username: "",
	//	Password: "",
	//}
	//Es.InitElasticSize()

	//正式环境
	MgoT = &mongodb.MongodbSim{
		MongodbAddr: "172.17.189.140:27080",
		//MongodbAddr: "127.0.0.1:27083",
		Size:     10,
		DbName:   "mixdata",
		UserName: "SJZY_RWbid_ES",
		Password: "SJZY@B4i4D5e6S",
		//Direct:   true,
	}
	MgoT.InitPool()

	Es = &elastic.Elastic{
		//S_esurl: "http://127.0.0.1:19908",
		S_esurl:  "http://172.17.4.184:19908",
		I_size:   10,
		Username: "jybid",
		Password: "Top2023_JEB01i@31",
	}
	Es.InitElasticSize()

}

func main() {
	Init()

	appID := "2021003111633206"
	privateKeyPEM := "-----BEGIN RSA PRIVATE KEY-----\nMIIEvAIBADANBgkqhkiG9w0BAQEFAASCBKYwggSiAgEAAoIBAQCaJnmz3N1Z43jxtmZJmPkInl3FbfudxjicThHj+Hkndp+muzZXlAPcWoYTFLAfVa6xg9xAc58fWTPUGOpouh0ZDEP8TGvD784AljDNwRXPI9+afjcxN8gzsNPlqA6KVkAu5u5uEI1HHQgtdyRoCEnXgQRXj6DqAs9olbx8HrTxHjZzjPd+HXv+v4na3zAvZsvPdhLB2ggpVW6uQN7ur8C+hVpB3ZO+gGVFHy3m1t08XgscljyOA9Vpk4T8TnKZqDvnI6ehWFCGeD11yoe3ipHJJdP/+u9/tXiKhm29mJvcM9C6gab6BOQABByQPb2nPFFyfPglbRYrYxgKp/eXUQTzAgMBAAECggEAO6BCVyZJoKlycNzC+Bjpy0Gmt+Vw/HheEWfJV8sW4gLqgk3MGVgxnhyN2/eWoKg96IA+w9d1Z2TTvWjd0yCafMPX0FGnJaUuo990hUH4s6UGqvXDV5BiKkxtylmVNkFilkHRXFWesDE0ppF9TZhXJ7rJ+Syfa6UmoSe6PrCKAPG4FD8MH1ODj5b42l/TRIFM948nTtTLFAFgd77P5ULBCKVfBB5V9XdKhLRdwG9RxugWwkxuCiMOtmk45DzA/PQgvkIzEViD/2NJX67/S88B0G63zdQizJrvxsLp9fThmjhlo/+CaF+a+nfIzg81TfQHlRuTjiYc7iE9n5K0sU82wQKBgQDydI9pEjkPH+ZEixEymdx8D4workBfa+GXgobJ2UwatqCJOCYqZJR0MhjTVeDJiiTGFiemNyZHT0WFQedAQ/O9mkRf0MRhzR6ZOiMnCG5pSSOKdmQXoqjOqcMzdgWBQRol0eLB/vto8M3bpt1VautR0qQOR7BrFsj57K3asofikQKBgQCiwwie5gyCAYB1gkp4QkLmQDSTJdZaTlzzx2oa2/ZB4Pg6wWWNJIxEfSwtuGVH4jGJl0Eer2YpHzaY7f2NfOWJCpNI/aRhM/RilQsFXTox/5bqlbAfspPMqSHnry2OgbZb9CRF3jH7kt65390SKde4A4JNE+jJ0Xb07Tzcr7+pQwKBgCDJ8KcWBs9r0flNEVtVgYLpm1vZDO2aqlySE2Zy0579OcK07Dxpc8YvDDpwb8DxxL1hO6V36d5U8yLtYbfIOMtYdIy8sAJF+jBgs2ZQ9pWYhYzxK+IiZk/kl3h7Ko1rX3zj4JW5xQhRIJXnACHQHtjm8rbXrR0ebLmX738XC9dxAoGAQfUDQ1E6dFyjb/syrnU7ow8VvyOwL7xd0rRW/TdyHdRLA5Rn0MCkzwRiqMx2zqCNLpLK4uJ1ZeiuOV8IRxP7+IL+mK+gs3tYZ7K6Wpwi/yvyPnIWg2Hcny1S/g5xSbSq76yKHn4ihy5Hb8+WSMwIDRvu68lLpq1ppWuWJGBcLP0CgYAnGtHlFY887zSI0aRdzV3EYm/sPnmYy0TWIUVLuGL8kkprcSIW2yRxx1GtpB+3A7cPu61WwRbiswJA6VmQHvn8f9ds81+aiO5KQLMk4MbrTwWhWLDpP4BxjWANVzPr19ZwmDJcYkh7YUI2iLqH7dXho5k62OYCM1T53xlql6HDcg==\n-----END RSA PRIVATE KEY-----"
	//epCertNo := "914400007211976169" //对应企业库 credit_no
	sceneCode := "JIANYU_LABEL"

	//1.查询qyxy 数据
	defer util.Catch()
	sess := MgoT.GetMgoConn()
	defer MgoT.DestoryMongoConn(sess)
	it := sess.DB("mixdata").C("qyxy_std").Find(nil).Select(nil).Sort("_id").Iter()

	fmt.Println("taskRun 开始")
	count := 0
	ch := make(chan bool, 5)
	wg := &sync.WaitGroup{}
	for tmp := make(map[string]interface{}); it.Next(&tmp); count++ {
		if count%1000 == 0 {
			log.Println("current", count)
		}
		// 存在芝麻标签，直接跳过
		//if _, ok := tmp["zhima_labels"]; ok {
		//	continue
		//}

		if util.ObjToString(tmp["credit_no"]) == "" {
			continue
		}

		ch <- true
		wg.Add(1)
		go func(tmp map[string]interface{}) {
			defer func() {
				<-ch
				wg.Done()
			}()
			//
			id := mongodb.BsonIdToSId(tmp["_id"])
			epCertNo := util.ObjToString(tmp["credit_no"])
			createLabels := getCreditLabel(appID, privateKeyPEM, epCertNo, sceneCode)
			if len(createLabels) > 0 {
				updateEs := map[string]interface{}{
					"zhima_labels": convertToInterfaceSliceEs(createLabels),
				}

				updateMgo := map[string]interface{}{
					"zhima_labels": convertToInterfaceSliceMgo(createLabels),
				}
				where := map[string]interface{}{
					"_id": tmp["_id"],
				}
				MgoT.Update("qyxy_std", where, map[string]interface{}{"$set": updateMgo}, true, false)
				err := Es.UpdateDocument("qyxy", id, updateEs)
				if err != nil && err.Error() != "Document not updated: noop" {
					log.Println("qyxy es update err", err, id)
				}
			}

		}(tmp)
		tmp = make(map[string]interface{})
	}
	wg.Wait()
	log.Println("企业芝麻标签结束")
}

// getCreditLabel 获取芝麻企业信用企业标签
func getCreditLabel(appID, privateKeyPEM, certNo, sceneCode string) (createLabels []CreditLabel) {
	BizContent := BizContentForLabelQuery{
		EPCertNo:  certNo,
		ShowFlag:  true,
		SceneCode: sceneCode,
	}

	curl := "https://openapi.alipay.com/gateway.do"
	method := "zhima.credit.ep.acceptance.label.query"
	bizbyte, _ := json.Marshal(BizContent)
	priKey, err := parsePrivateKey(privateKeyPEM)
	if err != nil {
		log.Fatalln("parsePrivateKey err", err)
	}

	resMap, err := DoRequest(appID, string(bizbyte), curl, method, priKey)
	if err != nil {
		log.Fatalln("request err", err)
	}
	responseData := util.ObjToMap((*resMap)["zhima_credit_ep_acceptance_label_query_response"])
	if util.ObjToString(((*responseData)["code"])) == "10000" && util.ObjToString(((*responseData)["msg"])) == "Success" {
		if (*responseData)["label_content"] == nil {
			//log.Println("请求出错，没有返回企业标签内容模型", BizContent)
		} else {
			if labelContent, ok := (*responseData)["label_content"].([]interface{}); ok {
				for _, v := range labelContent {
					label := util.ObjToMap(v)
					labelName := util.ObjToString((*label)["label_name"])         //外显
					labelCategory := util.ObjToString((*label)["label_category"]) //一级分类
					labelCode := util.ObjToString((*label)["label_code"])
					labelDe := util.ObjToString((*label)["label_definition"])

					newName, subtype := getLabel(labelName)
					data := CreditLabel{}
					data.ZhimaCode = labelCode
					data.ZhimaDefinition = labelDe
					data.ZhimaToptype = labelCategory
					data.ZhimaSubtype = subtype
					data.ZhimaName = newName
					data.ZhimaValue = labelName
					if subtype != "" {
						createLabels = append(createLabels, data)
					}
				}
			}
		}
	} else {
		log.Println("请求失败", BizContent, (*responseData))
	}

	return
}

// getLabel 根据labelTop  labelName 获取对应芝麻标签
func getLabel(labelName string) (newName, subtype string) {
	// for _, v := range JianZhiLabel {
	// 	if v.ZhimaToptype == labelTop && v.ZhimaName == labelName {
	// 		return v
	// 	}
	// }
	newName = getNewName(labelName)
	subtype = ""
	if ZhimaMap[newName] != "" {
		subtype = ZhimaMap[newName]
	}
	return
}

func getNewName(name string) string {
	if strings.Contains(name, "年企业信用") {
		return "芝麻企业信用等级连续情况"
	} else if strings.Contains(name, "芝麻企业信用等级") {
		return "芝麻企业信用等级"
	} else if strings.Contains(name, "企业信用等级") {
		return "企业信用等级_个体"
	} else if strings.Contains(name, "公开项目中标") {
		return "年度公开项目中标"
	} else if ZhimaMap[name] != "" {
		return ZhimaMap[name]
	} else if strings.Contains(name, "年A级纳税人") {
		return "A级纳税人连续年限"
	} else if strings.Contains(name, "年A级纳税人") {
		return "A级纳税人连续年限"
	} else if strings.Contains(name, "融资") && strings.Contains(name, "获") {
		return "获得融资"
	} else if strings.Contains(name, "融资") && strings.Contains(name, "轮") {
		return "融资轮次"
	}
	for k, _ := range ZhimaMap {
		if strings.Contains(name, k) {
			return k
		}
	}
	return ""
}

func convertToInterfaceSliceEs(labels []CreditLabel) []map[string]interface{} {
	result := make([]map[string]interface{}, len(labels))
	for i, label := range labels {
		result[i] = map[string]interface{}{
			"zhima_toptype":    label.ZhimaToptype,
			"zhima_subtype":    label.ZhimaSubtype,
			"zhima_name":       label.ZhimaName,
			"zhima_definition": label.ZhimaDefinition,
			"zhima_value":      label.ZhimaValue,
		}
	}
	return result
}

func convertToInterfaceSliceMgo(labels []CreditLabel) []map[string]interface{} {
	result := make([]map[string]interface{}, len(labels))
	for i, label := range labels {
		result[i] = map[string]interface{}{
			"zhima_toptype":    label.ZhimaToptype,
			"zhima_subtype":    label.ZhimaSubtype,
			"zhima_name":       label.ZhimaName,
			"zhima_code":       label.ZhimaCode,
			"zhima_definition": label.ZhimaDefinition,
			"zhima_value":      label.ZhimaValue,
		}
	}
	return result
}