From 99d02d5de0946020c6332238b86319905b5a91a4 Mon Sep 17 00:00:00 2001 From: Evgenii Stratonikov Date: Fri, 13 Mar 2020 10:03:58 +0300 Subject: [PATCH] rpc: move getpeers RPC to a separate func --- pkg/rpc/server/server.go | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/pkg/rpc/server/server.go b/pkg/rpc/server/server.go index 27e8ee4e8..84e2e3620 100644 --- a/pkg/rpc/server/server.go +++ b/pkg/rpc/server/server.go @@ -251,11 +251,7 @@ Methods: } case "getpeers": - peers := result.NewGetPeers() - peers.AddUnconnected(s.coreServer.UnconnectedPeers()) - peers.AddConnected(s.coreServer.ConnectedPeers()) - peers.AddBad(s.coreServer.BadPeers()) - results = peers + results, resultsErr = s.getPeers(reqParams) case "getrawmempool": results, resultsErr = s.getRawMempool(reqParams) @@ -317,6 +313,14 @@ Methods: s.WriteResponse(req, w, results) } +func (s *Server) getPeers(_ request.Params) (interface{}, error) { + peers := result.NewGetPeers() + peers.AddUnconnected(s.coreServer.UnconnectedPeers()) + peers.AddConnected(s.coreServer.ConnectedPeers()) + peers.AddBad(s.coreServer.BadPeers()) + return peers, nil +} + func (s *Server) getRawMempool(_ request.Params) (interface{}, error) { mp := s.chain.GetMemPool() hashList := make([]util.Uint256, 0)