improve performance of getting single neighbor information

avoid calculation of the number of adjout that will not be used.

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

2 files changed, 6 insertions, 3 deletions

diff --git a/server/server.go b/server/server.go
index da6feba2..ed245ff8 100644
--- a/server/server.go
+++ b/server/server.go
@@ -1542,10 +1542,13 @@ func (s *BgpServer) GetServer() (c *config.Global) {
 	return c
 }
 
-func (s *BgpServer) GetNeighbor(getAdvertised bool) (l []*config.Neighbor) {
+func (s *BgpServer) GetNeighbor(address string, getAdvertised bool) (l []*config.Neighbor) {
 	s.mgmtOperation(func() error {
 		l = make([]*config.Neighbor, 0, len(s.neighborMap))
-		for _, peer := range s.neighborMap {
+		for k, peer := range s.neighborMap {
+			if address != "" && address != k {
+				continue
+			}
 			l = append(l, peer.ToConfig(getAdvertised))
 		}
 		return nil
diff --git a/server/server_test.go b/server/server_test.go
index 3aaa7374..9caf8a62 100644
--- a/server/server_test.go
+++ b/server/server_test.go
@@ -109,7 +109,7 @@ func TestMonitor(test *testing.T) {
 
 	for {
 		time.Sleep(time.Second)
-		if t.GetNeighbor(false)[0].State.SessionState == config.SESSION_STATE_ESTABLISHED {
+		if t.GetNeighbor("", false)[0].State.SessionState == config.SESSION_STATE_ESTABLISHED {
 			break
 		}
 	}