mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-27 07:06:42 +00:00
fix : lint
This commit is contained in:
parent
8a6c16dcde
commit
58ef6fdbb7
3 changed files with 32 additions and 8 deletions
|
|
@ -122,6 +122,7 @@ func benchmarkBloomBits(b *testing.B, sectionSize uint64) {
|
|||
|
||||
b.Log("Running filter benchmarks...")
|
||||
start = time.Now()
|
||||
|
||||
var backend *TestBackend
|
||||
|
||||
for i := 0; i < benchFilterCnt; i++ {
|
||||
|
|
|
|||
|
|
@ -10,6 +10,7 @@ import (
|
|||
types "github.com/ethereum/go-ethereum/core/types"
|
||||
"github.com/ethereum/go-ethereum/crypto"
|
||||
"github.com/ethereum/go-ethereum/params"
|
||||
|
||||
gomock "github.com/golang/mock/gomock"
|
||||
)
|
||||
|
||||
|
|
@ -27,6 +28,7 @@ func newTestHeader(blockNumber uint) *types.Header {
|
|||
head := types.Header{
|
||||
Number: big.NewInt(int64(blockNumber)),
|
||||
}
|
||||
|
||||
return &head
|
||||
}
|
||||
|
||||
|
|
@ -40,6 +42,7 @@ func newTestReceipt(contractAddr common.Address, topicAddress common.Hash) *type
|
|||
}
|
||||
|
||||
receipt.Bloom = types.CreateBloom(types.Receipts{receipt})
|
||||
|
||||
return receipt
|
||||
}
|
||||
|
||||
|
|
@ -49,6 +52,7 @@ func (backend *MockBackend) expectBorReceiptsFromMock(hashes []*common.Hash) {
|
|||
backend.EXPECT().GetBorBlockReceipt(gomock.Any(), gomock.Any()).Return(nil, nil)
|
||||
continue
|
||||
}
|
||||
|
||||
backend.EXPECT().GetBorBlockReceipt(gomock.Any(), gomock.Any()).Return(newTestReceipt(addr, *hashes[i]), nil)
|
||||
}
|
||||
}
|
||||
|
|
@ -78,9 +82,11 @@ func TestBorFilters(t *testing.T) {
|
|||
|
||||
filter := NewBorBlockLogsRangeFilter(backend, sprint, 0, 18, []common.Address{addr}, [][]common.Hash{{hash1, hash2, hash3, hash4}})
|
||||
logs, err := filter.Logs(context.Background())
|
||||
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
if len(logs) != 4 {
|
||||
t.Error("expected 4 log, got", len(logs))
|
||||
}
|
||||
|
|
@ -119,6 +125,7 @@ func TestBorFilters(t *testing.T) {
|
|||
filter = NewBorBlockLogsRangeFilter(backend, sprint, 1, 16, []common.Address{addr}, [][]common.Hash{{hash1, hash2}})
|
||||
|
||||
logs, _ = filter.Logs(context.Background())
|
||||
|
||||
if len(logs) != 2 {
|
||||
t.Error("expected 2 log, got", len(logs))
|
||||
}
|
||||
|
|
|
|||
|
|
@ -2,7 +2,8 @@ package filters
|
|||
|
||||
import (
|
||||
context "context"
|
||||
"math/rand"
|
||||
"crypto/rand"
|
||||
"math/big"
|
||||
|
||||
common "github.com/ethereum/go-ethereum/common"
|
||||
core "github.com/ethereum/go-ethereum/core"
|
||||
|
|
@ -16,7 +17,6 @@ import (
|
|||
)
|
||||
|
||||
type TestBackend struct {
|
||||
mux *event.TypeMux
|
||||
DB ethdb.Database
|
||||
sections uint64
|
||||
txFeed event.Feed
|
||||
|
|
@ -35,21 +35,23 @@ func (b *TestBackend) BloomStatus() (uint64, uint64) {
|
|||
func (b *TestBackend) GetBorBlockReceipt(ctx context.Context, hash common.Hash) (*types.Receipt, error) {
|
||||
number := rawdb.ReadHeaderNumber(b.DB, hash)
|
||||
if number == nil {
|
||||
return nil, nil
|
||||
return &types.Receipt{}, nil
|
||||
}
|
||||
|
||||
receipt := rawdb.ReadBorReceipt(b.DB, hash, *number)
|
||||
if receipt == nil {
|
||||
return nil, nil
|
||||
return &types.Receipt{}, nil
|
||||
}
|
||||
|
||||
return receipt, nil
|
||||
}
|
||||
|
||||
func (b *TestBackend) GetBorBlockLogs(ctx context.Context, hash common.Hash) ([]*types.Log, error) {
|
||||
receipt, err := b.GetBorBlockReceipt(ctx, hash)
|
||||
if receipt == nil || err != nil {
|
||||
return nil, nil
|
||||
return []*types.Log{}, nil
|
||||
}
|
||||
|
||||
return receipt.Logs, nil
|
||||
}
|
||||
|
||||
|
|
@ -62,25 +64,30 @@ func (b *TestBackend) HeaderByNumber(ctx context.Context, blockNr rpc.BlockNumbe
|
|||
hash common.Hash
|
||||
num uint64
|
||||
)
|
||||
|
||||
if blockNr == rpc.LatestBlockNumber {
|
||||
hash = rawdb.ReadHeadBlockHash(b.DB)
|
||||
number := rawdb.ReadHeaderNumber(b.DB, hash)
|
||||
|
||||
if number == nil {
|
||||
return nil, nil
|
||||
return &types.Header{}, nil
|
||||
}
|
||||
|
||||
num = *number
|
||||
} else {
|
||||
num = uint64(blockNr)
|
||||
hash = rawdb.ReadCanonicalHash(b.DB, num)
|
||||
}
|
||||
|
||||
return rawdb.ReadHeader(b.DB, hash, num), nil
|
||||
}
|
||||
|
||||
func (b *TestBackend) HeaderByHash(ctx context.Context, hash common.Hash) (*types.Header, error) {
|
||||
number := rawdb.ReadHeaderNumber(b.DB, hash)
|
||||
if number == nil {
|
||||
return nil, nil
|
||||
return &types.Header{}, nil
|
||||
}
|
||||
|
||||
return rawdb.ReadHeader(b.DB, hash, *number), nil
|
||||
}
|
||||
|
||||
|
|
@ -88,6 +95,7 @@ func (b *TestBackend) GetReceipts(ctx context.Context, hash common.Hash) (types.
|
|||
if number := rawdb.ReadHeaderNumber(b.DB, hash); number != nil {
|
||||
return rawdb.ReadReceipts(b.DB, hash, *number, params.TestChainConfig), nil
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
|
|
@ -96,12 +104,14 @@ func (b *TestBackend) GetLogs(ctx context.Context, hash common.Hash) ([][]*types
|
|||
if number == nil {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
receipts := rawdb.ReadReceipts(b.DB, hash, *number, params.TestChainConfig)
|
||||
|
||||
logs := make([][]*types.Log, len(receipts))
|
||||
for i, receipt := range receipts {
|
||||
logs[i] = receipt.Logs
|
||||
}
|
||||
|
||||
return logs, nil
|
||||
}
|
||||
|
||||
|
|
@ -141,7 +151,13 @@ func (b *TestBackend) ServiceFilter(ctx context.Context, session *bloombits.Matc
|
|||
|
||||
task.Bitsets = make([][]byte, len(task.Sections))
|
||||
for i, section := range task.Sections {
|
||||
if rand.Int()%4 != 0 { // Handle occasional missing deliveries
|
||||
nBig, err := rand.Int(rand.Reader, big.NewInt(100))
|
||||
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
if nBig.Int64()%4 != 0 { // Handle occasional missing deliveries
|
||||
head := rawdb.ReadCanonicalHash(b.DB, (section+1)*params.BloomBitsBlocks-1)
|
||||
task.Bitsets[i], _ = rawdb.ReadBloomBits(b.DB, task.Bit, section, head)
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue