浏览代码

feat:新增项目标识

xuzhiheng 3 年之前
父节点
当前提交
b0adacd732
共有 2 个文件被更改,包括 24 次插入0 次删除
  1. 1 0
      src/util/util.go
  2. 23 0
      src/util/utiltag.go

+ 1 - 0
src/util/util.go

@@ -17,6 +17,7 @@ type SimpleEncrypt struct {
 }
 
 var SEPreview = &SimpleEncrypt{Key: "topnet2015topnet2015"}
+var SE2 = qu.SimpleEncrypt{Key: "topJYBX2019"}
 
 //数据类型转换
 func GetPostForm(r *http.Request) map[string]interface{} {

+ 23 - 0
src/util/utiltag.go

@@ -234,6 +234,11 @@ func UtilEsFind1(tags map[string]interface{}) (error, int64) {
 					} else {
 						log.Println("id", item["info_id"], "bidding为空")
 					}
+					//项目id
+					projectId := GetProjectId(info_id)
+					if projectId != "" {
+						item["projectId"] = SE2.EncodeString(projectId)
+					}
 					datas = append(datas, item)
 				} else {
 					log.Println("id", item["_id"], "信息为空")
@@ -899,3 +904,21 @@ func KeyWordToData(types, item, key string, keyWord []string) []string {
 	}
 	return keyWord
 }
+
+func GetProjectId(id string) string {
+	query := `{"query": {"bool": {"must": [{"term": {"projectset.ids": "%s"}}],"must_not": [],"should": []}}}`
+	querys := fmt.Sprintf(query, id)
+	projectId := ""
+	projectIdArr := []string{}
+	data := Es.Get("projectset", "projectset", querys)
+	if data != nil && *data != nil && len(*data) > 0 {
+		for _, v := range *data {
+			projectIdArr = append(projectIdArr, util.ObjToString(v["_id"]))
+		}
+		projectId = strings.Join(projectIdArr, ",")
+		log.Println("projectId", projectId)
+	} else {
+		log.Println("ES未查到项目id", id)
+	}
+	return projectId
+}