benjamyn/any-sync
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

testPool: add peers

Browse Source
This commit is contained in:
Sergey Cherepanov 2023-02-02 13:18:30 +03:00
parent 3fcbe3aba9
commit 6d2178dfc9
No known key found for this signature in database
GPG Key ID: 87F8EDE8FBDF637C
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
net/rpc/rpctest/pool.go
View File

@ -15,12 +15,15 @@ import (
var ErrCantConnect = errors.New("can't connect to test server") var ErrCantConnect = errors.New("can't connect to test server")
func NewTestPool() *TestPool { func NewTestPool() *TestPool {
return &TestPool{} return &TestPool{
peers: map[string]peer.Peer{},
}
} }
type TestPool struct { type TestPool struct {
ts *TesServer ts *TesServer
mu sync.Mutex peers map[string]peer.Peer
mu sync.Mutex
} }
func (t *TestPool) WithServer(ts *TesServer) *TestPool { func (t *TestPool) WithServer(ts *TesServer) *TestPool {
@ -33,6 +36,9 @@ func (t *TestPool) WithServer(ts *TesServer) *TestPool {
func (t *TestPool) Get(ctx context.Context, id string) (peer.Peer, error) { func (t *TestPool) Get(ctx context.Context, id string) (peer.Peer, error) {
t.mu.Lock() t.mu.Lock()
defer t.mu.Unlock() defer t.mu.Unlock()
if p, ok := t.peers[id]; ok {
return p, nil
}
if t.ts == nil { if t.ts == nil {
return nil, ErrCantConnect return nil, ErrCantConnect
} }
@ -51,6 +57,11 @@ func (t *TestPool) Dial(ctx context.Context, id string) (peer.Peer, error) {
func (t *TestPool) GetOneOf(ctx context.Context, peerIds []string) (peer.Peer, error) { func (t *TestPool) GetOneOf(ctx context.Context, peerIds []string) (peer.Peer, error) {
t.mu.Lock() t.mu.Lock()
defer t.mu.Unlock() defer t.mu.Unlock()
for _, peerId := range peerIds {
if p, ok := t.peers[peerId]; ok {
return p, nil
}
}
if t.ts == nil { if t.ts == nil {
return nil, ErrCantConnect return nil, ErrCantConnect
} }
@ -70,6 +81,12 @@ func (t *TestPool) NewPool(name string) pool.Pool {
return t return t
} }
func (t *TestPool) AddPeer(p peer.Peer) {
t.mu.Lock()
defer t.mu.Unlock()
t.peers[p.Id()] = p
}
func (t *TestPool) Init(a *app.App) (err error) { func (t *TestPool) Init(a *app.App) (err error) {
return nil return nil
} }