修复设置抓包率引起的查询时间异常的问题

session-dealer/mysql/cover_range.go

@@ -28,6 +28,7 @@ func NewCoverRanges() *coverRanges {
 	return &coverRanges{
 		head: &coverageNode{
 			begin: -1,
+			end:   -1,
 		},
 	}
 }
@@ -52,10 +53,9 @@ func (crs *coverRanges) addRange(node *coverageNode)  {
 			node.next = checkRange
 			node = nil
 			break
-
-		} else {
-			currRange = checkRange
 		}
+
+		currRange = checkRange
 	}
 
 	if node != nil && currRange != nil {
@@ -70,7 +70,7 @@ func (crs *coverRanges) mergeRanges()  {
 	currRange := crs.head.next
 	for currRange != nil && currRange.next != nil {
 		checkRange := currRange.next
-		if currRange.end >= checkRange.begin && currRange.end < checkRange.end {
+		if currRange.begin <= checkRange.begin && currRange.end >= checkRange.begin && currRange.end < checkRange.end {
 			currRange.end = checkRange.end
 			currRange.next = checkRange.next
 			checkRange.Recovery()
@@ -81,12 +81,10 @@ func (crs *coverRanges) mergeRanges()  {
 	}
 }
 
-
 type coveragePool struct {
 	queue chan *coverageNode
 }
 
-
 func NewCoveragePool() (cp *coveragePool) {
 	return &coveragePool{
 		queue: make(chan *coverageNode, 256),

session-dealer/mysql/session.go

@@ -87,7 +87,7 @@ func (ms *MysqlSession) ReceiveTCPPacket(newPkt *model.TCPPacket) {
 		ms.readFromClient(newPkt.Seq, newPkt.Payload)
 
 	} else {
-		ms.readFromServer(newPkt.Payload)
+		ms.readFromServer(newPkt.Seq, newPkt.Payload)
 		qp := ms.GenerateQueryPiece()
 		if qp != nil {
 			ms.queryPieceReceiver <- qp
@@ -99,7 +99,7 @@ func (ms *MysqlSession) resetBeginTime() {
 	ms.stmtBeginTime = time.Now().UnixNano() / millSecondUnit
 }
 
-func (ms *MysqlSession) readFromServer(bytes []byte) {
+func (ms *MysqlSession) readFromServer(respSeq int64, bytes []byte) {
 	if ms.expectSendSize < 1 && len(bytes) > 4 {
 		ms.expectSendSize = extractMysqlPayloadSize(bytes[:4])
 		contents := bytes[4:]
@@ -107,6 +107,10 @@ func (ms *MysqlSession) readFromServer(bytes []byte) {
 			ms.prepareInfo.prepareStmtID = bytesToInt(contents[1:5])
 		}
 	}
+
+	if ms.coverRanges.head.next == nil || ms.coverRanges.head.next.end != respSeq {
+		ms.clear()
+	}
 }
 
 func (ms *MysqlSession) checkFinish() bool {