mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-11-10 14:17:56 +01:00
dee8ad3754
Use the new feature of Go 1.24, fix linter warnings. This change was produced by: - running golangci-lint run --fix - sed 's/context.Background/t.Context/' -i `git grep -l context.Background | grep test.go` - manually fixing broken tests - itest, lntest: use ht.Context() where ht or hn is available - in HarnessNode.Stop() we keep using context.Background(), because it is called from a cleanup handler in which t.Context() is canceled already.
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
//go:build kvdb_etcd
|
|
// +build kvdb_etcd
|
|
|
|
package etcd
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestCommitQueue tests that non-conflicting transactions commit concurrently,
|
|
// while conflicting transactions are queued up.
|
|
func TestCommitQueue(t *testing.T) {
|
|
// The duration of each commit.
|
|
const commitDuration = time.Millisecond * 500
|
|
const numCommits = 5
|
|
|
|
var wg sync.WaitGroup
|
|
commits := make([]string, numCommits)
|
|
idx := int32(-1)
|
|
|
|
commit := func(tag string, sleep bool) func() {
|
|
return func() {
|
|
defer wg.Done()
|
|
|
|
// Update our log of commit order. Avoid blocking
|
|
// by preallocating the commit log and increasing
|
|
// the log index atomically.
|
|
if sleep {
|
|
time.Sleep(commitDuration)
|
|
}
|
|
|
|
i := atomic.AddInt32(&idx, 1)
|
|
commits[i] = tag
|
|
}
|
|
}
|
|
|
|
ctx := t.Context()
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
q := NewCommitQueue(ctx)
|
|
defer q.Stop()
|
|
defer cancel()
|
|
|
|
wg.Add(numCommits)
|
|
t1 := time.Now()
|
|
|
|
// Tx1 (long): reads: key1, key2, writes: key3, conflict: none
|
|
q.Add(
|
|
commit("free", true),
|
|
[]string{"key1", "key2"},
|
|
[]string{"key3"},
|
|
)
|
|
// Tx2: reads: key1, key2, writes: key3, conflict: Tx1
|
|
q.Add(
|
|
commit("blocked1", false),
|
|
[]string{"key1", "key2"},
|
|
[]string{"key3"},
|
|
)
|
|
// Tx3 (long): reads: key1, writes: key4, conflict: none
|
|
q.Add(
|
|
commit("free", true),
|
|
[]string{"key1", "key2"},
|
|
[]string{"key4"},
|
|
)
|
|
// Tx4 (long): reads: key1, writes: none, conflict: none
|
|
q.Add(
|
|
commit("free", true),
|
|
[]string{"key1", "key2"},
|
|
[]string{},
|
|
)
|
|
// Tx4: reads: key2, writes: key4 conflict: Tx3
|
|
q.Add(
|
|
commit("blocked2", false),
|
|
[]string{"key2"},
|
|
[]string{"key4"},
|
|
)
|
|
|
|
// Wait for all commits.
|
|
wg.Wait()
|
|
t2 := time.Now()
|
|
|
|
// Expected total execution time: delta.
|
|
// 2 * commitDuration <= delta < 3 * commitDuration
|
|
delta := t2.Sub(t1)
|
|
require.LessOrEqual(t, int64(commitDuration*2), int64(delta))
|
|
require.Greater(t, int64(commitDuration*3), int64(delta))
|
|
|
|
// Expect that the non-conflicting "free" transactions are executed
|
|
// before the blocking ones, and the blocking ones are executed in
|
|
// the order of addition.
|
|
require.Equal(t,
|
|
[]string{"free", "blocked1", "free", "free", "blocked2"},
|
|
commits,
|
|
)
|
|
}
|