swarm/storage: fix mockNetFetcher data races (#18462)

fixes: ethersphere/go-ethereum#1117
(cherry picked from commit f728837ee6)
This commit is contained in:
Janoš Guljaš 2019-01-16 14:31:32 +01:00 committed by Rafael Matias
parent df355eceb4
commit dd3e894747
No known key found for this signature in database
GPG Key ID: 1BC39532FB4A2DBD
1 changed files with 4 additions and 0 deletions

View File

@ -43,6 +43,7 @@ type mockNetFetcher struct {
quit <-chan struct{}
ctx context.Context
hopCounts []uint8
mu sync.Mutex
}
func (m *mockNetFetcher) Offer(ctx context.Context, source *enode.ID) {
@ -51,6 +52,9 @@ func (m *mockNetFetcher) Offer(ctx context.Context, source *enode.ID) {
}
func (m *mockNetFetcher) Request(ctx context.Context, hopCount uint8) {
m.mu.Lock()
defer m.mu.Unlock()
m.requestCalled = true
var peers []Address
m.peers.Range(func(key interface{}, _ interface{}) bool {