Fix invalid type assertion in BGP FSM

bio-routing · Jul 28, 2022 · 225a40b · 225a40b
1 parent eddc5ab
commit 225a40b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/protocols/bgp/server/fsm_address_family.go b/protocols/bgp/server/fsm_address_family.go
@@ -119,7 +119,7 @@ func (f *fsmAddressFamily) bmpInit() {
 func (f *fsmAddressFamily) bmpDispose() {
 	f.rib.GetContributingASNs().Remove(f.fsm.peer.localASN)
 
-	f.adjRIBIn.(*adjRIBIn.AdjRIBIn).Flush()
+	f.adjRIBIn.Flush()
 
 	f.adjRIBIn.Unregister(f.rib)
 

diff --git a/routingtable/client_interface.go b/routingtable/client_interface.go
@@ -31,6 +31,7 @@ type AdjRIB interface {
 // AdjRIBIn is the interface any AdjRIBIn must implement
 type AdjRIBIn interface {
 	AdjRIB
+	Flush()
 }
 
 // AdjRIBOut is the interface any AdjRIBOut must implement

diff --git a/routingtable/mock_client.go b/routingtable/mock_client.go
@@ -83,3 +83,5 @@ func (m *RTMockClient) ReplaceFilterChain(filter.Chain) {}
 func (m *RTMockClient) ReplacePath(*net.Prefix, *route.Path, *route.Path) {}
 
 func (m *RTMockClient) Dispose() {}
+
+func (m *RTMockClient) Flush() {}
-Original file line number
+Diff line change
@@ Expand Up / @@ -31,6 +31,7 @@ type AdjRIB interface { @@
     // AdjRIBIn is the interface any AdjRIBIn must implement
     type AdjRIBIn interface {
     	AdjRIB
+    	Flush()
     }
     // AdjRIBOut is the interface any AdjRIBOut must implement
@@ Expand Down @@
Original file line number	Diff line number	Diff line change
Expand Up		@@ -83,3 +83,5 @@ func (m *RTMockClient) ReplaceFilterChain(filter.Chain) {}
		func (m RTMockClient) ReplacePath(net.Prefix, route.Path, route.Path) {}

		func (m *RTMockClient) Dispose() {}

		func (m *RTMockClient) Flush() {}