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
|
package flydns
import (
"fmt"
"net"
"net/netip"
"strings"
)
func combineString(inp []string) string {
return strings.Join(inp, "")
}
func GetAppAddresses(appName string) ([]netip.Addr, error) {
addrs, err := net.LookupAddr(fmt.Sprintf("%s.internal", appName))
if err != nil {
return nil, err
}
var result []netip.Addr
for _, addr := range addrs {
ip, err := netip.ParseAddr(addr)
if err != nil {
return nil, err
}
result = append(result, ip)
}
return result, nil
}
func GetIndividualInstanceAddress(instanceID string, appName string) (netip.Addr, error) {
var ip netip.Addr
addr, err := net.LookupAddr(fmt.Sprintf("%s.vm.%s.internal", instanceID, appName))
if err != nil {
return ip, err
}
ip, err = netip.ParseAddr(addr[0])
if err != nil {
return ip, err
}
return ip, nil
}
func GetAppInstancesInRegion(appName string, region string) ([]netip.Addr, error) {
instances, err := net.LookupHost(fmt.Sprintf("%s.%s.internal", appName, region))
if err != nil {
return nil, err
}
var result []netip.Addr
for _, instance := range instances {
ip, err := netip.ParseAddr(instance)
if err != nil {
return nil, err
}
result = append(result, ip)
}
return result, nil
}
func GetClosestInstances(appName string, count int) ([]netip.Addr, error) {
instances, err := net.LookupHost(fmt.Sprintf("top%d.nearest.of.%s.internal", count, appName))
if err != nil {
return nil, err
}
var result []netip.Addr
for _, instance := range instances {
ip, err := netip.ParseAddr(instance)
if err != nil {
return nil, err
}
result = append(result, ip)
}
return result, nil
}
func GetApps() ([]string, error) {
appString, err := net.LookupTXT("_apps.internal")
if err != nil {
return nil, err
}
var result []string
for _, resp := range strings.Split(appString[0], ",") {
result = append(result, strings.Split(resp, ",")...)
}
return result, nil
}
type Machine struct {
ID string
Region string
}
func parseMachine(input string) (Machine, error) {
var result Machine
split := strings.Split(input, " ")
if len(split) != 2 {
return result, fmt.Errorf("invalid input")
}
result.ID = split[0]
result.Region = split[1]
return result, nil
}
func GetMachines(appName string) ([]Machine, error) {
machines, err := net.LookupTXT(fmt.Sprintf("vms.%s.internal", appName))
if err != nil {
return nil, err
}
var result []Machine
for _, machine := range strings.Split(combineString(machines), ",") {
m, err := parseMachine(machine)
if err != nil {
return nil, err
}
result = append(result, m)
}
return result, nil
}
func GetRegions(appName string) ([]string, error) {
regions, err := net.LookupTXT(fmt.Sprintf("regions.%s.internal", appName))
if err != nil {
return nil, err
}
return strings.Split(combineString(regions), ","), nil
}
|
