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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
| package lib_hosts
import (
"bufio"
"context"
"fmt"
"io"
"log"
"net"
"os"
"regexp"
"strings"
"time"
"github.com/buger/jsonparser"
"github.com/redis/go-redis/v9"
)
func UpdateHostsFileFormRedisAndLookupIP(rdb redis.ClusterClient, ctx context.Context) {
EnvHostsTag := os.Getenv("EnvHostsTag")
if EnvHostsTag == "" {
log.Println("== 不需要检查和修改 域名解析")
return
}
//是否更新 hosts文件，华为云那边不能修改，需要通过vpc的内网域名的A记录来控制
enableUpdataHostsFile := os.Getenv("EnvHostsEndEnableUpdateFile") == "1"
//执行完成后是否ping一下
EnablePing := os.Getenv("EnvHostsEndEnablePing") == "1"
val, err := rdb.Get(ctx, "f501:Config:/etc/hosts").Result()
if err != nil {
log.Println("UpdateHostsFileFormRedis f501:Config:/etc/hosts 错误！！！", err)
return
} else {
linesString, err := jsonparser.GetString([]byte(val), EnvHostsTag, "Lines")
if err != nil {
log.Println("UpdateHostsFileFormRedis 没找到相关信息 ", err)
return
}
if enableUpdataHostsFile {
UpdateHostsFile(linesString) //更新hosts文件 华为那边zip部署的fc不支持需要手动配置修改vpc
} else {
log.Println("部分环境不启用/etc/hosts更新,如果要启动请 到redis修改...")
log.Println("跳过/etc/hosts配置,你可能需要手动配置dns或vpc组网的内网域名的A记录 ...")
}
if EnablePing { //执行ping命令
log.Println("所有域名分别LookupIP ...")
time.Sleep(200 * time.Microsecond)
re := regexp.MustCompile(`\s+`)
linesString = re.ReplaceAllString(linesString, " ")
lines := strings.Split(linesString, ",")
for _, line := range lines {
linesArr := strings.Split(line, " ")
if len(linesArr) > 1 {
ipAddresses, err := net.LookupIP(linesArr[1])
if err != nil {
log.Println("net.LookupIP Error:", err)
return
}
for _, ip := range ipAddresses {
log.Println("net.LookupIP :", linesArr[1], ip)
}
} else {
log.Println("UpdateHostsFileFormRedis 长度不符", linesArr)
}
}
}
}
}
// 修改hosts文件
func UpdateHostsFile(linesString string) {
log.Println("更新hosts 文件...")
lines := strings.Split(linesString, ",")
exists := FileExists("/etc/hosts-fish-source")
if !exists {
err := CopyFile("/etc/hosts", "/etc/hosts-fish-source")
if err != nil {
log.Println(err)
return
} else {
//log.Println("复制hosts文件 到 /etc/hosts-fish-source")
}
}
fileAppend, err := os.Create("/etc/hosts-fish-append")
if err != nil {
log.Println("无法创建目标文件: ", err)
return
}
defer fileAppend.Close()
writer := bufio.NewWriter(fileAppend)
defer writer.Flush()
for _, line := range lines {
_, err := writer.WriteString(line + "\n")
if err != nil {
log.Println("写入文件失败:", err)
return
}
}
writer.Flush()
/* buf2, err := os.ReadFile("/etc/hosts-fish-append")
if err != nil {
log.Println("文件读取失败", "append", err)
return
}
log.Println("###---###内容 wjx append", string(buf2)) // 将内容转换为字符串并打印
*/
//log.Println("新建文件 到 /etc/hosts-fish-append 准备合并")
err = MergeFiles("/etc/hosts-fish-source", "/etc/hosts-fish-append", "/etc/hosts")
if err != nil {
log.Println("文件合并成功错误", err)
} else {
//log.Println("文件合并成功")
}
}
func FileExists(filename string) bool {
_, err := os.Stat(filename)
if err == nil {
return true // 文件存在
}
if os.IsNotExist(err) {
return false // 文件不存在
}
return false // 其他错误，无法确定文件是否存在
}
func MergeFiles(file1, file2, outputFile string) error {
err := ClearFile(outputFile)
if err != nil {
log.Println("文件清空失败", err)
return err
}
/* content, err := os.ReadFile(outputFile) // 读取文件内容
if err != nil {
log.Println("文件清空后重新读取失败", outputFile, err)
return err
}
log.Println("###---###内容", outputFile, string(content)) // 将内容转换为字符串并打印
*/
buf1, err := os.ReadFile(file1)
if err != nil {
log.Println("文件读取失败", file1, err)
return err
}
//log.Println("###---###内容", file1, string(buf1)) // 将内容转换为字符串并打印
buf2, err := os.ReadFile(file2)
if err != nil {
log.Println("文件读取失败", file2, err)
return err
}
//log.Println("###---###内容", file2, string(buf2)) // 将内容转换为字符串并打印
file, err := os.OpenFile(outputFile, os.O_CREATE|os.O_RDWR|os.O_APPEND, 0666)
if err != nil {
log.Println("文件打开失败", outputFile, err)
return err
}
defer file.Close()
writer := bufio.NewWriter(file)
writer.Write(buf1)
writer.Write(buf2)
writer.Flush()
return nil
}
func ClearFile(filename string) error {
// 打开文件并截断内容
file, err := os.OpenFile(filename, os.O_RDWR|os.O_TRUNC, 0666)
if err != nil {
return err
}
defer file.Close()
// 清空文件内容
err = file.Truncate(0)
if err != nil {
return err
}
//log.Printf("文件 %s 已成功清空", filename)
return nil
}
func CopyFile(sourceFile string, destinationFile string) error {
src, err := os.Open(sourceFile)
if err != nil {
return fmt.Errorf("无法打开源文件 %s: %s", sourceFile, err)
}
defer src.Close()
dst, err := os.Create(destinationFile)
if err != nil {
return fmt.Errorf("无法创建目标文件 %s: %s", destinationFile, err)
}
defer dst.Close()
_, err = io.Copy(dst, src)
if err != nil {
return fmt.Errorf("复制文件时出错: %s", err)
}
return nil
}
|