mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-22 08:26:19 +01:00
25e54d2b50
add progress bars to the VM importer The new progress bars supposed to display the processing speed per each VM importer worker. This info should help to identify if there is a bottleneck on the VM side during the import process, without waiting for its finish. The new progress bars can be disabled by passing `vm-disable-progress-bar` flag. Plotting multiple progress bars requires using experimental progress bar pool from github.com/cheggaaa/pb/v3. Switch to progress bar pool required changes in all import modes. The openTSDB mode wasn't changed due to its implementation, which implies individual progress bars per each series. Because of this, using the pool wasn't possible. Signed-off-by: dmitryk-dk <kozlovdmitriyy@gmail.com> Co-authored-by: hagen1778 <roman@victoriametrics.com>
44 lines
844 B
Go
44 lines
844 B
Go
// +build linux darwin freebsd netbsd openbsd solaris dragonfly plan9 aix
|
|
|
|
package pb
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/cheggaaa/pb/v3/termutil"
|
|
)
|
|
|
|
func (p *Pool) print(first bool) bool {
|
|
p.m.Lock()
|
|
defer p.m.Unlock()
|
|
var out string
|
|
if !first {
|
|
out = fmt.Sprintf("\033[%dA", p.lastBarsCount)
|
|
}
|
|
isFinished := true
|
|
bars := p.bars
|
|
rows, cols, err := termutil.TerminalSize()
|
|
if err != nil {
|
|
cols = defaultBarWidth
|
|
}
|
|
if rows > 0 && len(bars) > rows {
|
|
// we need to hide bars that overflow terminal height
|
|
bars = bars[len(bars)-rows:]
|
|
}
|
|
for _, bar := range bars {
|
|
if !bar.IsFinished() {
|
|
isFinished = false
|
|
}
|
|
bar.SetWidth(cols)
|
|
out += fmt.Sprintf("\r%s\n", bar.String())
|
|
}
|
|
if p.Output != nil {
|
|
fmt.Fprint(p.Output, out)
|
|
} else {
|
|
fmt.Fprint(os.Stderr, out)
|
|
}
|
|
p.lastBarsCount = len(bars)
|
|
return isFinished
|
|
}
|