server: adj-in/out supports route families rather than v4/v6

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>

1 files changed, 5 insertions, 5 deletions

diff --git a/table/table_manager.go b/table/table_manager.go
index e135f58e..d14fec5b 100644
--- a/table/table_manager.go
+++ b/table/table_manager.go
@@ -287,15 +287,15 @@ type AdjRib struct {
 	adjRibOut map[bgp.RouteFamily]map[string]*ReceivedRoute
 }
 
-func NewAdjRib() *AdjRib {
+func NewAdjRib(rfList []bgp.RouteFamily) *AdjRib {
 	r := &AdjRib{
 		adjRibIn:  make(map[bgp.RouteFamily]map[string]*ReceivedRoute),
 		adjRibOut: make(map[bgp.RouteFamily]map[string]*ReceivedRoute),
 	}
-	r.adjRibIn[bgp.RF_IPv4_UC] = make(map[string]*ReceivedRoute)
-	r.adjRibIn[bgp.RF_IPv6_UC] = make(map[string]*ReceivedRoute)
-	r.adjRibOut[bgp.RF_IPv4_UC] = make(map[string]*ReceivedRoute)
-	r.adjRibOut[bgp.RF_IPv6_UC] = make(map[string]*ReceivedRoute)
+	for _, rf := range rfList {
+		r.adjRibIn[rf] = make(map[string]*ReceivedRoute)
+		r.adjRibOut[rf] = make(map[string]*ReceivedRoute)
+	}
 	return r
 }