1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
// Copyright 2020 PingCAP, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.
package utils
import (
"math"
"net"
"strconv"
"strings"
"time"
"github.com/spf13/pflag"
)
// JoinInt joins a slice of int to string
func JoinInt(nums []int, delim string) string {
result := ""
for _, i := range nums {
result += strconv.Itoa(i)
result += delim
}
return strings.TrimSuffix(result, delim)
}
// IsFlagSetByUser check if the a flag is set by user explicitly
func IsFlagSetByUser(flagSet *pflag.FlagSet, flagName string) bool {
setByUser := false
flagSet.Visit(func(f *pflag.Flag) {
if f.Name == flagName {
setByUser = true
}
})
return setByUser
}
// MustAtoI calls strconv.Atoi and ignores error
func MustAtoI(a string) int {
v, _ := strconv.Atoi(a)
return v
}
// Base62Tag returns a tag based on time
func Base62Tag() string {
const base = 62
const sets = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
b := make([]byte, 0)
num := time.Now().UnixNano() / int64(time.Millisecond)
for num > 0 {
r := math.Mod(float64(num), float64(base))
num /= base
b = append([]byte{sets[int(r)]}, b...)
}
return string(b)
}
// Ternary operator
func Ternary(condition bool, a, b any) any {
if condition {
return a
}
return b
}
// JoinHostPort return host and port
func JoinHostPort(host string, port int) string {
return net.JoinHostPort(host, strconv.Itoa(port))
}
// ParseHostPort Prase host and port
func ParseHostPort(hostport string) (host, port string) {
colon := strings.LastIndex(hostport, ":")
host = strings.TrimSuffix(strings.TrimPrefix(hostport[:colon], "["), "]")
port = hostport[colon+1:]
return
}
|
