forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-20 10:52:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

swarm/pss: minor refactoring

Browse source
This commit is contained in:
Vlad 2019-04-16 14:39:45 +02:00
parent c5e2a6dc7a
commit 0b23499afb
1 changed files with 18 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
swarm/pss/prox_test.go
View file

@ -63,15 +63,15 @@ func (td *testData) pushNotification(val handlerNotification) {
td.mu.Unlock()
}
func (td *testData) popNotification() (first handlerNotification, ok bool) {
func (td *testData) popNotification() (first handlerNotification, exist bool) {
td.mu.Lock()
if len(td.notifications) > 0 {
ok = true
exist = true
first = td.notifications[0]
td.notifications = td.notifications[1:]
}
td.mu.Unlock()
return first, ok
return first, exist
}
func (td *testData) getMsgCount() int {
@ -272,33 +272,34 @@ func (td *testData) sendAllMsgs() error {
return nil
}
// testRoutine is the main test function, called by Simulation.Run()
func testRoutine(td *testData, ctx context.Context) error {
if err := td.sendAllMsgs(); err != nil {
return err
}
isMoreTimeLeft := func() bool {
func isMoreTimeLeft(ctx context.Context) bool {
select {
case <-ctx.Done():
return false
default:
return true
}
}
}
// testRoutine is the main test function, called by Simulation.Run()
func testRoutine(td *testData, ctx context.Context) error {
hasMoreRound := func(err error, hadMessage bool) bool {
return err == nil && (hadMessage || isMoreTimeLeft())
return err == nil && (hadMessage || isMoreTimeLeft(ctx))
}
if err := td.sendAllMsgs(); err != nil {
return err
}
var err error
received := 0
hadMessage := false
for oneMoreRound := true; oneMoreRound; oneMoreRound = hasMoreRound(err, hadMessage) {
message, hadMessage := td.popNotification()
if !isMoreTimeLeft() {
if !isMoreTimeLeft(ctx) {
// Stop handlers from sending more messages.
// Note: only best effort, race is possible.
td.setDone()
@ -314,8 +315,6 @@ func testRoutine(td *testData, ctx context.Context) error {
} else {
time.Sleep(32 * time.Millisecond)
}
oneMoreRound = err == nil && (hadMessage || isMoreTimeLeft())
}
if err != nil {
@ -326,7 +325,6 @@ func testRoutine(td *testData, ctx context.Context) error {
return ctx.Err()
}
return nil
}
func (td *testData) isAllowedMessage(n handlerNotification) bool {