118 lines
3.2 KiB
Go
118 lines
3.2 KiB
Go
// Copyright 2015 PingCAP, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package variable_test
|
|
|
|
import (
|
|
"time"
|
|
|
|
. "github.com/pingcap/check"
|
|
"github.com/pingcap/parser/auth"
|
|
"github.com/pingcap/tidb/util/execdetails"
|
|
"github.com/pingcap/tidb/util/mock"
|
|
)
|
|
|
|
var _ = Suite(&testSessionSuite{})
|
|
|
|
type testSessionSuite struct {
|
|
}
|
|
|
|
func (*testSessionSuite) TestSession(c *C) {
|
|
ctx := mock.NewContext()
|
|
|
|
ss := ctx.GetSessionVars().StmtCtx
|
|
c.Assert(ss, NotNil)
|
|
|
|
// For AffectedRows
|
|
ss.AddAffectedRows(1)
|
|
c.Assert(ss.AffectedRows(), Equals, uint64(1))
|
|
ss.AddAffectedRows(1)
|
|
c.Assert(ss.AffectedRows(), Equals, uint64(2))
|
|
|
|
// For RecordRows
|
|
ss.AddRecordRows(1)
|
|
c.Assert(ss.RecordRows(), Equals, uint64(1))
|
|
ss.AddRecordRows(1)
|
|
c.Assert(ss.RecordRows(), Equals, uint64(2))
|
|
|
|
// For FoundRows
|
|
ss.AddFoundRows(1)
|
|
c.Assert(ss.FoundRows(), Equals, uint64(1))
|
|
ss.AddFoundRows(1)
|
|
c.Assert(ss.FoundRows(), Equals, uint64(2))
|
|
|
|
// For UpdatedRows
|
|
ss.AddUpdatedRows(1)
|
|
c.Assert(ss.UpdatedRows(), Equals, uint64(1))
|
|
ss.AddUpdatedRows(1)
|
|
c.Assert(ss.UpdatedRows(), Equals, uint64(2))
|
|
|
|
// For TouchedRows
|
|
ss.AddTouchedRows(1)
|
|
c.Assert(ss.TouchedRows(), Equals, uint64(1))
|
|
ss.AddTouchedRows(1)
|
|
c.Assert(ss.TouchedRows(), Equals, uint64(2))
|
|
|
|
// For CopiedRows
|
|
ss.AddCopiedRows(1)
|
|
c.Assert(ss.CopiedRows(), Equals, uint64(1))
|
|
ss.AddCopiedRows(1)
|
|
c.Assert(ss.CopiedRows(), Equals, uint64(2))
|
|
|
|
// For last insert id
|
|
ctx.GetSessionVars().SetLastInsertID(1)
|
|
c.Assert(ctx.GetSessionVars().StmtCtx.LastInsertID, Equals, uint64(1))
|
|
|
|
ss.ResetForRetry()
|
|
c.Assert(ss.AffectedRows(), Equals, uint64(0))
|
|
c.Assert(ss.FoundRows(), Equals, uint64(0))
|
|
c.Assert(ss.UpdatedRows(), Equals, uint64(0))
|
|
c.Assert(ss.RecordRows(), Equals, uint64(0))
|
|
c.Assert(ss.TouchedRows(), Equals, uint64(0))
|
|
c.Assert(ss.CopiedRows(), Equals, uint64(0))
|
|
c.Assert(ss.WarningCount(), Equals, uint16(0))
|
|
}
|
|
|
|
func (*testSessionSuite) TestSlowLogFormat(c *C) {
|
|
ctx := mock.NewContext()
|
|
|
|
seVar := ctx.GetSessionVars()
|
|
c.Assert(seVar, NotNil)
|
|
|
|
seVar.User = &auth.UserIdentity{Username: "root", Hostname: "192.168.0.1"}
|
|
seVar.ConnectionID = 1
|
|
seVar.CurrentDB = "test"
|
|
seVar.InRestrictedSQL = true
|
|
txnTS := uint64(406649736972468225)
|
|
costTime := time.Second
|
|
execDetail := execdetails.ExecDetails{
|
|
ProcessTime: time.Second * time.Duration(2),
|
|
WaitTime: time.Minute,
|
|
BackoffTime: time.Millisecond,
|
|
RequestCount: 2,
|
|
TotalKeys: 10000,
|
|
ProcessedKeys: 20001,
|
|
}
|
|
resultString := `# Txn_start_ts: 406649736972468225
|
|
# User: root@192.168.0.1
|
|
# Conn_ID: 1
|
|
# Query_time: 1
|
|
# Process_time: 2 Wait_time: 60 Backoff_time: 0.001 Request_count: 2 Total_keys: 10000 Process_keys: 20001
|
|
# DB: test
|
|
# Index_ids: [1,2]
|
|
# Is_internal: true
|
|
select * from t;`
|
|
logString := seVar.SlowLogFormat(txnTS, costTime, execDetail, "[1,2]", "select * from t")
|
|
c.Assert(logString, Equals, resultString)
|
|
}
|