浏览代码

fix:方法调用修改

duxin 2 年之前
父节点
当前提交
4633e41db4
共有 2 个文件被更改,包括 27 次插入10 次删除
  1. 16 6
      common/advancedProject.go
  2. 11 4
      common/customizedAnalysis.go

+ 16 - 6
common/advancedProject.go

@@ -12,16 +12,23 @@ import (
 	"time"
 )
 
-func (mae *MarketAnalysisEntity) AdvancedProject() map[string]interface{} {
-	finalSql := fmt.Sprintf(mae.GetCommonQuerySqlWithAggs(), projectsNumber, mae.Size, projectsSort)
-	log.Println("定制化分析报告sql查询:", finalSql)
+func AdvancedProject(userid, keyWords string) (map[string]interface{}, error) {
+	mae := new(MarketAnalysisEntity)
+	if key := KeyWordFormat(userid, keyWords); key != "" {
+		if err := json.Unmarshal([]byte(key), &mae.FormatParam.KeysItems); err != nil {
+			log.Println("关键词格式化失败")
+			return nil, err
+		}
+	}
 	mae.Types = 2
 	mae.FormatParam.SubType = []string{"拟建", "采购意向"}
 	mae.Size = 2
 	mae.FormatParam.STime = time.Now().AddDate(0, -3, 0).Unix()
+	finalSql := fmt.Sprintf(mae.GetCommonQuerySqlWithAggs(), projectsNumber, mae.Size, projectsSort)
+	log.Println("超前项目es查询:", finalSql)
 	res, _, data := util.GetAggs("bidding", "bidding", finalSql)
 	if res == nil || len(res) == 0 || data == nil || len(data) == 0 {
-		return nil
+		return nil, nil
 	}
 	var thisRow SuperProjects
 	for name, object := range res {
@@ -43,7 +50,7 @@ func (mae *MarketAnalysisEntity) AdvancedProject() map[string]interface{} {
 		redisData[v.Key] = v.DocCount
 	}
 	if len(redisData) == 0 {
-		return nil
+		return nil, nil
 	}
 	resData := make([]map[string]interface{}, len(data))
 	for i, v := range data {
@@ -52,7 +59,10 @@ func (mae *MarketAnalysisEntity) AdvancedProject() map[string]interface{} {
 		}
 	}
 	log.Println("AdvancedProject redis data :", resData)
-	return nil
+	return map[string]interface{}{
+		"projectTop2":  resData,
+		"projectCount": resData,
+	}, nil
 }
 
 func KeyWordFormat(userid, keyWords string) string {

+ 11 - 4
common/customizedAnalysis.go

@@ -11,16 +11,23 @@ import (
 
 var Analysis = []string{"<10万", "10万-50万", "50万-100万", "100万-500万", "500万-1000万", "1000万-1亿", "≥1亿"}
 
-func (mae *MarketAnalysisEntity) PotentialCustomizeAnalysis() map[string]interface{} {
+func PotentialCustomizeAnalysis(userid, keyWords string) (map[string]interface{}, error) {
+	mae := new(MarketAnalysisEntity)
+	if key := KeyWordFormat(userid, keyWords); key != "" {
+		if err := json.Unmarshal([]byte(key), &mae.FormatParam.KeysItems); err != nil {
+			log.Println("关键词格式化失败")
+			return nil, err
+		}
+	}
 	var aggs []string
 	aggs = append(aggs, aggs_buyerclass, buyer_procurement_scale, winner_procurement_scale)
 	mae.Types = 1
 	mae.FormatParam.STime = time.Now().AddDate(-1, 0, 0).Unix()
 	finalSql := fmt.Sprintf(mae.GetCommonQuerySqlWithAggs(), strings.Join(aggs, ","), mae.Size, "")
-	log.Println("定制化分析报告sql查询:", finalSql)
+	log.Println("定制化分析报告es查询:", finalSql)
 	res, _, _ := util.GetAggs("projectset", "projectset", finalSql)
 	if res == nil || len(res) == 0 {
-		return nil
+		return nil, nil
 	}
 	thisRow := Aggregation{}
 	for name, object := range res {
@@ -50,7 +57,7 @@ func (mae *MarketAnalysisEntity) PotentialCustomizeAnalysis() map[string]interfa
 	CustomerDistribute(thisRow, rMap) //客户分布
 	BuyerAnalysis(thisRow, rMap)      //采购单位分布
 	WinningAnalysis(thisRow, rMap)    //中标单位分布
-	return rMap
+	return rMap, nil
 }
 
 // CustomerDistribute 客户分布