e1a443b04a
Instead of providing the database from outside to testplanet create it inside and then allow wrapping and modifying it. This is more convenient to use. Change-Id: I9b8f69e6e0a19ff984b4e2bfe927c9100c77bc6c
62 lines
2.2 KiB
Go
62 lines
2.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package testplanet
|
|
|
|
import (
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/common/identity/testidentity"
|
|
"storj.io/common/pb/pbgrpc"
|
|
"storj.io/common/storj"
|
|
"storj.io/storj/satellite"
|
|
"storj.io/storj/satellite/metainfo"
|
|
"storj.io/storj/storagenode"
|
|
)
|
|
|
|
// Reconfigure allows to change node configurations
|
|
type Reconfigure struct {
|
|
SatelliteDB func(log *zap.Logger, index int, db satellite.DB) (satellite.DB, error)
|
|
SatellitePointerDB func(log *zap.Logger, index int, db metainfo.PointerDB) (metainfo.PointerDB, error)
|
|
Satellite func(log *zap.Logger, index int, config *satellite.Config)
|
|
|
|
ReferralManagerServer func(log *zap.Logger) pbgrpc.ReferralManagerServer
|
|
|
|
StorageNodeDB func(index int, db storagenode.DB, log *zap.Logger) (storagenode.DB, error)
|
|
StorageNode func(index int, config *storagenode.Config)
|
|
UniqueIPCount int
|
|
|
|
Identities func(log *zap.Logger, version storj.IDVersion) *testidentity.Identities
|
|
}
|
|
|
|
// DisablePeerCAWhitelist returns a `Reconfigure` that sets `UsePeerCAWhitelist` for
|
|
// all node types that use kademlia.
|
|
var DisablePeerCAWhitelist = Reconfigure{
|
|
Satellite: func(log *zap.Logger, index int, config *satellite.Config) {
|
|
config.Server.UsePeerCAWhitelist = false
|
|
},
|
|
StorageNode: func(index int, config *storagenode.Config) {
|
|
config.Server.UsePeerCAWhitelist = false
|
|
},
|
|
}
|
|
|
|
// ShortenOnlineWindow returns a `Reconfigure` that sets the NodeSelection
|
|
// OnlineWindow to 1 second, meaning a connection failure leads to marking the nodes as offline
|
|
var ShortenOnlineWindow = Reconfigure{
|
|
Satellite: func(log *zap.Logger, index int, config *satellite.Config) {
|
|
config.Overlay.Node.OnlineWindow = 1 * time.Second
|
|
},
|
|
}
|
|
|
|
// ReconfigureRS returns function to change satellite redundancy scheme values
|
|
var ReconfigureRS = func(minThreshold, repairThreshold, successThreshold, totalThreshold int) func(log *zap.Logger, index int, config *satellite.Config) {
|
|
return func(log *zap.Logger, index int, config *satellite.Config) {
|
|
config.Metainfo.RS.MinThreshold = minThreshold
|
|
config.Metainfo.RS.RepairThreshold = repairThreshold
|
|
config.Metainfo.RS.SuccessThreshold = successThreshold
|
|
config.Metainfo.RS.TotalThreshold = totalThreshold
|
|
}
|
|
}
|