send msg to peer

node/node.go

@@ -268,8 +268,10 @@ func (n *Node) runTimeProof(sig <-chan struct{}, wait chan<- struct{}) {
 }
 
 func (n Node) broadcastMsg(msg *core.Message) error {
-	for pid, peer := range n.peers {
-		log.Info("peer id", common.Hash2String(pid), "url", peer.Url())
+	for _, peer := range n.peers {
+		if err := peer.SendMsg(msg); err != nil {
+			return err
+		}
 	}
 	return nil
 }

peer/peer.go

@@ -17,10 +17,17 @@
 package peer
 
 import (
+	"encoding/json"
+	"errors"
 	"fmt"
+	"github.com/pdupub/go-pdu/core"
 	"golang.org/x/net/websocket"
 )
 
+var (
+	errPeerNotReachable = errors.New("this peer not reachable right now")
+)
+
 // Peer contain the info of websocket connection
 type Peer struct {
 	IP      string `json:"ip"`
@@ -57,6 +64,31 @@ func (p Peer) Url() string {
 	return fmt.Sprintf("ws://%s:%d/%s", p.IP, p.Port, p.NodeKey)
 }
 
+// Connected return true if this peer is connected right now
+func (p *Peer) Connected() bool {
+	if p.conn != nil {
+		return true
+	}
+	return false
+}
+
+// SendMsg is used to send msg to this peer
+func (p *Peer) SendMsg(msg *core.Message) error {
+	if !p.Connected() {
+		return errPeerNotReachable
+	}
+
+	msgBytes, err := json.Marshal(msg)
+	if err != nil {
+		return err
+	}
+	_, err = p.conn.Write(msgBytes)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 // origin used when peer dial
 func (p Peer) origin() string {
 	return fmt.Sprintf("http://%s:%d/", p.IP, p.Port)