jyMicroservices/jyBXBase/rpc/model/newestBidding.go

package model

import (
      MC "app.yhyue.com/moapp/jybase/common"
      ME "app.yhyue.com/moapp/jybase/encrypt"
      elastic "app.yhyue.com/moapp/jybase/esv1"
      "app.yhyue.com/moapp/jybase/mongodb"
      "app.yhyue.com/moapp/jybase/mysql"
      "fmt"
      "github.com/zeromicro/go-zero/core/logx"
      "go.mongodb.org/mongo-driver/bson/primitive"
      "jyBXBase/rpc/bxbase"
      IC "jyBXBase/rpc/init"
      "log"
      "strings"
      "time"
)

const (
      search_index        = "bidding"
      search_type         = "bidding"
      mongodb_fields      = `{"_id":1,"area":1,"publishtime":1,"s_subscopeclass":1,"subtype":1,"title":1,"toptype":1,"type":1, "buyerclass":1,"budget":1,"bidamount":1,"s_winner":1,"bidopentime":1,"buyer":1,"projectname":1,"spidercode":1,"site":1}`
      query               = `{"query":{"terms":{"_id":["%s"]}},"_source":["_id","area", "publishtime", "s_subscopeclass", "subtype", "title", "toptype", "type", "buyerclass","bidamount","budget","projectname","buyer","bidopentime","s_winner","filetext","spidercode","site"],"from":0,"size":%d}`
      multi_match         = `{"multi_match": {"query": %s,"type": "phrase", "fields": ["title"]}}`
      query_bool_must     = `{"terms":{"%s":[%s]}}`
      query_bool_must_and = `{"bool":{"must":[%s],"must_not":[%s]}}`
      search_field        = `"_id","area", "publishtime", "s_subscopeclass", "subtype", "title", "toptype", "type", "buyerclass","bidamount","budget","projectname","buyer","bidopentime","s_winner","filetext","isValidFile","spidercode","site"`
      query_city_hkeys    = `{"query":{"bool":{"must":[%s],"should":[%s],"minimum_should_match": 1}},"highlight": {"pre_tags": ["<a>"],"post_tags": ["</a>"],"fields": {"title": {"fragment_size": 0,"number_of_fragments": 1}}},"_source":[` + search_field + `],"sort":[{"publishtime":"desc"},{"budget":"desc"}],"from":0,"size":20}`
      Pushbidding         = "pushbidding"
      Province            = "province"
)

type NewestInfo struct {
TableName string
UserId    string
MysqlDb   *mysql.Mysql
}

var mysqlTables = map[string]string{
      "f": "pushsubscribe",
      "v": "pushsubscribe",
      "m": "pushmember",
      "e": "pushentniche",
}

