-
Notifications
You must be signed in to change notification settings - Fork 1
/
serial.go
52 lines (47 loc) · 1.03 KB
/
serial.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
package algorithm
func Serialize(nodeHostMap, svcIpMap map[string]string) JobMap {
return Parallelize(nodeHostMap, svcIpMap, 1)
}
func Parallelize(nodeHostMap, svcIpMap map[string]string, parallel int) JobMap {
r := []map[string]string{}
ipHostMap := map[string]string{}
for _, serverHost := range nodeHostMap {
serverIp := svcIpMap[serverHost]
ipHostMap[serverIp] = serverHost
for _, client := range nodeHostMap {
if client == serverHost {
continue
}
m := map[string]string{client: serverIp}
r = append(r, m)
}
}
jobs := map[int][]JobNode{}
jobMap := JobMap{
JobNodeSize: 0,
EpochSize: 0,
Jobs: jobs,
}
i := 0
for i < len(r) {
js := []JobNode{}
for j := 0; j < parallel; j++ {
if i+j >= len(r) {
break
}
m := r[i+j]
for k, v := range m {
js = append(js, JobNode{
ServerHost: ipHostMap[v],
ClientHost: k,
ServerIp: v,
})
}
}
jobMap.Jobs[jobMap.EpochSize] = js
jobMap.JobNodeSize += len(js)
jobMap.EpochSize++
i += parallel
}
return jobMap
}