Browse Source

增加剑鱼页面后台缺失的分享签名后台

李广朋 9 years ago
parent
commit
917d1028f2
2 changed files with 81 additions and 40 deletions
  1. 72 40
      common/src/main.go
  2. 9 0
      core/src/qfw/swordfish/swordfishmanage.go

+ 72 - 40
common/src/main.go

@@ -15,11 +15,13 @@ import (
 var N = 200
 
 //生enterprise表
-func main() {
+func main1() {
 	// 参数 mongodb(网站库) mongodb(公示库) elastic collection area index-db inde-type N
-
 	initConfig(os.Args[1], "web", "qfw")
-	initConfig(os.Args[2], "ecps", "qfw")
+	if os.Args[4] == "enterprise" {
+		initConfig(os.Args[2], "ecps", "qfw")
+
+	}
 	InitMongodbPool()
 	elastic.InitElastic(os.Args[3])
 	if len(os.Args[8]) > 0 {
@@ -40,7 +42,7 @@ func main() {
 	query := ecps_session.DB("qfw").C(os.Args[4]).Find(m).Iter()
 	arr := make([]map[string]interface{}, 0)
 	for tmp := make(map[string]interface{}); query.Next(tmp); i++ {
-		arr = append(arr, elastic.ConverData(&tmp))
+		arr = append(arr, tmp)
 		if i%N == 0 && i != 0 {
 			elastic.BulkSave(os.Args[6], os.Args[7], &arr, true)
 			arr = make([]map[string]interface{}, 0)
@@ -51,53 +53,83 @@ func main() {
 		elastic.BulkSave(os.Args[6], os.Args[7], &arr, true)
 		arr = make([]map[string]interface{}, 0)
 	}
-
-	//生成带名片索引
-	i = 0
-	count := Count("enterprise", "web", "qfw", m)
-	log.Println("网站库,一共", count, "条数据")
-	for {
-		limit := i + N
-		if limit > count {
-			limit = count
-		}
-		query := *Find("enterprise", "web", "qfw", m, nil, nil, false, i, i+N)
-		regNos := make([]string, 0)
-		arr := make([]map[string]interface{}, 0)
-		if len(query) > 0 {
-			for _, v := range query {
-				if v["_id"] != nil {
-					regno := fmt.Sprint(v["RegNo"])
-					regNos = append(regNos, "\""+regno+"\"")
-					delete(v, "_id")
-				}
+	if os.Args[4] == "enterprise" {
+		//生成带名片索引
+		i = 0
+		count := Count("enterprise", "web", "qfw", m)
+		log.Println("网站库,一共", count, "条数据")
+		for {
+			limit := i + N
+			if limit > count {
+				limit = count
 			}
-
-			findstr := `{"RegNo":{"$in":[` + strings.Join(regNos, ",") + `]}}`
-			query1 := *Find("enterprise", "ecps", "qfw", findstr, nil, nil, false, -1, -1)
-			for _, v := range query {
-				for _, _v := range query1 {
-					if fmt.Sprint(_v["RegNo"]) == fmt.Sprint(v["RegNo"]) {
+			query := *Find("enterprise", "web", "qfw", m, nil, nil, false, i, i+N)
+			regNos := make([]string, 0)
+			arr := make([]map[string]interface{}, 0)
+			if len(query) > 0 {
+				for _, v := range query {
+					if v["_id"] != nil {
+						regno := fmt.Sprint(v["RegNo"])
+						regNos = append(regNos, "\""+regno+"\"")
 						delete(v, "_id")
-						if len(v) > 1 {
-							for field, value := range v {
-								_v[field] = value
+					}
+				}
+
+				findstr := `{"RegNo":{"$in":[` + strings.Join(regNos, ",") + `]}}`
+				query1 := *Find("enterprise", "ecps", "qfw", findstr, nil, nil, false, -1, -1)
+				for _, v := range query {
+					for _, _v := range query1 {
+						if fmt.Sprint(_v["RegNo"]) == fmt.Sprint(v["RegNo"]) {
+							delete(v, "_id")
+							if len(v) > 1 {
+								for field, value := range v {
+									_v[field] = value
+								}
+								arr = append(arr, elastic.ConverData(&_v))
 							}
-							arr = append(arr, elastic.ConverData(&_v))
 						}
 					}
 				}
+				elastic.BulkSave(os.Args[6], os.Args[7], &arr, true)
 			}
-			elastic.BulkSave(os.Args[6], os.Args[7], &arr, true)
+			if i >= count {
+				break
+			} else {
+				i = i + N
+			}
+			log.Printf("当前索引位置:=%d \n", i)
 		}
-		if i >= count {
-			break
+	}
+
+}
+func main() {
+	// 参数 mongodb(网站库) elastic collection query
+	initConfig(os.Args[1], "web", "qfw")
+	InitMongodbPool()
+	log.Println("初始化完毕")
+	elastic.InitElastic(os.Args[2])
+	log.Println("elatic初始化完毕")
+	query := ""
+	if len(os.Args) > 4 && os.Args[4] != "" {
+		query = os.Args[4]
+		query = strings.Replace(query, "^", "$", -1)
+	}
+	log.Println("query:", query)
+	list := *Find(os.Args[3], "web", "qfw", query, nil, nil, false, -1, -1)
+	log.Println("len:", len(list))
+	log.Println("查询完毕")
+	var i = 0
+	for i <= len(list) {
+		var tmp []map[string]interface{}
+		if i+100 >= len(list) {
+			tmp = list[i:len(list)]
 		} else {
-			i = i + N
+			tmp = list[i : i+100]
 		}
-		log.Printf("当前索引位置:=%d \n", i)
+		elastic.BulkSave(os.Args[3], os.Args[3], &tmp, true)
+		log.Println("当前索引:", i)
+		i = i + 100
 	}
-
 }
 func initConfig(ip, alias, db string) {
 	conf := *new(PoolConfig)

+ 9 - 0
core/src/qfw/swordfish/swordfishmanage.go

@@ -7,6 +7,8 @@ import (
 	"gopkg.in/mgo.v2/bson"
 	"net/rpc"
 	"qfw/coreconfig"
+	cutil "qfw/coreutil"
+	"qfw/mobile"
 	"qfw/util"
 	"qfw/util/mongodb"
 	"qfw/util/redis"
@@ -94,6 +96,8 @@ func (s *SwordFish) Wxprotocol() error {
 
 //剑鱼微信查询页面
 func (s *SwordFish) Wxsearch() error {
+	s.T["signature "] = mobile.GetSignature(s.Action)
+	s.T["shareid"] = cutil.FindMyShareId("topjy", s.Session().Get("s_m_openid").(string))
 	return s.Render("/swordfish/wxsearch.html")
 }
 
@@ -133,6 +137,9 @@ func (s *SwordFish) Wxsearchlist() error {
 			}
 		}
 	}
+	//搜索列表增加分享
+	s.T["signature "] = mobile.GetSignature(s.Action)
+	s.T["shareid"] = cutil.FindMyShareId("topjy", s.Session().Get("s_m_openid").(string))
 	return s.Render("/swordfish/wxsearchlist.html", &s.T)
 }
 
@@ -256,6 +263,8 @@ func getSwordFish(lasttime int64, sid string, res []map[string]interface{}, coun
 func (s *SwordFish) VisitRedirect() {
 	sid := s.GetString("id")
 	surl := s.GetString("url")
+	s.T["signature "] = mobile.GetSignature(s.Action)
+	s.T["shareid"] = cutil.FindMyShareId("topjy", s.Session().Get("s_m_openid").(string))
 	if len(sid) > 5 {
 		obj := *mongodb.FindById("bidding", sid, `{"publishtime":1,"_id":0,"detail":1,"title":1}`)
 		if obj != nil && len(obj) == 3 {