diff --git a/pkg/innerring/processors/netmap/cleanup_table.go b/pkg/innerring/processors/netmap/cleanup_table.go index 2033946547..7a41d7a80b 100644 --- a/pkg/innerring/processors/netmap/cleanup_table.go +++ b/pkg/innerring/processors/netmap/cleanup_table.go @@ -4,7 +4,7 @@ import ( "encoding/hex" "sync" - "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc" + netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc" ) type ( diff --git a/pkg/innerring/processors/netmap/cleanup_table_test.go b/pkg/innerring/processors/netmap/cleanup_table_test.go index 7a9c3dea35..4f1375ef13 100644 --- a/pkg/innerring/processors/netmap/cleanup_table_test.go +++ b/pkg/innerring/processors/netmap/cleanup_table_test.go @@ -4,20 +4,20 @@ import ( "encoding/hex" "testing" - "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc" + netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc" crypto "github.com/nspcc-dev/neofs-crypto" "github.com/nspcc-dev/neofs-node/pkg/util/test" "github.com/stretchr/testify/require" ) func TestCleanupTable(t *testing.T) { - var infos = []netmap.NodeInfo{ + infos := []netmap.NodeInfo{ {PublicKey: crypto.MarshalPublicKey(&test.DecodeKey(1).PublicKey)}, {PublicKey: crypto.MarshalPublicKey(&test.DecodeKey(2).PublicKey)}, {PublicKey: crypto.MarshalPublicKey(&test.DecodeKey(3).PublicKey)}, } - var mapInfos = map[string]struct{}{ + mapInfos := map[string]struct{}{ hex.EncodeToString(infos[0].PublicKey): {}, hex.EncodeToString(infos[1].PublicKey): {}, hex.EncodeToString(infos[2].PublicKey): {}, @@ -110,6 +110,5 @@ func TestCleanupTable(t *testing.T) { })) require.EqualValues(t, len(infos)-1, cnt) }) - }) } diff --git a/pkg/innerring/processors/netmap/processor.go b/pkg/innerring/processors/netmap/processor.go index 2827523b0f..e406d08967 100644 --- a/pkg/innerring/processors/netmap/processor.go +++ b/pkg/innerring/processors/netmap/processor.go @@ -38,17 +38,21 @@ type ( epochState EpochState activeState ActiveState morphClient *client.Client + + netmapSnapshot cleanupTable } // Params of the processor constructor. Params struct { - Log *zap.Logger - PoolSize int - NetmapContract util.Uint160 - EpochTimer EpochTimerReseter - MorphClient *client.Client - EpochState EpochState - ActiveState ActiveState + Log *zap.Logger + PoolSize int + NetmapContract util.Uint160 + EpochTimer EpochTimerReseter + MorphClient *client.Client + EpochState EpochState + ActiveState ActiveState + CleanupEnabled bool + CleanupThreshold uint64 // in epochs } ) @@ -88,6 +92,7 @@ func New(p *Params) (*Processor, error) { epochState: p.EpochState, activeState: p.ActiveState, morphClient: p.MorphClient, + netmapSnapshot: newCleanupTable(p.CleanupEnabled, p.CleanupThreshold), }, nil }