a550ca3d64
Since we will use peer flap rate to determine how we rate limit, we store this value on disk per peer per channel. This allows us to restart with memory of our peers past behaviour, so we don't give badly behaving peers have a fresh start on restart. Last flap timestamp is stored with our flap count so that we can degrade this all time flap count over time for peers that have not recently flapped.
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package channeldb
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/lightningnetwork/lnd/routing/route"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestFlapCount tests lookup and writing of flap count to disk.
|
|
func TestFlapCount(t *testing.T) {
|
|
db, cleanup, err := MakeTestDB()
|
|
require.NoError(t, err)
|
|
defer cleanup()
|
|
|
|
// Try to read flap count for a peer that we have no records for.
|
|
_, err = db.ReadFlapCount(testPub)
|
|
require.Equal(t, ErrNoPeerBucket, err)
|
|
|
|
var (
|
|
testPub2 = route.Vertex{2, 2, 2}
|
|
peer1FlapCount = &FlapCount{
|
|
Count: 20,
|
|
LastFlap: time.Unix(100, 23),
|
|
}
|
|
peer2FlapCount = &FlapCount{
|
|
Count: 39,
|
|
LastFlap: time.Unix(200, 23),
|
|
}
|
|
)
|
|
|
|
peers := map[route.Vertex]*FlapCount{
|
|
testPub: peer1FlapCount,
|
|
testPub2: peer2FlapCount,
|
|
}
|
|
|
|
err = db.WriteFlapCounts(peers)
|
|
require.NoError(t, err)
|
|
|
|
// Lookup flap count for our first pubkey.
|
|
count, err := db.ReadFlapCount(testPub)
|
|
require.NoError(t, err)
|
|
require.Equal(t, peer1FlapCount, count)
|
|
|
|
// Lookup our flap count for the second peer.
|
|
count, err = db.ReadFlapCount(testPub2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, peer2FlapCount, count)
|
|
}
|