-
Notifications
You must be signed in to change notification settings - Fork 0
/
util.go
127 lines (112 loc) · 2.48 KB
/
util.go
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package dns
import (
"fmt"
"math/rand"
"net"
"os"
"reflect"
"regexp"
"strconv"
"strings"
"github.com/wwqdrh/gokit/logger"
)
var (
ResolvConf = "/etc/resolv.conf"
)
const (
StandardDnsPort = 53
AlternativeDnsPort = 10053
)
func ArrayEquals(src, target []string) bool {
if len(src) != len(target) {
return false
}
for i := 0; i < len(src); i++ {
found := false
for j := 0; j < len(target); j++ {
if src[i] == target[j] {
found = true
break
}
}
if !found {
return false
}
}
return true
}
// CreateDirIfNotExist create dir
func CreateDirIfNotExist(dir string) error {
if _, err := os.Stat(dir); os.IsNotExist(err) {
err = os.MkdirAll(dir, 0755)
if err != nil {
return err
}
}
return nil
}
// Contains check whether target exist in container, the type of container can be an array, slice or map
func Contains(container any, target any) bool {
targetValue := reflect.ValueOf(container)
switch reflect.TypeOf(container).Kind() {
case reflect.Slice, reflect.Array:
for i := 0; i < targetValue.Len(); i++ {
if targetValue.Index(i).Interface() == target {
return true
}
}
case reflect.Map:
if targetValue.MapIndex(reflect.ValueOf(target)).IsValid() {
return true
}
}
return false
}
const IpAddrPattern = "[0-9]+\\.[0-9]+\\.[0-9]+\\.[0-9]+"
func IsValidIp(ip string) bool {
if ok, err := regexp.MatchString("^"+IpAddrPattern+"$", ip); ok && err == nil {
return true
}
return false
}
// DomainNotExistError ...
type DomainNotExistError struct {
name string
qtype uint16
}
func (e DomainNotExistError) Error() string {
return fmt.Sprintf("domain %s (%d) not exist", e.name, e.qtype)
}
// IsDomainNotExist check the error type
func IsDomainNotExist(err error) bool {
_, exists := err.(DomainNotExistError)
return exists
}
// GetRandomTcpPort get pod random ssh port
func GetRandomTcpPort() int {
for i := 0; i < 20; i++ {
port := RandomPort()
conn, err := net.Dial("tcp", fmt.Sprintf(":%d", port))
if err == nil {
logger.DefaultLogger.Debugx("Port %d not available", nil, port)
_ = conn.Close()
} else {
logger.DefaultLogger.Debugx("Using port %d", nil, port)
return port
}
}
port := RandomPort()
logger.DefaultLogger.Debugx("Using random port %d", nil, port)
return port
}
func RandomPort() int {
return rand.Intn(65535-1024) + 1024
}
func string2IP(ip string) [4]byte {
res := [4]byte{}
for i, item := range strings.SplitN(ip, ".", 4) {
ch, _ := strconv.Atoi(item)
res[i] = byte(ch)
}
return res
}