*: refactor slow log format and parse slow query log to SLOW_QUERY table.

executor/adapter.go

@@ -379,52 +379,35 @@ func (a *ExecStmt) LogSlowQuery(txnTS uint64, succ bool) {
 	sessVars := a.Ctx.GetSessionVars()
 	sql = QueryReplacer.Replace(sql) + sessVars.GetExecuteArgumentsInfo()
 
-	connID := sessVars.ConnectionID
-	currentDB := sessVars.CurrentDB
 	var tableIDs, indexIDs string
 	if len(sessVars.StmtCtx.TableIDs) > 0 {
 		tableIDs = strings.Replace(fmt.Sprintf("table_ids:%v ", a.Ctx.GetSessionVars().StmtCtx.TableIDs), " ", ",", -1)
 	}
 	if len(sessVars.StmtCtx.IndexIDs) > 0 {
 		indexIDs = strings.Replace(fmt.Sprintf("index_ids:%v ", a.Ctx.GetSessionVars().StmtCtx.IndexIDs), " ", ",", -1)
 	}
-	user := sessVars.User
-	var internal string
-	if sessVars.InRestrictedSQL {
-		internal = "[INTERNAL] "
-	}
 	execDetail := sessVars.StmtCtx.GetExecDetails()
 	if costTime < threshold {
-		logutil.SlowQueryLogger.Debugf(
-			"[QUERY] %vcost_time:%vs %s succ:%v con:%v user:%s txn_start_ts:%v database:%v %v%vsql:%v",
-			internal, costTime.Seconds(), execDetail, succ, connID, user, txnTS, currentDB, tableIDs, indexIDs, sql)
+		logutil.SlowQueryLogger.Debugf(sessVars.SlowLogFormat(txnTS, costTime, execDetail, indexIDs, sql))
 	} else {
-		logutil.SlowQueryLogger.Warnf(
-			"[SLOW_QUERY] %vcost_time:%vs %s succ:%v con:%v user:%s txn_start_ts:%v database:%v %v%vsql:%v",
-			internal, costTime.Seconds(), execDetail, succ, connID, user, txnTS, currentDB, tableIDs, indexIDs, sql)
+		logutil.SlowQueryLogger.Warnf(sessVars.SlowLogFormat(txnTS, costTime, execDetail, indexIDs, sql))
 		metrics.TotalQueryProcHistogram.Observe(costTime.Seconds())
 		metrics.TotalCopProcHistogram.Observe(execDetail.ProcessTime.Seconds())
 		metrics.TotalCopWaitHistogram.Observe(execDetail.WaitTime.Seconds())
 		var userString string
-		if user != nil {
-			userString = user.String()
-		}
-		if len(tableIDs) > 10 {
-			tableIDs = tableIDs[10 : len(tableIDs)-1] // Remove "table_ids:" and the last ","
-		}
-		if len(indexIDs) > 10 {
-			indexIDs = indexIDs[10 : len(indexIDs)-1] // Remove "index_ids:" and the last ","
+		if sessVars.User != nil {
+			userString = sessVars.User.String()
 		}
 		domain.GetDomain(a.Ctx).LogSlowQuery(&domain.SlowQueryInfo{
 			SQL:      sql,
 			Start:    a.StartTime,
 			Duration: costTime,
 			Detail:   sessVars.StmtCtx.GetExecDetails(),
 			Succ:     succ,
-			ConnID:   connID,
+			ConnID:   sessVars.ConnectionID,
 			TxnTS:    txnTS,
 			User:     userString,
-			DB:       currentDB,
+			DB:       sessVars.CurrentDB,
 			TableIDs: tableIDs,
 			IndexIDs: indexIDs,
 			Internal: sessVars.InRestrictedSQL,

infoschema/slow_log.go

@@ -0,0 +1,194 @@
+// Copyright 2019 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 infoschema
+
+import (
+	"bufio"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/pingcap/errors"
+	"github.com/pingcap/parser/mysql"
+	"github.com/pingcap/tidb/sessionctx"
+	"github.com/pingcap/tidb/sessionctx/variable"
+	"github.com/pingcap/tidb/types"
+	"github.com/pingcap/tidb/util/execdetails"
+	"github.com/pingcap/tidb/util/hack"
+	"github.com/pingcap/tidb/util/logutil"
+	log "github.com/sirupsen/logrus"
+)
+
+var slowQueryCols = []columnInfo{
+	{variable.SlowLogTimeStr, mysql.TypeDatetime, -1, 0, nil, nil},
+	{variable.SlowLogTxnStartTSStr, mysql.TypeLonglong, 20, mysql.UnsignedFlag, nil, nil},
+	{variable.SlowLogUserStr, mysql.TypeVarchar, 64, 0, nil, nil},
+	{variable.SlowLogConnIDStr, mysql.TypeLonglong, 20, mysql.UnsignedFlag, nil, nil},
+	{variable.SlowLogQueryTimeStr, mysql.TypeDouble, 22, 0, nil, nil},
+	{execdetails.ProcessTimeStr, mysql.TypeDouble, 22, 0, nil, nil},
+	{execdetails.WaitTimeStr, mysql.TypeDouble, 22, 0, nil, nil},
+	{execdetails.BackoffTimeStr, mysql.TypeDouble, 22, 0, nil, nil},
+	{execdetails.RequestCountStr, mysql.TypeLonglong, 20, mysql.UnsignedFlag, nil, nil},
+	{execdetails.TotalKeysStr, mysql.TypeLonglong, 20, mysql.UnsignedFlag, nil, nil},
+	{execdetails.ProcessedKeysStr, mysql.TypeLonglong, 20, mysql.UnsignedFlag, nil, nil},
+	{variable.SlowLogDBStr, mysql.TypeVarchar, 64, 0, nil, nil},
+	{variable.SlowLogIsInternalStr, mysql.TypeTiny, 1, 0, nil, nil},
+	{variable.SlowLogQuerySQLStr, mysql.TypeVarchar, 4096, 0, nil, nil},
+}
+
+func dataForSlowLog(ctx sessionctx.Context) ([][]types.Datum, error) {
+	rowsMap, err := parseSlowLogFile(ctx.GetSessionVars().SlowQueryFile)
+	if err != nil {
+		return nil, errors.Trace(err)
+	}
+	var rows [][]types.Datum
+	for _, row := range rowsMap {
+		record := make([]types.Datum, 0, len(slowQueryCols))
+		for _, col := range slowQueryCols {
+			if v, ok := row[col.name]; ok {
+				record = append(record, v)
+			} else {
+				record = append(record, types.NewDatum(nil))
+			}
+		}
+		rows = append(rows, record)
+	}
+	return rows, nil
+}
+
+// TODO: Support parse multiple log-files.
+// parseSlowLogFile uses to parse slow log file.
+func parseSlowLogFile(filePath string) ([]map[string]types.Datum, error) {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return nil, errors.Trace(err)
+	}
+	defer func() {
+		if err = file.Close(); err != nil {
+			log.Error(err)
+		}
+	}()
+
+	return parseSlowLog(bufio.NewScanner(file))
+}
+
+// TODO: optimize for parse huge log-file.
+func parseSlowLog(scanner *bufio.Scanner) ([]map[string]types.Datum, error) {
+	rows := make([]map[string]types.Datum, 0)
+	rowMap := make(map[string]types.Datum, len(slowQueryCols))
+	startFlag := false
+
+	for scanner.Scan() {
+		line := scanner.Text()
+		// Check slow log entry start flag.
+		if !startFlag && strings.Contains(line, variable.SlowLogStartPrefixStr) {
+			value, err := parseSlowLogField(variable.SlowLogTimeStr, line[len(variable.SlowLogStartPrefixStr):])
+			if err != nil {
+				log.Errorf("parse slow log error: %v", err)
+				continue
+			}
+			rowMap[variable.SlowLogTimeStr] = *value
+			startFlag = true
+			continue
+		}
+
+		if startFlag {
+			// Parse slow log field.
+			if strings.Contains(line, variable.SlowLogPrefixStr) {
+				line = line[len(variable.SlowLogPrefixStr):]
+				fieldValues := strings.Split(line, " ")
+				for i := 0; i < len(fieldValues)-1; i += 2 {
+					field := fieldValues[i]
+					if strings.HasSuffix(field, ":") {
+						field = field[:len(field)-1]
+					}
+					value, err := parseSlowLogField(field, fieldValues[i+1])
+					if err != nil {
+						log.Errorf("parse slow log error: %v", err)
+						continue
+					}
+					rowMap[field] = *value
+
+				}
+			} else if strings.HasSuffix(line, variable.SlowLogSQLSuffixStr) {
+				// Get the sql string, and mark the start flag to false.
+				rowMap[variable.SlowLogQuerySQLStr] = types.NewStringDatum(string(hack.Slice(line)))
+				rows = append(rows, rowMap)
+				rowMap = make(map[string]types.Datum, len(slowQueryCols))
+				startFlag = false
+			}
+		}
+	}
+
+	if err := scanner.Err(); err != nil {
+		return nil, errors.Trace(err)
+	}
+	return rows, nil
+}
+
+func parseSlowLogField(field, value string) (*types.Datum, error) {
+	col := findColumnByName(slowQueryCols, field)
+	if col == nil {
+		return nil, errors.Errorf("can't found column %v", field)
+	}
+	var val types.Datum
+	switch col.tp {
+	case mysql.TypeLonglong:
+		num, err := strconv.ParseUint(value, 10, 64)
+		if err != nil {
+			return nil, errors.Trace(err)
+		}
+		val = types.NewUintDatum(num)
+	case mysql.TypeVarchar:
+		val = types.NewStringDatum(value)
+	case mysql.TypeDouble:
+		num, err := strconv.ParseFloat(value, 64)
+		if err != nil {
+			return nil, errors.Trace(err)
+		}
+		val = types.NewDatum(num)
+	case mysql.TypeTiny:
+		// parse bool
+		val = types.NewDatum(value == "true")
+	case mysql.TypeDatetime:
+		t, err := parseTime(value)
+		if err != nil {
+			return nil, errors.Trace(err)
+		}
+		val = types.NewTimeDatum(types.Time{
+			Time: types.FromGoTime(t),
+			Type: mysql.TypeDatetime,
+			Fsp:  types.MaxFsp,
+		})
+	}
+	return &val, nil
+}
+
+func parseTime(s string) (time.Time, error) {
+	t, err := time.Parse(logutil.SlowLogTimeFormat, s)
+	if err != nil {
+		err = errors.Errorf("string \"%v\" doesn't has a prefix that matches format \"%v\", err: %v", s, logutil.SlowLogTimeFormat, err)
+	}
+	return t, err
+}
+
+func findColumnByName(cols []columnInfo, colName string) *columnInfo {
+	for _, col := range cols {
+		if col.name == colName {
+			return &col
+		}
+	}
+	return nil
+}

infoschema/slow_log_test.go

@@ -0,0 +1,74 @@
+// Copyright 2019 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 infoschema
+
+import (
+	"bufio"
+	"bytes"
+	"testing"
+
+	"github.com/pingcap/parser/mysql"
+	"github.com/pingcap/tidb/types"
+)
+
+func TestParseSlowLogFile(t *testing.T) {
+	slowLog := bytes.NewBufferString(
+		`# Time: 2019-01-24-22:32:29.313255 +0800
+# Txn_start_ts: 405888132465033227
+# Query_time: 0.216905
+# Process_time: 0.021 Request_count: 1 Total_keys: 637 Processed_keys: 436
+# Is_internal: true
+select * from t;`)
+	scanner := bufio.NewScanner(slowLog)
+	rows, err := parseSlowLog(scanner)
+	if err != nil {
+		t.Fatalf("parse slow log failed")
+	}
+	if len(rows) != 1 {
+		t.Fatalf("parse slow log failed")
+	}
+	row := rows[0]
+	t1 := types.Time{
+		Time: types.FromDate(2019, 01, 24, 22, 32, 29, 313255),
+		Type: mysql.TypeDatetime,
+		Fsp:  types.MaxFsp,
+	}
+	if logTime, ok := row["Time"]; !ok || logTime.GetMysqlTime() != t1 {
+		t.Fatalf("parse slow log failed")
+	}
+	if ts, ok := row["Txn_start_ts"]; !ok || ts.GetUint64() != 405888132465033227 {
+		t.Fatalf("parse slow log failed")
+	}
+	if queryTime, ok := row["Query_time"]; !ok || queryTime.GetFloat64() != 0.216905 {
+		t.Fatalf("parse slow log failed")
+	}
+	if ProcessTime, ok := row["Process_time"]; !ok || ProcessTime.GetFloat64() != 0.021 {
+		t.Fatalf("parse slow log failed")
+	}
+	if requestCount, ok := row["Request_count"]; !ok || requestCount.GetUint64() != 1 {
+		t.Fatalf("parse slow log failed")
+	}
+	if totalKeys, ok := row["Total_keys"]; !ok || totalKeys.GetUint64() != 637 {
+		t.Fatalf("parse slow log failed")
+	}
+	if processedKeys, ok := row["Processed_keys"]; !ok || processedKeys.GetUint64() != 436 {
+		t.Fatalf("parse slow log failed")
+	}
+	if isInternal, ok := row["Is_internal"]; !ok || isInternal.GetInt64() != 1 {
+		t.Fatalf("parse slow log failed")
+	}
+	if sql, ok := row["Query"]; !ok || sql.GetString() != "select * from t;" {
+		t.Fatalf("parse slow log failed")
+	}
+}

infoschema/tables.go

@@ -67,6 +67,7 @@ const (
 	tableCollationCharacterSetApplicability = "COLLATION_CHARACTER_SET_APPLICABILITY"
 	tableProcesslist                        = "PROCESSLIST"
 	tableTiDBIndexes                        = "TIDB_INDEXES"
+	tableSlowLog                            = "SLOW_QUERY"
 )
 
 type columnInfo struct {
@@ -1469,6 +1470,7 @@ var tableNameToColumns = map[string][]columnInfo{
 	tableCollationCharacterSetApplicability: tableCollationCharacterSetApplicabilityCols,
 	tableProcesslist:                        tableProcesslistCols,
 	tableTiDBIndexes:                        tableTiDBIndexesCols,
+	tableSlowLog:                            slowQueryCols,
 }
 
 func createInfoSchemaTable(handle *Handle, meta *model.TableInfo) *infoschemaTable {
@@ -1560,6 +1562,8 @@ func (it *infoschemaTable) getRows(ctx sessionctx.Context, cols []*table.Column)
 		fullRows = dataForCollationCharacterSetApplicability()
 	case tableProcesslist:
 		fullRows = dataForProcesslist(ctx)
+	case tableSlowLog:
+		fullRows, err = dataForSlowLog(ctx)
 	}
 	if err != nil {
 		return nil, errors.Trace(err)