func GetNewestInfo(userId, userType string) *NewestInfo {
      nt := &NewestInfo{
	  UserId:    userId,
	  TableName: mysqlTables[userType],
	  MysqlDb:   IC.BaseServiceMysql,
      }
      return nt
}
func (n *NewestInfo) GetPushHistory() (res []*bxbase.NewestList) {
      findSQL := "select b.infoid,a.matchkeys,b.attachment_count,b.budget,b.bidamount from %s a LEFT JOIN %s b ON a.bidid = b.id LEFT JOIN %s p on b.area=p.id or  b.city=p.id  where a.userid = '%s'  and a.date >= %d order by a.id desc limit 50"
      findSQL = fmt.Sprintf(findSQL, n.TableName, Pushbidding, Province, n.UserId, time.Now().AddDate(0, 0, -7).Unix())
      log.Println(findSQL)
      list := n.MysqlDb.SelectBySql(findSQL)
      if len(*list) > 0 && list != nil {
	  m := map[string]bool{}
	  es_ids := []string{}
	  infos := map[string]*bxbase.NewestList{}
	  for _, v := range *list {
	        infoId := MC.ObjToString(v["infoid"])
	        if m[infoId] {
		    continue
	        }
	        es_ids = append(es_ids, infoId)
	        m[MC.ObjToString(v["infoid"])] = true
	        //
	        infos[infoId] = &bxbase.NewestList{
		    Id:         ME.EncodeArticleId2ByCheck(MC.ObjToString(v["infoid"])),
		    Matchkeys:  MC.ObjToString(v["matchkeys"]),
		    Budget:     MC.Int64All(v["budget"]),
		    Bidamount:  MC.Int64All(v["bidamount"]),
		    FileExists: MC.Int64All(v["attachment_count"]) > 0,
	        }
	  }
	  if len(es_ids) > 0 {
	        list := elastic.Get(search_index, search_type, fmt.Sprintf(query, strings.Join(es_ids, `","`), len(es_ids)))
	        if list != nil {
		    for _, v := range *list {
			_id := MC.ObjToString(v["_id"])
			bn := infos[_id]
			bn.Title = MC.ObjToString(v["title"])
			bn.PublishTime = MC.Int64All(v["publishtime"])
			bn.Subtype = MC.If(v["subtype"] != nil, MC.ObjToString(v["subtype"]), MC.ObjToString(v["toptype"])).(string)
			bn.Area = MC.If(MC.ObjToString(v["area"]) == "A", "全国", MC.ObjToString(v["area"])).(string)
			bn.Buyerclass = MC.ObjToString(v["buyerclass"])
			bn.City = MC.ObjToString(v["city"])
			bn.Industry = MC.If(MC.ObjToString(v["s_subscopeclass"]) != "", strings.Split(strings.Split(MC.ObjToString(v["s_subscopeclass"]), ",")[0], "_")[0], "").(string)
			bn.SpiderCode = MC.ObjToString(v["spidercode"])
			bn.Site = MC.ObjToString(v["site"])
		    }
	        }
	  }
	  //mongodb bidding
	  mgo_ids := []primitive.ObjectID{}
	  for _, v := range es_ids {
	        if infos[v].Title == "" {
		    mgo_ids = append(mgo_ids, mongodb.StringTOBsonId(v))
	        }
	  }
	  if len(mgo_ids) > 0 {
	        list, ok := IC.MgoBidding.Find("bidding", map[string]interface{}{"_id": map[string]interface{}{"$in": mgo_ids}}, nil, mongodb_fields, false, -1, -1)
	        if ok && *list != nil {
		    for _, v := range *list {
			_id := mongodb.BsonIdToSId(v["_id"])
			bn := infos[_id]
			bn.Title = MC.ObjToString(v["title"])
			bn.PublishTime = MC.Int64All(v["publishtime"])
			bn.Subtype = MC.If(v["subtype"] != nil, MC.ObjToString(v["subtype"]), MC.ObjToString(v["toptype"])).(string)
			bn.Area = MC.If(MC.ObjToString(v["area"]) == "A", "全国", MC.ObjToString(v["area"])).(string)
			bn.Buyerclass = MC.ObjToString(v["buyerclass"])
			bn.City = MC.ObjToString(v["city"])
			bn.Industry = MC.If(MC.ObjToString(v["s_subscopeclass"]) != "", strings.Split(strings.Split(MC.ObjToString(v["s_subscopeclass"]), ",")[0], "_")[0], "").(string)
			bn.SpiderCode = MC.ObjToString(v["spidercode"])
			bn.Site = MC.ObjToString(v["site"])
		    }
	        }
	  }
	  //mongodb bidding_back
	  mgo_back_ids := []primitive.ObjectID{}
	  for _, v := range mgo_ids {
	        if infos[mongodb.BsonIdToSId(v)].Title == "" {
		    mgo_back_ids = append(mgo_back_ids, v)
	        }
	  }
	  if len(mgo_back_ids) > 0 {
	        list, ok := IC.MgoBidding.Find("bidding_back", map[string]interface{}{"_id": map[string]interface{}{"$in": mgo_back_ids}}, nil, mongodb_fields, false, -1, -1)
	        if ok && *list != nil {
		    for _, v := range *list {
			_id := mongodb.BsonIdToSId(v["_id"])
			bn := infos[_id]
			bn.Title = MC.ObjToString(v["title"])
			bn.PublishTime = MC.Int64All(v["publishtime"])
			bn.Subtype = MC.If(v["subtype"] != nil, MC.ObjToString(v["subtype"]), MC.ObjToString(v["toptype"])).(string)
			bn.Area = MC.If(MC.ObjToString(v["area"]) == "A", "全国", MC.ObjToString(v["area"])).(string)
			bn.Buyerclass = MC.ObjToString(v["buyerclass"])
			bn.City = MC.ObjToString(v["city"])
			bn.Industry = MC.If(MC.ObjToString(v["s_subscopeclass"]) != "", strings.Split(strings.Split(MC.ObjToString(v["s_subscopeclass"]), ",")[0], "_")[0], "").(string)
			bn.SpiderCode = MC.ObjToString(v["spidercode"])
			bn.Site = MC.ObjToString(v["site"])
		    }
	        }
	  }
	  //
	  for _, v := range infos {
	        res = append(res, v)
	  }
      }
      return
}

//根据定位或者搜索历史 查es
func NewestQuery(city, keys string) (str string) {
      var musts, bools []string
      if keys != "" {
	  for _, v := range strings.Split(keys, ",") {
	        keys := strings.Split(v, " ") //历史搜索 空格划分
	        must_tmp := []string{}
	        for _, key := range keys {
		    must_tmp = append(must_tmp, fmt.Sprintf(multi_match, "\""+key+"\""))
	        }
	        bools = append(bools, fmt.Sprintf(query_bool_must_and, strings.Join(must_tmp, ","), ""))
	  }

      }
      if city != "" {
	  musts = append(musts, fmt.Sprintf(query_bool_must, "city", `"`+city+`"`))
      }
      str = fmt.Sprintf(query_city_hkeys, strings.Join(musts, ","), strings.Join(bools, ","))
      logx.Info("str:", str)
      return
}

//es查询
func NewestES(doSearchStr string) (res []*bxbase.NewestList) {
      list := elastic.Get(search_index, search_type, doSearchStr)
      if list != nil && len(*list) > 0 {
	  for _, v := range *list {
	        _id := mongodb.BsonIdToSId(v["_id"])
	        isValidFile, _ := v["isValidFile"].(bool)
	        res = append(res, &bxbase.NewestList{
		    Id:          ME.EncodeArticleId2ByCheck(_id),
		    Title:       MC.ObjToString(v["title"]),
		    Subtype:     MC.If(v["subtype"] != nil, MC.ObjToString(v["subtype"]), MC.ObjToString(v["toptype"])).(string),
		    Area:        MC.If(MC.ObjToString(v["area"]) == "A", "全国", MC.ObjToString(v["area"])).(string),
		    Buyerclass:  MC.ObjToString(v["buyerclass"]),
		    City:        MC.ObjToString(v["city"]),
		    Industry:    MC.If(MC.ObjToString(v["s_subscopeclass"]) != "", strings.Split(strings.Split(MC.ObjToString(v["s_subscopeclass"]), ",")[0], "_")[0], "").(string),
		    Budget:      MC.Int64All(v["budget"]),
		    Bidamount:   MC.Int64All(v["bidamount"]),
		    FileExists:  isValidFile, //附件
		    PublishTime: MC.Int64All(v["publishtime"]),
		    Site:        MC.ObjToString(v["site"]),
		    SpiderCode:  MC.ObjToString(v["spidercode"]),
	        })
	  }
      }
      return
}