mirror of
https://github.com/gluster/glusterd2.git
synced 2026-02-07 09:46:04 +01:00
486e23ed22
Added capacity field to volume GET AP response. This is currently returned only for volumes provisioned dynamically and their volume snapshots. The CLI prints the volume capacity in a human friendly format. Signed-off-by: Hari Gowtham <hgowtham@redhat.com>
106 lines
2.1 KiB
Go
106 lines
2.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"math"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
validSizeFormat = regexp.MustCompile(`^([0-9]+)([GMKT])?$`)
|
|
)
|
|
|
|
func formatBoolYesNo(value bool) string {
|
|
if value == true {
|
|
return "yes"
|
|
}
|
|
return "no"
|
|
}
|
|
|
|
func formatPID(pid int) string {
|
|
if pid == 0 {
|
|
return ""
|
|
}
|
|
return strconv.Itoa(pid)
|
|
}
|
|
|
|
func sizeToMb(value string) (uint64, error) {
|
|
sizeParts := validSizeFormat.FindStringSubmatch(value)
|
|
if len(sizeParts) == 0 {
|
|
return 0, errors.New("invalid size format")
|
|
}
|
|
|
|
// If Size unit is specified as M/K/G/T, Default Size unit is M
|
|
sizeUnit := "M"
|
|
if len(sizeParts) == 3 {
|
|
sizeUnit = sizeParts[2]
|
|
}
|
|
|
|
sizeValue, err := strconv.ParseUint(sizeParts[1], 10, 64)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
var size uint64
|
|
switch sizeUnit {
|
|
case "K":
|
|
size = sizeValue / 1024
|
|
case "G":
|
|
size = sizeValue * 1024
|
|
case "T":
|
|
size = sizeValue * 1024 * 1024
|
|
default:
|
|
size = sizeValue
|
|
}
|
|
return size, nil
|
|
}
|
|
|
|
// logn is used to find the unit size the given MB belongs to.
|
|
// 1024 MB will return 1
|
|
// 1048576 MB will return 2 and so on
|
|
func logn(n, b float64) float64 {
|
|
return math.Log(n) / math.Log(b)
|
|
}
|
|
|
|
// humanReadable converts size given in MB into a more human readable unit
|
|
//
|
|
// humanReadable(1024) returns 1.0 GB
|
|
// humanReadable(1536) returns 1.5 GB
|
|
// humanReadable(1048576) returns 1.0 TB
|
|
func humanReadable(value uint64) string {
|
|
units := []string{"MB", "GB", "TB", "PB", "EB"}
|
|
// If less than 1024MB we return it as such
|
|
if value < 1024 {
|
|
return fmt.Sprintf("%.1f MB", float64(value))
|
|
}
|
|
e := math.Floor(logn(float64(value), 1024))
|
|
suffix := units[int(e)]
|
|
size := math.Floor(float64(value)/math.Pow(1024, e)*10+0.5) / 10
|
|
f := "%.0f %s"
|
|
if size < 10 {
|
|
f = "%.1f %s"
|
|
}
|
|
|
|
return fmt.Sprintf(f, size, suffix)
|
|
}
|
|
|
|
func readString(prompt string, args ...interface{}) string {
|
|
var s string
|
|
fmt.Printf(prompt, args...)
|
|
fmt.Scanln(&s)
|
|
return s
|
|
}
|
|
|
|
// PromptConfirm prompts for confirmation
|
|
func PromptConfirm(prompt string, args ...interface{}) bool {
|
|
switch strings.ToLower(readString(prompt, args...)) {
|
|
case "yes", "y":
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|