Simplify handler

dot/network/config.go

@@ -66,7 +66,7 @@ type Config struct {
 	// Service interfaces
 	BlockState         BlockState
 	Syncer             Syncer
-	warpSyncHandler    WarpSyncRequestHandler
+	warpSyncProvider   WarpSyncProvider
 	TransactionHandler TransactionHandler
 
 	// Used to specify the address broadcasted to other peers, and avoids using pubip.Get

dot/network/service.go

@@ -130,7 +130,7 @@ type Service struct {
 	blockState         BlockState
 	syncer             Syncer
 	transactionHandler TransactionHandler
-	warpSyncHandler    WarpSyncRequestHandler
+	warpSyncProvider   WarpSyncProvider
 
 	// Configuration options
 	noBootstrap bool
@@ -217,7 +217,7 @@ func NewService(cfg *Config) (*Service, error) {
 		noBootstrap:            cfg.NoBootstrap,
 		noMDNS:                 cfg.NoMDNS,
 		syncer:                 cfg.Syncer,
-		warpSyncHandler:        cfg.warpSyncHandler,
+		warpSyncProvider:       cfg.warpSyncProvider,
 		notificationsProtocols: make(map[MessageType]*notificationsProtocol),
 		lightRequest:           make(map[peer.ID]struct{}),
 		telemetryInterval:      cfg.telemetryInterval,

dot/network/warp_sync.go

@@ -2,24 +2,13 @@ package network
 
 import (
 	"github.com/ChainSafe/gossamer/dot/network/messages"
-	"github.com/ChainSafe/gossamer/lib/common"
 	libp2pnetwork "github.com/libp2p/go-libp2p/core/network"
 	"github.com/libp2p/go-libp2p/core/peer"
 )
 
-type WarpSyncProvider interface {
-	// Generate proof starting at given block hash. The proof is accumulated until maximum proof
-	// size is reached.
-	generate(start common.Hash) (encodedProof []byte, err error)
-}
-
-type WarpSyncRequestHandler struct {
-	backend WarpSyncProvider
-}
-
-func (w *WarpSyncRequestHandler) handleRequest(req messages.WarpProofRequest) ([]byte, error) {
+func (s *Service) handleWarpSyncRequest(req messages.WarpProofRequest) ([]byte, error) {
 	// use the backend to generate the warp proof
-	proof, err := w.backend.generate(req.Begin)
+	proof, err := s.warpSyncProvider.generate(req.Begin)
 	if err != nil {
 		return nil, err
 	}
@@ -54,7 +43,7 @@ func (s *Service) handleWarpSyncMessage(stream libp2pnetwork.Stream, msg message
 	}()
 
 	if req, ok := msg.(*messages.WarpProofRequest); ok {
-		resp, err := s.warpSyncHandler.handleRequest(*req)
+		resp, err := s.handleWarpSyncRequest(*req)
 		if err != nil {
 			logger.Debugf("cannot create response for request: %s", err)
 			return nil