7x31
/
sniffer-agent
mirror of https://github.com/zr-hebo/sniffer-agent.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix bug of lost user info and use db stmt

This commit is contained in:
hebo 2019-08-30 12:45:04 +08:00
parent a09af431ee
commit 10b08dec52
1 changed files with 49 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,8 +6,8 @@ import (
"sync" "sync"
"time" "time"
"github.com/siddontang/go/hack"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/pingcap/tidb/util/hack"
"github.com/zr-hebo/sniffer-agent/model" "github.com/zr-hebo/sniffer-agent/model"
) )
@ -269,8 +269,7 @@ func (ms *MysqlSession) GenerateQueryPiece() (qp model.QueryPiece) {
var mqp *model.PooledMysqlQueryPiece var mqp *model.PooledMysqlQueryPiece
var querySQLInBytes []byte var querySQLInBytes []byte
switch ms.cachedStmtBytes[0] { if ms.cachedStmtBytes[0] > 32 {
case ComAuth:
var userName, dbName string var userName, dbName string
var err error var err error
userName, dbName, err = parseAuthInfo(ms.cachedStmtBytes) userName, dbName, err = parseAuthInfo(ms.cachedStmtBytes)
@ -280,9 +279,12 @@ func (ms *MysqlSession) GenerateQueryPiece() (qp model.QueryPiece) {
ms.visitUser = &userName ms.visitUser = &userName
ms.visitDB = &dbName ms.visitDB = &dbName
} else {
switch ms.cachedStmtBytes[0] {
case ComInitDB: case ComInitDB:
newDBName := string(ms.cachedStmtBytes[1:]) newDBName := string(ms.cachedStmtBytes[1:])
useSQL := fmt.Sprintf("use %s", newDBName) useSQL := fmt.Sprintf("use %s", newDBName)
querySQLInBytes = hack.Slice(useSQL)
mqp = ms.composeQueryPiece() mqp = ms.composeQueryPiece()
mqp.QuerySQL = &useSQL mqp.QuerySQL = &useSQL
// update session database // update session database
@ -324,6 +326,8 @@ func (ms *MysqlSession) GenerateQueryPiece() (qp model.QueryPiece) {
default: default:
return return
} }
}
if strictMode && mqp != nil && mqp.VisitUser == nil { if strictMode && mqp != nil && mqp.VisitUser == nil {
user, db, err := querySessionInfo(ms.serverPort, mqp.SessionID) user, db, err := querySessionInfo(ms.serverPort, mqp.SessionID)
@ -341,10 +345,12 @@ func (ms *MysqlSession) GenerateQueryPiece() (qp model.QueryPiece) {
func filterQueryPieceBySQL(mqp *model.PooledMysqlQueryPiece, querySQL []byte) (model.QueryPiece) { func filterQueryPieceBySQL(mqp *model.PooledMysqlQueryPiece, querySQL []byte) (model.QueryPiece) {
if mqp == nil || querySQL == nil { if mqp == nil || querySQL == nil {
return nil return nil
}
} else if (uselessSQLPattern.Match(querySQL)) { /*
else if (uselessSQLPattern.Match(querySQL)) {
return nil return nil
} }
*/
if ddlPatern.Match(querySQL) { if ddlPatern.Match(querySQL) {
mqp.SetNeedSyncSend(true) mqp.SetNeedSyncSend(true)