7x31/sniffer-agent
1
0
Fork 0
mirror of https://github.com/zr-hebo/sniffer-agent.git synced 2025-09-18 09:47:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug of recover early

Browse Source
This commit is contained in:
hebo
2019-10-08 17:54:25 +08:00
parent 1d64e41b58
commit d019ac76cf
3 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
model/mysql_query_piece.go
View File

@@ -62,10 +62,15 @@ func (mqp *MysqlQueryPiece) Bytes() (content []byte) {
return mqp.jsonContent return mqp.jsonContent
} }
mqp.jsonContent = marsharQueryPieceMonopolize(mqp) mqp.GenerateJsonBytes()
return mqp.jsonContent return mqp.jsonContent
} }
func (mqp *MysqlQueryPiece) GenerateJsonBytes() {
mqp.jsonContent = marsharQueryPieceMonopolize(mqp)
return
}
func (mqp *MysqlQueryPiece) GetSQL() (str *string) { func (mqp *MysqlQueryPiece) GetSQL() (str *string) {
return mqp.QuerySQL return mqp.QuerySQL
} }

1
model/query_piece_pool.go
View File

@@ -19,7 +19,6 @@ func (mqpp *mysqlQueryPiecePool) Enqueue(pmqp *PooledMysqlQueryPiece) {
mqpp.lock.Lock() mqpp.lock.Lock()
defer mqpp.lock.Unlock() defer mqpp.lock.Unlock()
mqpp.queue <- pmqp mqpp.queue <- pmqp
} }

8
session-dealer/mysql/session.go
View File

@@ -301,10 +301,14 @@ func (ms *MysqlSession) GenerateQueryPiece() (qp model.QueryPiece) {
} }
} }
return filterQueryPieceBySQL(mqp, querySQLInBytes) mqp = filterQueryPieceBySQL(mqp, querySQLInBytes)
if mqp != nil {
mqp.GenerateJsonBytes()
}
return mqp
} }
func filterQueryPieceBySQL(mqp *model.PooledMysqlQueryPiece, querySQL []byte) (model.QueryPiece) { func filterQueryPieceBySQL(mqp *model.PooledMysqlQueryPiece, querySQL []byte) (*model.PooledMysqlQueryPiece) {
if mqp == nil || querySQL == nil { if mqp == nil || querySQL == nil {
return nil return nil