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

This commit is contained in:
hebo 2019-11-14 21:52:19 +08:00
parent 16722c878c
commit 85bef28022
2 changed files with 20 additions and 18 deletions

View File

@ -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,19 +81,17 @@ func (crs *coverRanges) mergeRanges() {
}
}
type coveragePool struct {
queue chan *coverageNode
}
func NewCoveragePool() (cp *coveragePool) {
return &coveragePool{
queue: make(chan *coverageNode, 256),
}
}
func (crp *coveragePool) NewCoverage(begin, end int64)(cn *coverageNode) {
func (crp *coveragePool) NewCoverage(begin, end int64) (cn *coverageNode) {
cn = crp.Dequeue()
cn.begin = begin
cn.end = end
@ -126,7 +124,7 @@ func (crp *coveragePool) Dequeue() (cn *coverageNode) {
}()
select {
case cn = <- crp.queue:
case cn = <-crp.queue:
return
default:

View File

@ -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 {