be5b2d46a5
In this commit, we’ve moved away from the internal queryHandler within
the packetQueue entirely. We now use an internal queueLen variable
internally to allow callers to sample the queue’s size, and also for
synchronization purposes internally.
This commit also introduces a chan struct{} (freeSlots) that is used
internally as a semaphore. The current value of freeSlots reflects the
number of available slots within the commitment transaction. Within the
link, after an HTLC has been removed/modified, then a “slot” is freed
up. The main packetConsumer then interprets these messages as a signal
to attempt to free up a new slot within the queue itself by dumping off
to the commitment transaction.
49 lines
941 B
Go
49 lines
941 B
Go
package htlcswitch
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
)
|
|
|
|
// TestWaitingQueueThreadSafety test the thread safety properties of the
|
|
// waiting queue, by executing methods in separate goroutines which operates
|
|
// with the same data.
|
|
func TestWaitingQueueThreadSafety(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
const numPkts = 1000
|
|
|
|
q := newPacketQueue(numPkts)
|
|
q.Start()
|
|
defer q.Stop()
|
|
|
|
a := make([]lnwire.MilliSatoshi, numPkts)
|
|
for i := 0; i < numPkts; i++ {
|
|
a[i] = lnwire.MilliSatoshi(i)
|
|
q.AddPkt(&htlcPacket{
|
|
amount: lnwire.MilliSatoshi(i),
|
|
htlc: &lnwire.UpdateAddHTLC{},
|
|
})
|
|
}
|
|
|
|
var b []lnwire.MilliSatoshi
|
|
for i := 0; i < numPkts; i++ {
|
|
q.SignalFreeSlot()
|
|
|
|
select {
|
|
case packet := <-q.outgoingPkts:
|
|
b = append(b, packet.amount)
|
|
|
|
case <-time.After(2 * time.Second):
|
|
t.Fatal("timeout")
|
|
}
|
|
}
|
|
|
|
if !reflect.DeepEqual(b, a) {
|
|
t.Fatal("wrong order of the objects")
|
|
}
|
|
}
|