diff options
author | Marcus Wichelmann <mail@marcusw.de> | 2021-02-20 21:25:39 +0100 |
---|---|---|
committer | Marcus Wichelmann <mail@marcusw.de> | 2021-03-29 20:53:53 +0200 |
commit | 3c437e5aecab18454c5d70fd35272060bc3c95cf (patch) | |
tree | 489423ce9d54a5502fe58c6db1072c003c3efb05 /pkg/server/grpc_server.go | |
parent | 76f40ede0d466dfc7b1151d4e36b619a0daae51e (diff) |
Added ListDynamicNeighbor message to API
Diffstat (limited to 'pkg/server/grpc_server.go')
-rw-r--r-- | pkg/server/grpc_server.go | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/pkg/server/grpc_server.go b/pkg/server/grpc_server.go index 1ae58055..b0cae8d3 100644 --- a/pkg/server/grpc_server.go +++ b/pkg/server/grpc_server.go @@ -102,6 +102,18 @@ func (s *server) serve() error { return nil } +func (s *server) ListDynamicNeighbor(r *api.ListDynamicNeighborRequest, stream api.GobgpApi_ListDynamicNeighborServer) error { + ctx, cancel := context.WithCancel(context.Background()) + defer cancel() + fn := func(dn *api.DynamicNeighbor) { + if err := stream.Send(&api.ListDynamicNeighborResponse{DynamicNeighbor: dn}); err != nil { + cancel() + return + } + } + return s.bgpServer.ListDynamicNeighbor(ctx, r, fn) +} + func (s *server) ListPeerGroup(r *api.ListPeerGroupRequest, stream api.GobgpApi_ListPeerGroupServer) error { ctx, cancel := context.WithCancel(context.Background()) defer cancel() @@ -861,6 +873,10 @@ func (s *server) AddDynamicNeighbor(ctx context.Context, r *api.AddDynamicNeighb return &empty.Empty{}, s.bgpServer.AddDynamicNeighbor(ctx, r) } +func (s *server) DeleteDynamicNeighbor(ctx context.Context, r *api.DeleteDynamicNeighborRequest) (*empty.Empty, error) { + return &empty.Empty{}, s.bgpServer.DeleteDynamicNeighbor(ctx, r) +} + func newPrefixFromApiStruct(a *api.Prefix) (*table.Prefix, error) { _, prefix, err := net.ParseCIDR(a.IpPrefix) if err != nil { |