소스 검색

id区间查询逻辑调整

mxs 1 년 전
부모
커밋
06c713389e
2개의 변경된 파일6개의 추가작업 그리고 13개의 파일을 삭제
  1. 1 2
      download_file/src/main.go
  2. 5 11
      download_file/src/util/util.go

+ 1 - 2
download_file/src/main.go

@@ -3,7 +3,6 @@ package main
 import (
 	"flag"
 	"mongodb"
-	"net/http"
 	"os"
 	qu "qfw/util"
 	sp "spiderutil"
@@ -72,7 +71,7 @@ func main() {
 			}
 		}
 	}()
-	go http.ListenAndServe(UdpPort, nil)
+	//go http.ListenAndServe(UdpPort, nil)
 	ch := make(chan bool, 1)
 	<-ch
 }

+ 5 - 11
download_file/src/util/util.go

@@ -36,17 +36,11 @@ func GetIdInterval(id string) (gtid, lteid string) {
 		},
 		"dataprocess": 8,
 	}
-	list, _ := MgoB.Find(Coll, query, map[string]interface{}{"_id": 1}, nil, false, -1, -1)
-	dataLength := len(*list)
-	if dataLength == 0 { //无新ID段
-		return
-	}
-
-	gtid = qu.ObjToString((*list)[0]["gtid"])              //起始id
-	lteid = qu.ObjToString((*list)[dataLength-1]["lteid"]) //结束id
-	//成功获取id区间
-	if gtid < lteid {
-		StartID = mongodb.BsonIdToSId((*list)[dataLength-1]["_id"])
+	list, _ := MgoB.Find(Coll, query, map[string]interface{}{"_id": 1}, nil, false, 0, 1) //查找一个id区间
+	if len(*list) > 0 {
+		gtid = qu.ObjToString((*list)[0]["gtid"])   //起始id
+		lteid = qu.ObjToString((*list)[0]["lteid"]) //结束id
+		StartID = mongodb.BsonIdToSId((*list)[0]["_id"])
 		qu.Debug("当前轮ID区间:", gtid, lteid, "表ID:", StartID)
 		return
 	}