send ping message to new peers

Signed-off-by: Leo Chen <leo@harmony.one>
pull/372/head
Leo Chen 6 years ago
parent 47545789c2
commit ac26de397f
  1. 16
      api/service/discovery/service.go

@ -2,10 +2,10 @@ package discovery
import ( import (
"context" "context"
"io"
"sync" "sync"
"github.com/ethereum/go-ethereum/log" "github.com/ethereum/go-ethereum/log"
proto_discovery "github.com/harmony-one/harmony/api/proto/discovery"
"github.com/harmony-one/harmony/internal/utils" "github.com/harmony-one/harmony/internal/utils"
"github.com/harmony-one/harmony/p2p" "github.com/harmony-one/harmony/p2p"
@ -26,8 +26,6 @@ type Service struct {
Host p2p.Host Host p2p.Host
DHT *libp2pdht.IpfsDHT DHT *libp2pdht.IpfsDHT
Rendezvous string Rendezvous string
reader io.Reader
writer io.Writer
ctx context.Context ctx context.Context
} }
@ -75,9 +73,11 @@ func (s *Service) StartService() {
if peer.ID == s.Host.GetP2PHost().ID() { if peer.ID == s.Host.GetP2PHost().ID() {
continue continue
} }
log.Debug("Found peer", "adding peer", peer) log.Debug("Found Peer", "peer", peer.ID, "addr", peer.Addrs)
p := p2p.Peer{PeerID: peer.ID, Addrs: peer.Addrs} p := p2p.Peer{PeerID: peer.ID, Addrs: peer.Addrs}
s.Host.AddPeer(&p) s.Host.AddPeer(&p)
// TODO: stop ping if pinged before
s.pingPeer(p)
} }
select {} select {}
@ -116,3 +116,11 @@ func (s *Service) Init() error {
return nil return nil
} }
func (s *Service) pingPeer(peer p2p.Peer) {
ping := proto_discovery.NewPingMessage(s.Host.GetSelfPeer())
buffer := ping.ConstructPingMessage()
log.Debug("Sending Ping Message to", "peer", peer)
s.Host.SendMessage(peer, buffer)
log.Debug("Sent Ping Message to", "peer", peer)
}

Loading…
Cancel
Save