update utils

Author: zmh-program

conf.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"fmt"
-	"log"
 	"math/rand"
 	"os"
 	"strings"
@@ -31,7 +30,7 @@ func GetTokenFromEnv() []string {
 
 func ValidateToken() {
 	if len(tokenList) == 0 {
-		log.Fatal("No token found!")
+		logger.Fatal("No token found! Please set TOKEN environment variable.")
 	}
 	logger.Debug(fmt.Sprintf("Found %d available token(s).", len(tokenList)))
 }

main.go

@@ -10,5 +10,4 @@ func main() {
 
 	tokenList = GetTokenFromEnv()
 	ValidateToken()
-
 }

utils.go

@@ -0,0 +1,42 @@
+package main
+
+import "fmt"
+
+func Sum(array []int) int {
+	total := 0
+	for _, v := range array {
+		total += v
+	}
+	return total
+}
+
+var sizeUnits = []string{"B", "KiB", "MiB", "GiB", "TiB", "PiB"}
+
+func SizeConvert(size float64, idx int) string {
+	if size <= 0 {
+		return "0"
+	}
+	for idx < len(sizeUnits)-1 && size > 1024 {
+		size /= 1024
+		idx++
+	}
+	return fmt.Sprintf("%.1f %s", size, sizeUnits[idx])
+}
+
+var scaleUnits = []string{"", "k", "m"}
+
+func ScaleConvert(n float64, useSmallScale bool) string {
+	idx := 0
+	condition := 100.0
+	if !useSmallScale {
+		condition = 1000.0
+	}
+	for idx < len(scaleUnits)-1 && n > condition {
+		n /= 1000
+		idx++
+	}
+	if idx == 0 {
+		return fmt.Sprintf("%.0f", n)
+	}
+	return fmt.Sprintf("%.1f%s", n, scaleUnits[idx])
+}