mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-16 00:41:24 +01:00
7f4fb34182
It is better developing vmctl tool in VictoriaMetrics repository, so it could be released together with the rest of vmutils tools such as vmalert, vmagent, vmbackup, vmrestore and vmauth.
116 lines
2.2 KiB
Go
116 lines
2.2 KiB
Go
package pb
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"github.com/mattn/go-runewidth"
|
|
"math"
|
|
"regexp"
|
|
//"unicode/utf8"
|
|
)
|
|
|
|
const (
|
|
_KiB = 1024
|
|
_MiB = 1048576
|
|
_GiB = 1073741824
|
|
_TiB = 1099511627776
|
|
|
|
_kB = 1e3
|
|
_MB = 1e6
|
|
_GB = 1e9
|
|
_TB = 1e12
|
|
)
|
|
|
|
var ctrlFinder = regexp.MustCompile("\x1b\x5b[0-9]+\x6d")
|
|
|
|
func CellCount(s string) int {
|
|
n := runewidth.StringWidth(s)
|
|
for _, sm := range ctrlFinder.FindAllString(s, -1) {
|
|
n -= runewidth.StringWidth(sm)
|
|
}
|
|
return n
|
|
}
|
|
|
|
func StripString(s string, w int) string {
|
|
l := CellCount(s)
|
|
if l <= w {
|
|
return s
|
|
}
|
|
var buf = bytes.NewBuffer(make([]byte, 0, len(s)))
|
|
StripStringToBuffer(s, w, buf)
|
|
return buf.String()
|
|
}
|
|
|
|
func StripStringToBuffer(s string, w int, buf *bytes.Buffer) {
|
|
var seqs = ctrlFinder.FindAllStringIndex(s, -1)
|
|
mainloop:
|
|
for i, r := range s {
|
|
for _, seq := range seqs {
|
|
if i >= seq[0] && i < seq[1] {
|
|
buf.WriteRune(r)
|
|
continue mainloop
|
|
}
|
|
}
|
|
if rw := CellCount(string(r)); rw <= w {
|
|
w -= rw
|
|
buf.WriteRune(r)
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
for w > 0 {
|
|
buf.WriteByte(' ')
|
|
w--
|
|
}
|
|
return
|
|
}
|
|
|
|
func round(val float64) (newVal float64) {
|
|
roundOn := 0.5
|
|
places := 0
|
|
var round float64
|
|
pow := math.Pow(10, float64(places))
|
|
digit := pow * val
|
|
_, div := math.Modf(digit)
|
|
if div >= roundOn {
|
|
round = math.Ceil(digit)
|
|
} else {
|
|
round = math.Floor(digit)
|
|
}
|
|
newVal = round / pow
|
|
return
|
|
}
|
|
|
|
// Convert bytes to human readable string. Like a 2 MiB, 64.2 KiB, or 2 MB, 64.2 kB
|
|
// if useSIPrefix is set to true
|
|
func formatBytes(i int64, useSIPrefix bool) (result string) {
|
|
if !useSIPrefix {
|
|
switch {
|
|
case i >= _TiB:
|
|
result = fmt.Sprintf("%.02f TiB", float64(i)/_TiB)
|
|
case i >= _GiB:
|
|
result = fmt.Sprintf("%.02f GiB", float64(i)/_GiB)
|
|
case i >= _MiB:
|
|
result = fmt.Sprintf("%.02f MiB", float64(i)/_MiB)
|
|
case i >= _KiB:
|
|
result = fmt.Sprintf("%.02f KiB", float64(i)/_KiB)
|
|
default:
|
|
result = fmt.Sprintf("%d B", i)
|
|
}
|
|
} else {
|
|
switch {
|
|
case i >= _TB:
|
|
result = fmt.Sprintf("%.02f TB", float64(i)/_TB)
|
|
case i >= _GB:
|
|
result = fmt.Sprintf("%.02f GB", float64(i)/_GB)
|
|
case i >= _MB:
|
|
result = fmt.Sprintf("%.02f MB", float64(i)/_MB)
|
|
case i >= _kB:
|
|
result = fmt.Sprintf("%.02f kB", float64(i)/_kB)
|
|
default:
|
|
result = fmt.Sprintf("%d B", i)
|
|
}
|
|
}
|
|
return
|
|
}
|