github.com/yinchengtsinghua/golang-Eos-dpos-Ethereum@v0.0.0-20190121132951-92cc4225ed8e/p2p/testing/peerpool.go (about)

     1  
     2  //此源码被清华学神尹成大魔王专业翻译分析并修改
     3  //尹成QQ77025077
     4  //尹成微信18510341407
     5  //尹成所在QQ群721929980
     6  //尹成邮箱 yinc13@mails.tsinghua.edu.cn
     7  //尹成毕业于清华大学,微软区块链领域全球最有价值专家
     8  //https://mvp.microsoft.com/zh-cn/PublicProfile/4033620
     9  //版权所有2017 Go Ethereum作者
    10  //此文件是Go以太坊库的一部分。
    11  //
    12  //Go-Ethereum库是免费软件：您可以重新分发它和/或修改
    13  //根据GNU发布的较低通用公共许可证的条款
    14  //自由软件基金会，或者许可证的第3版，或者
    15  //（由您选择）任何更高版本。
    16  //
    17  //Go以太坊图书馆的发行目的是希望它会有用，
    18  //但没有任何保证；甚至没有
    19  //适销性或特定用途的适用性。见
    20  //GNU较低的通用公共许可证，了解更多详细信息。
    21  //
    22  //你应该收到一份GNU较低级别的公共许可证副本
    23  //以及Go以太坊图书馆。如果没有，请参见<http://www.gnu.org/licenses/>。
    24  
    25  package testing
    26  
    27  import (
    28  	"fmt"
    29  	"sync"
    30  
    31  	"github.com/ethereum/go-ethereum/log"
    32  	"github.com/ethereum/go-ethereum/p2p/discover"
    33  )
    34  
    35  type TestPeer interface {
    36  	ID() discover.NodeID
    37  	Drop(error)
    38  }
    39  
    40  //testpeerpool是演示对等连接注册的示例对等池
    41  type TestPeerPool struct {
    42  	lock  sync.Mutex
    43  	peers map[discover.NodeID]TestPeer
    44  }
    45  
    46  func NewTestPeerPool() *TestPeerPool {
    47  	return &TestPeerPool{peers: make(map[discover.NodeID]TestPeer)}
    48  }
    49  
    50  func (p *TestPeerPool) Add(peer TestPeer) {
    51  	p.lock.Lock()
    52  	defer p.lock.Unlock()
    53  	log.Trace(fmt.Sprintf("pp add peer  %v", peer.ID()))
    54  	p.peers[peer.ID()] = peer
    55  
    56  }
    57  
    58  func (p *TestPeerPool) Remove(peer TestPeer) {
    59  	p.lock.Lock()
    60  	defer p.lock.Unlock()
    61  	delete(p.peers, peer.ID())
    62  }
    63  
    64  func (p *TestPeerPool) Has(id discover.NodeID) bool {
    65  	p.lock.Lock()
    66  	defer p.lock.Unlock()
    67  	_, ok := p.peers[id]
    68  	return ok
    69  }
    70  
    71  func (p *TestPeerPool) Get(id discover.NodeID) TestPeer {
    72  	p.lock.Lock()
    73  	defer p.lock.Unlock()
    74  	return p.peers[id]
    75  }