netsync add test case (#365)

[bytom/vapor.git] / netsync / peers / peer.go

diff --git a/netsync/peers/peer.go b/netsync/peers/peer.go
index 431de06..ef90812 100644 (file)
--- a/netsync/peers/peer.go
+++ b/netsync/peers/peer.go
@@ -673,6 +673,15 @@ func (ps *PeerSet) SetStatus(peerID string, height uint64, hash *bc.Hash) {
        peer.SetBestStatus(height, hash)
 }
 
        peer.SetBestStatus(height, hash)
 }
 

+func (ps *PeerSet) SetIrreversibleStatus(peerID string, height uint64, hash *bc.Hash) {
+       peer := ps.GetPeer(peerID)
+       if peer == nil {
+               return
+       }
+
+       peer.SetIrreversibleStatus(height, hash)
+}
+

 func (ps *PeerSet) Size() int {
        ps.mtx.RLock()
        defer ps.mtx.RUnlock()
 func (ps *PeerSet) Size() int {
        ps.mtx.RLock()
        defer ps.mtx.RUnlock()