forked from kubeedge/kubeedge
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.go
executable file
·89 lines (77 loc) · 2.09 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
/*
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 edge
import (
"encoding/json"
"math/rand"
"os"
"path/filepath"
"time"
"github.com/kubeedge/kubeedge/edge/test/integration/utils/common"
)
//config.json decode struct
type Config struct {
MqttEndpoint string `json:"mqttEndpoint"`
TestManager string `json:"testManager"`
EdgedEndpoint string `json:"edgedEndpoint"`
AppImageUrl []string `json:"image_url"`
NodeId string `json:"nodeId"`
}
//config struct
var config *Config
//get config.json path
func LoadConfig() Config {
if config == nil {
config = loadConfigJsonFromPath()
}
return *config
}
//Load Config.json from the PWD, and decode the config.
func loadConfigJsonFromPath() *Config {
path := getConfigPath()
_, err := filepath.Abs(filepath.Dir(path))
if err != nil {
common.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
}
//Get config path from Env or hard code the file path
func getConfigPath() string {
path := os.Getenv("TESTCONFIG")
if path == "" {
path = "config.json"
}
return path
}
//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)
}