Add HybridRelayer interface

data/txHandler.go

@@ -148,6 +148,11 @@ type TxHandlerOpts struct {
 	Config        config.Local
 }
 
+// HybridRelayer is an interface for relaying p2p transactions to WS network
+type HybridRelayer interface {
+	BridgeP2PToWS(ctx context.Context, tag protocol.Tag, data []byte, wait bool, except network.Peer) error
+}
+
 // MakeTxHandler makes a new handler for transaction messages
 func MakeTxHandler(opts TxHandlerOpts) (*TxHandler, error) {
 
@@ -859,7 +864,9 @@ func (handler *TxHandler) validateIncomingTxMessage(rawmsg network.IncomingMessa
 		logging.Base().Infof("unable to pin transaction: %v", err)
 	}
 
-	_ = handler.net.Relay(handler.ctx, protocol.TxnTag, reencoded, false, wi.rawmsg.Sender)
+	if hybridNet, ok := handler.net.(HybridRelayer); ok {
+		_ = hybridNet.BridgeP2PToWS(handler.ctx, protocol.TxnTag, reencoded, false, wi.rawmsg.Sender)
+	}
 
 	return network.OutgoingMessage{
 		Action: network.Accept,

network/hybridNetwork.go

@@ -125,6 +125,11 @@ func (n *HybridP2PNetwork) Relay(ctx context.Context, tag protocol.Tag, data []b
 	})
 }
 
+// BridgeP2PToWS skips Relay/Broadcast to both networks and only sends to WS
+func (n *HybridP2PNetwork) BridgeP2PToWS(ctx context.Context, tag protocol.Tag, data []byte, wait bool, except Peer) error {
+	return n.wsNetwork.Relay(ctx, tag, data, wait, except)
+}
+
 // Disconnect implements GossipNode
 func (n *HybridP2PNetwork) Disconnect(badnode DisconnectablePeer) {
 	net := badnode.GetNetwork()