|
@@ -13,18 +13,13 @@ import (
|
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
|
)
|
|
|
|
|
|
-const (
|
|
|
- ProjectCache = "info" //存放每条项目信息,key为项目ID
|
|
|
-)
|
|
|
-
|
|
|
var (
|
|
|
Sysconfig map[string]interface{} //读取配置文件
|
|
|
- MongoTool *MongodbSim //mongodb连接
|
|
|
+ MongoTool, MgoBidding *MongodbSim //mongodb连接
|
|
|
ExtractColl, ProjectColl, BackupColl, SiteColl string //抽取表、项目表、项目快照表、站点表
|
|
|
Thread int //配置项线程数
|
|
|
- //NextNode []interface{}
|
|
|
- BlackList []interface{}
|
|
|
- BlaskListMap map[string]bool
|
|
|
+ BlackList []interface{}
|
|
|
+ BlaskListMap map[string]bool
|
|
|
)
|
|
|
|
|
|
var (
|
|
@@ -62,6 +57,13 @@ func init() {
|
|
|
DbName: Sysconfig["mongodbName"].(string),
|
|
|
}
|
|
|
MongoTool.InitPool()
|
|
|
+ bidding, _ := Sysconfig["bidding"].(map[string]interface{})
|
|
|
+ MgoBidding = &MongodbSim{
|
|
|
+ MongodbAddr: bidding["addr"].(string),
|
|
|
+ Size: util.IntAll(bidding["dbsize"]),
|
|
|
+ DbName: bidding["dbname"].(string),
|
|
|
+ }
|
|
|
+ MgoBidding.InitPool()
|
|
|
|
|
|
ExtractColl = Sysconfig["extractColl"].(string)
|
|
|
ProjectColl = Sysconfig["projectColl"].(string)
|
|
@@ -416,7 +418,6 @@ func CosineSimilar(srcWords1, dstWords1 string) float64 {
|
|
|
return v1
|
|
|
}
|
|
|
|
|
|
-
|
|
|
func initWinnerRegexp() {
|
|
|
winRegMap := Sysconfig["winner"].(map[string]interface{})
|
|
|
preRegexps := winRegMap["pre_regexp"].([]interface{})
|
|
@@ -458,7 +459,6 @@ func initWinnerRegexp() {
|
|
|
BlackRegexp["winner"] = winBlackRegexps
|
|
|
}
|
|
|
|
|
|
-
|
|
|
func initBuyerRegexp() {
|
|
|
buyRegMap := Sysconfig["buyer"].(map[string]interface{})
|
|
|
preRegexps := buyRegMap["pre_regexp"].([]interface{})
|
|
@@ -540,4 +540,3 @@ func initAgencyRegexp() {
|
|
|
}
|
|
|
BlackRegexp["agency"] = winBlackRegexps
|
|
|
}
|
|
|
-
|