diff --git a/net/service/service.go b/net/service/service.go index c0f3d8be8..caa2e2354 100644 --- a/net/service/service.go +++ b/net/service/service.go @@ -204,3 +204,7 @@ func (s *Service) handleIncomingMessage(ctx context.Context, m msg.NetMessage) { case <-ctx.Done(): } } + +func (s *Service) SetHandler(h Handler) { + s.Handler = h +}