117 lines
3.0 KiB
Go
117 lines
3.0 KiB
Go
// Copyright 2018 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 execdetails
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
// ExecDetails contains execution detail information.
|
|
type ExecDetails struct {
|
|
ProcessTime time.Duration
|
|
WaitTime time.Duration
|
|
BackoffTime time.Duration
|
|
RequestCount int
|
|
TotalKeys int64
|
|
ProcessedKeys int64
|
|
}
|
|
|
|
// String implements the fmt.Stringer interface.
|
|
func (d ExecDetails) String() string {
|
|
parts := make([]string, 0, 6)
|
|
if d.ProcessTime > 0 {
|
|
parts = append(parts, fmt.Sprintf("process_time:%v", d.ProcessTime))
|
|
}
|
|
if d.WaitTime > 0 {
|
|
parts = append(parts, fmt.Sprintf("wait_time:%v", d.WaitTime))
|
|
}
|
|
if d.BackoffTime > 0 {
|
|
parts = append(parts, fmt.Sprintf("backoff_time:%v", d.BackoffTime))
|
|
}
|
|
if d.RequestCount > 0 {
|
|
parts = append(parts, fmt.Sprintf("request_count:%d", d.RequestCount))
|
|
}
|
|
if d.TotalKeys > 0 {
|
|
parts = append(parts, fmt.Sprintf("total_keys:%d", d.TotalKeys))
|
|
}
|
|
if d.ProcessedKeys > 0 {
|
|
parts = append(parts, fmt.Sprintf("processed_keys:%d", d.ProcessedKeys))
|
|
}
|
|
return strings.Join(parts, " ")
|
|
}
|
|
|
|
// RuntimeStatsColl collects executors's execution info.
|
|
type RuntimeStatsColl struct {
|
|
mu sync.Mutex
|
|
stats map[string]*RuntimeStats
|
|
}
|
|
|
|
// RuntimeStats collects one executor's execution info.
|
|
type RuntimeStats struct {
|
|
// executor's Next() called times.
|
|
loop int32
|
|
// executor consume time.
|
|
consume int64
|
|
// executor return row count.
|
|
rows int64
|
|
}
|
|
|
|
// NewRuntimeStatsColl creates new executor collector.
|
|
func NewRuntimeStatsColl() *RuntimeStatsColl {
|
|
return &RuntimeStatsColl{stats: make(map[string]*RuntimeStats)}
|
|
}
|
|
|
|
// Get gets execStat for a executor.
|
|
func (e *RuntimeStatsColl) Get(planID string) *RuntimeStats {
|
|
e.mu.Lock()
|
|
defer e.mu.Unlock()
|
|
runtimeStats, exists := e.stats[planID]
|
|
if !exists {
|
|
runtimeStats = &RuntimeStats{}
|
|
e.stats[planID] = runtimeStats
|
|
}
|
|
return runtimeStats
|
|
}
|
|
|
|
// Exists checks if the planID exists in the stats collection.
|
|
func (e *RuntimeStatsColl) Exists(planID string) bool {
|
|
e.mu.Lock()
|
|
defer e.mu.Unlock()
|
|
_, exists := e.stats[planID]
|
|
return exists
|
|
}
|
|
|
|
// Record records executor's execution.
|
|
func (e *RuntimeStats) Record(d time.Duration, rowNum int) {
|
|
atomic.AddInt32(&e.loop, 1)
|
|
atomic.AddInt64(&e.consume, int64(d))
|
|
atomic.AddInt64(&e.rows, int64(rowNum))
|
|
}
|
|
|
|
// SetRowNum sets the row num.
|
|
func (e *RuntimeStats) SetRowNum(rowNum int64) {
|
|
atomic.StoreInt64(&e.rows, rowNum)
|
|
}
|
|
|
|
func (e *RuntimeStats) String() string {
|
|
if e == nil {
|
|
return ""
|
|
}
|
|
return fmt.Sprintf("time:%v, loops:%d, rows:%d", time.Duration(e.consume), e.loop, e.rows)
|
|
}
|