zhangjinkun 5 år sedan
förälder
incheckning
603fad5a00
3 ändrade filer med 19 tillägg och 3 borttagningar
  1. 3 0
      udpcreateindex/src/config.json
  2. 7 1
      udpcreateindex/src/main.go
  3. 9 2
      udpcreateindex/src/projectindex.go

+ 3 - 0
udpcreateindex/src/config.json

@@ -42,12 +42,15 @@
         "multiIndex": ""
     },
     "project": {
+		"addr": "172.17.4.189:27082",
+        "size": 2,
         "db": "extract_kf",
         "collect": "huawei_project",
         "index": "projectset_v1",
         "type": "projectset"
     },
     "project2": {
+
         "db": "extract_kf",
         "collect": "huawei_project",
         "index": "project_v2",

+ 7 - 1
udpcreateindex/src/main.go

@@ -16,6 +16,7 @@ var (
 	Sysconfig          map[string]interface{} //配置文件
 	mgo                *mongodb.MongodbSim    //mongodb操作对象
 	extractmgo         *mongodb.MongodbSim    //mongodb操作对象
+	project2db         *mongodb.MongodbSim    //mongodb操作对象
 	mgostandard        *mongodb.MongodbSim    //mongodb操作对象
 	udpclient          mu.UdpClient           //udp对象
 	updport            string
@@ -47,7 +48,12 @@ func init() {
 		DbName:      mconf["db"].(string),
 	}
 	mgo.InitPool()
-
+	project2db = &mongodb.MongodbSim{
+		MongodbAddr: project2["addr"].(string),
+		Size:        util.IntAllDef(project2["pool"], 5),
+		DbName:      project2["db"].(string),
+	}
+	project2db.InitPool()
 	savedb, _ := Sysconfig["savedb"].(map[string]interface{})
 	if savedb == nil {
 		log.Println("未设置保存数据库,默认使用招标库")

+ 9 - 2
udpcreateindex/src/projectindex.go

@@ -9,6 +9,7 @@ import (
 	"qfw/util"
 	elastic "qfw/util/elastic"
 
+	mgov "gopkg.in/mgo.v2"
 	"gopkg.in/mgo.v2/bson"
 )
 
@@ -23,8 +24,14 @@ func projectTask(data []byte, project, mapInfo map[string]interface{}) {
 			},
 		}
 	}
-	session := extractmgo.GetMgoConn(3600)
-	defer extractmgo.DestoryMongoConn(session)
+	var session *mgov.Session
+	if project["addr"] != nil {
+		session = project2db.GetMgoConn(3600)
+		defer project2db.DestoryMongoConn(session)
+	} else {
+		session = extractmgo.GetMgoConn(3600)
+		defer extractmgo.DestoryMongoConn(session)
+	}
 	c, _ := project["collect"].(string)
 	db, _ := project["db"].(string)
 	index, _ := project["index"].(string)