更改目录结构

server/pkg/utils/util.go

@@ -0,0 +1,74 @@
+package utils
+
+import (
+	"fmt"
+	"math/rand"
+	"time"
+)
+
+func init() {
+	rand.Seed(time.Now().UnixNano())
+}
+
+func InArrStr(arr []string, str string) bool {
+	for _, d := range arr {
+		if d == str {
+			return true
+		}
+	}
+	return false
+}
+
+const (
+	KB = 1024
+	MB = 1024 * KB
+	GB = 1024 * MB
+	TB = 1024 * GB
+	PB = 1024 * TB
+)
+
+func HumanByte(bf interface{}) string {
+	var hb string
+	var bAll float64
+	switch bi := bf.(type) {
+	case int:
+		bAll = float64(bi)
+	case int32:
+		bAll = float64(bi)
+	case uint32:
+		bAll = float64(bi)
+	case int64:
+		bAll = float64(bi)
+	case uint64:
+		bAll = float64(bi)
+	case float64:
+		bAll = float64(bi)
+	}
+
+	switch {
+	case bAll >= TB:
+		hb = fmt.Sprintf("%0.2f TB", bAll/TB)
+	case bAll >= GB:
+		hb = fmt.Sprintf("%0.2f GB", bAll/GB)
+	case bAll >= MB:
+		hb = fmt.Sprintf("%0.2f MB", bAll/MB)
+	case bAll >= KB:
+		hb = fmt.Sprintf("%0.2f KB", bAll/KB)
+	default:
+		hb = fmt.Sprintf("%0.2f B", bAll)
+	}
+
+	return hb
+}
+
+func RandomNum(length int) string {
+	letterRunes := []rune("abcdefghijklmnpqrstuvwxy1234567890")
+
+	bytes := make([]rune, length)
+
+	for i := range bytes {
+		bytes[i] = letterRunes[rand.Intn(len(letterRunes))]
+	}
+
+	return string(bytes)
+}