|
@@ -12,30 +12,30 @@ import (
|
|
|
)
|
|
|
|
|
|
var (
|
|
|
- MongoTool *mongodb.MongodbSim
|
|
|
+ MongoTool *mongodb.MongodbSim
|
|
|
|
|
|
- PnameLength int
|
|
|
- PurchasingLength int
|
|
|
+ PnameLength int
|
|
|
+ PurchasingLength int
|
|
|
)
|
|
|
|
|
|
func init() {
|
|
|
+ MongoTool = &mongodb.MongodbSim{
|
|
|
+ MongodbAddr: "172.17.4.85:27080", // 172.17.4.85:27080
|
|
|
+ Size: 10,
|
|
|
+ DbName: "qfw",
|
|
|
+ }
|
|
|
+ MongoTool.InitPool()
|
|
|
//MongoTool = &mongodb.MongodbSim{
|
|
|
- // MongodbAddr: "172.17.4.85:27080", // 172.17.4.85:27080
|
|
|
+ // MongodbAddr: "172.17.4.187:27082,172.17.145.163:27083",
|
|
|
// Size: 10,
|
|
|
// DbName: "mixdata",
|
|
|
+ // UserName: "SJZY_RWMIX_Other",
|
|
|
+ // Password: "SJZY@M34I6x7D9ata",
|
|
|
//}
|
|
|
//MongoTool.InitPool()
|
|
|
- MongoTool = &mongodb.MongodbSim{
|
|
|
- MongodbAddr: "172.17.4.187:27082,172.17.145.163:27083",
|
|
|
- Size: 10,
|
|
|
- DbName: "mixdata",
|
|
|
- UserName: "SJZY_RWMIX_Other",
|
|
|
- Password: "SJZY@M34I6x7D9ata",
|
|
|
- }
|
|
|
- MongoTool.InitPool()
|
|
|
|
|
|
- redis.InitRedis1("qyxy_id=172.17.4.189:8379", 2) // 企业company_id(中标单位)
|
|
|
- //redis.InitRedis1("qyxy_winner=172.17.4.189:8379", 3) // 剑鱼中标企业搜索搜索
|
|
|
+ //redis.InitRedis1("qyxy_id=172.17.4.189:8379", 2) // 企业company_id(中标单位)
|
|
|
+ redis.InitRedis1("qyxy=172.17.4.189:8379", 3) // 剑鱼企业搜索搜索
|
|
|
|
|
|
PnameLength = 500
|
|
|
PurchasingLength = 500
|
|
@@ -43,7 +43,7 @@ func init() {
|
|
|
|
|
|
var name string
|
|
|
|
|
|
-func main1() {
|
|
|
+func main1() {
|
|
|
flag.StringVar(&name, "n", "", "企业名称")
|
|
|
flag.Parse()
|
|
|
if name == "" {
|
|
@@ -63,7 +63,7 @@ func main() {
|
|
|
sess := MongoTool.GetMgoConn()
|
|
|
defer MongoTool.DestoryMongoConn(sess)
|
|
|
|
|
|
- ch := make(chan bool, 2)
|
|
|
+ ch := make(chan bool, 3)
|
|
|
wg := &sync.WaitGroup{}
|
|
|
|
|
|
//q := map[string]interface{}{
|
|
@@ -72,20 +72,14 @@ func main() {
|
|
|
// "$lte": mongodb.StringTOBsonId("5e0a1f000000000000000000"),
|
|
|
// },
|
|
|
//}
|
|
|
- //field := map[string]interface{}{"agency": 1, "buyer": 1, "s_winner": 1, "projectname": 1, "purchasing": 1, "area": 1}
|
|
|
- field := map[string]interface{}{"use_flag": 1, "company_type": 1, "company_name": 1}
|
|
|
- query := sess.DB("mixdata").C("qyxy_std").Find(nil).Select(field).Skip(52035000).Iter()
|
|
|
+ field := map[string]interface{}{"agency": 1, "buyer": 1, "s_winner": 1, "projectname": 1, "purchasing": 1, "area": 1}
|
|
|
+ //field := map[string]interface{}{"use_flag": 1, "company_type": 1, "company_name": 1}
|
|
|
+ query := sess.DB("qfw").C("projectset_20200918").Find(nil).Select(field).Iter()
|
|
|
count := 0
|
|
|
for tmp := make(map[string]interface{}); query.Next(tmp); count++ {
|
|
|
if count%5000 == 0 {
|
|
|
util.Debug("current ---", count)
|
|
|
}
|
|
|
- if util.IntAll(tmp["use_flag"]) > 5 {
|
|
|
- continue
|
|
|
- }
|
|
|
- if strings.Contains(util.ObjToString(tmp["company_type"]), "个体") {
|
|
|
- continue
|
|
|
- }
|
|
|
ch <- true
|
|
|
wg.Add(1)
|
|
|
go func(tmp map[string]interface{}) {
|
|
@@ -93,7 +87,8 @@ func main() {
|
|
|
<-ch
|
|
|
wg.Done()
|
|
|
}()
|
|
|
- redis.PutCKV("qyxy_id", util.ObjToString(tmp["company_name"]), util.ObjToString(tmp["_id"]))
|
|
|
+ //redis.PutCKV("qyxy_id", util.ObjToString(tmp["company_name"]), util.ObjToString(tmp["_id"]))
|
|
|
+ taskinfo1(tmp)
|
|
|
}(tmp)
|
|
|
tmp = make(map[string]interface{})
|
|
|
}
|
|
@@ -104,10 +99,10 @@ func main() {
|
|
|
func taskinfo1(tmp map[string]interface{}) {
|
|
|
if util.ObjToString(tmp["s_winner"]) != "" {
|
|
|
winners := strings.Split(util.ObjToString(tmp["s_winner"]), ",")
|
|
|
- for _, v := range winners{
|
|
|
+ for _, v := range winners {
|
|
|
winnermap := make(map[string]interface{})
|
|
|
- if b, err := redis.Exists("qyxy_winner", v); err == nil && b {
|
|
|
- text := redis.GetStr("qyxy_winner", v)
|
|
|
+ if b, err := redis.Exists("qyxy", v); err == nil && b {
|
|
|
+ text := redis.GetStr("qyxy", v)
|
|
|
err1 := json.Unmarshal([]byte(text), &winnermap)
|
|
|
if err1 != nil {
|
|
|
util.Debug(v, "winner-----map解析异常")
|
|
@@ -124,10 +119,10 @@ func taskinfo1(tmp map[string]interface{}) {
|
|
|
if len(pnameArr) > PnameLength {
|
|
|
pnameArr1 := pnameArr[len(pnameArr)-PnameLength:]
|
|
|
winnermap["bid_projectname"] = pnameArr1
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
winnermap["bid_projectname"] = pnameArr
|
|
|
}
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
pname := []string{util.ObjToString(tmp["projectname"])}
|
|
|
winnermap["bid_projectname"] = pname
|
|
|
}
|
|
@@ -143,10 +138,10 @@ func taskinfo1(tmp map[string]interface{}) {
|
|
|
if len(purs) > PurchasingLength {
|
|
|
purs1 := purs[len(purs)-PurchasingLength:]
|
|
|
winnermap["bid_purchasing"] = purs1
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
winnermap["bid_purchasing"] = purs
|
|
|
}
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
purs := []string{util.ObjToString(tmp["purchasing"])}
|
|
|
winnermap["bid_purchasing"] = purs
|
|
|
}
|
|
@@ -159,7 +154,7 @@ func taskinfo1(tmp map[string]interface{}) {
|
|
|
areas = append(areas, util.ObjToString(tmp["area"]))
|
|
|
}
|
|
|
winnermap["bid_area"] = areas
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
areas := []string{util.ObjToString(tmp["area"])}
|
|
|
winnermap["bid_area"] = areas
|
|
|
}
|
|
@@ -171,11 +166,11 @@ func taskinfo1(tmp map[string]interface{}) {
|
|
|
types = append(types, "中标单位")
|
|
|
}
|
|
|
winnermap["bid_unittype"] = types
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
types := []string{"中标单位"}
|
|
|
winnermap["bid_unittype"] = types
|
|
|
}
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
// bid_projectname
|
|
|
if tmp["projectname"] != nil && tmp["projectname"] != "" {
|
|
|
winnermap["bid_projectname"] = []string{util.ObjToString(tmp["projectname"])}
|
|
@@ -190,16 +185,16 @@ func taskinfo1(tmp map[string]interface{}) {
|
|
|
winnermap["bid_unittype"] = []string{"中标单位"}
|
|
|
}
|
|
|
if len(winnermap) > 0 {
|
|
|
- data , _ := json.Marshal(winnermap)
|
|
|
- redis.Put("qyxy_winner", v, string(data), 0)
|
|
|
+ data, _ := json.Marshal(winnermap)
|
|
|
+ redis.Put("qyxy", v, string(data), 0)
|
|
|
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
if buyer := util.ObjToString(tmp["buyer"]); buyer != "" {
|
|
|
buyermap := make(map[string]interface{})
|
|
|
- if b, err := redis.Exists("qyxy_winner", buyer); err == nil && b {
|
|
|
- text := redis.GetStr("qyxy_winner", buyer)
|
|
|
+ if b, err := redis.Exists("qyxy", buyer); err == nil && b {
|
|
|
+ text := redis.GetStr("qyxy", buyer)
|
|
|
err1 := json.Unmarshal([]byte(text), &buyermap)
|
|
|
if err1 != nil {
|
|
|
util.Debug(buyer, "buyer-----map解析异常")
|
|
@@ -212,22 +207,22 @@ func taskinfo1(tmp map[string]interface{}) {
|
|
|
types = append(types, "采购单位")
|
|
|
}
|
|
|
buyermap["bid_unittype"] = types
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
types := []string{"采购单位"}
|
|
|
buyermap["bid_unittype"] = types
|
|
|
}
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
buyermap["bid_unittype"] = []string{"采购单位"}
|
|
|
}
|
|
|
if len(buyermap) > 0 {
|
|
|
- data , _ := json.Marshal(buyermap)
|
|
|
- redis.Put("qyxy_winner", buyer, string(data), 0)
|
|
|
+ data, _ := json.Marshal(buyermap)
|
|
|
+ redis.Put("qyxy", buyer, string(data), 0)
|
|
|
}
|
|
|
}
|
|
|
if agency := util.ObjToString(tmp["agency"]); agency != "" {
|
|
|
agencymap := make(map[string]interface{})
|
|
|
- if b, err := redis.Exists("qyxy_winner", agency); err == nil && b {
|
|
|
- text := redis.GetStr("qyxy_winner", agency)
|
|
|
+ if b, err := redis.Exists("qyxy", agency); err == nil && b {
|
|
|
+ text := redis.GetStr("qyxy", agency)
|
|
|
err1 := json.Unmarshal([]byte(text), &agencymap)
|
|
|
if err1 != nil {
|
|
|
util.Debug(agency, "agency----map解析异常")
|
|
@@ -240,16 +235,16 @@ func taskinfo1(tmp map[string]interface{}) {
|
|
|
types = append(types, "代理机构")
|
|
|
}
|
|
|
agencymap["bid_unittype"] = types
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
types := []string{"代理机构"}
|
|
|
agencymap["bid_unittype"] = types
|
|
|
}
|
|
|
- }else {
|
|
|
+ } else {
|
|
|
agencymap["bid_unittype"] = []string{"代理机构"}
|
|
|
}
|
|
|
if len(agencymap) > 0 {
|
|
|
- data , _ := json.Marshal(agencymap)
|
|
|
- redis.Put("qyxy_winner", agency, string(data), 0)
|
|
|
+ data, _ := json.Marshal(agencymap)
|
|
|
+ redis.Put("qyxy", agency, string(data), 0)
|
|
|
}
|
|
|
}
|
|
|
}
|
|
@@ -268,4 +263,4 @@ func BinarySearch(s []string, k string) int {
|
|
|
}
|
|
|
}
|
|
|
return -1
|
|
|
-}
|
|
|
+}
|