This repository has been archived by the owner on Oct 24, 2022. It is now read-only.
forked from kubeedge/kubeedge
/
config.go
105 lines (92 loc) · 2.94 KB
/
config.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
/*
Copyright 2019 The KubeEdge Authors.
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,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package utils
import (
"encoding/json"
"math/rand"
"os"
"path/filepath"
"time"
)
const letterBytes = "abcdefghijklmnopqrstuvwxyz0123456789"
type vmSpec struct {
Ip string `json:"ip"`
Username string `json:"username"`
Passwd string `json:"password"`
}
//config.json decode struct
type Config struct {
AppImageUrl []string `json:"image_url"`
K8SMasterForKubeEdge string `json:"k8smasterforkubeedge"`
Nodes map[string]vmSpec `json:"k8snodes"`
NumOfNodes int `json:"node_num"`
ImageRepo string `json:"imagerepo"`
K8SMasterForProvisionEdgeNodes string `json:"k8smasterforprovisionedgenodes"`
CloudImageUrl string `json:"cloudimageurl"`
EdgeImageUrl string `json:"edgeimageurl"`
Namespace string `json:"namespace"`
ControllerStubPort int `json:"controllerstubport"`
Protocol string `json:"protocol"`
}
//config struct
var config *Config
//get config.json path
func LoadConfig() Config {
if config == nil {
config = loadConfigJsonFromPath()
}
return *config
}
//loadConfigJsonFromPath reads the test configuration and builds a Config object.
func loadConfigJsonFromPath() *Config {
path := getConfigPath()
_, err := filepath.Abs(filepath.Dir(path))
if err != nil {
InfoV6("Failed to get Abs path: %v", err)
panic(err)
}
var config *Config = &Config{}
configFile, err := os.Open(path)
if err != nil {
panic(err)
}
decoder := json.NewDecoder(configFile)
err = decoder.Decode(config)
if err != nil {
panic(err)
}
return config
}
//getConfigPath returns the configuration path provided in the env var name. In case the env var is not
//set, the default configuration path is returned
func getConfigPath() string {
path := os.Getenv("TESTCONFIG")
if path == "" {
path = "config.json"
}
return path
}
func RandomInt(min, max int) int {
return min + rand.Intn(max-min)
}
//function to Generate Random string
func GetRandomString(length int) string {
str := "0123456789abcdefghijklmnopqrstuvwxyz"
bytes := []byte(str)
result := []byte{}
r := rand.New(rand.NewSource(time.Now().UnixNano()))
for i := 0; i < length; i++ {
result = append(result, bytes[r.Intn(len(bytes))])
}
return string(result)
}