forked from zalando/skipper
-
Notifications
You must be signed in to change notification settings - Fork 0
/
fallbackgroup.go
57 lines (46 loc) · 1.27 KB
/
fallbackgroup.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
package routing
// QUESTION: does the fallback functionality always apply to load balanced routes?
// TODO: implement an additional predicate that is also checked for fallbacks, sg. like "fallback group"
// currently only for load balanced routes
func applyFallbackGroups(r []*Route) []*Route {
groups := make(map[string][]*Route)
for i := range r {
for j := range r[i].Predicates {
er := r[i].Route
// NOTE: this one "LBMember" is now hard coded to omit circular dependency.
// It can be fixed if we implement this logic as a post processor.
if er.Predicates[j].Name != "LBMember" {
continue
}
if len(er.Predicates[j].Args) == 0 {
continue
}
name, ok := er.Predicates[j].Args[0].(string)
if !ok {
continue
}
groups[name] = append(groups[name], r[i])
}
}
for name, group := range groups {
if len(group) == 0 {
continue
}
// TODO: here we could clean off load balancing, if there's only a single route
head := group[0]
head.Head = head
head.Me = head
head.Group = name
head.IsLoadBalanced = true
current := head
for _, route := range group[1:] {
current.Next = route
current = route
current.Head = head
current.Me = current
current.Group = name
current.IsLoadBalanced = true
}
}
return r
}