forked from TrueCloudLab/frostfs-node
[#493] cmd/node: Add profiler section to config
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
922d29ff88
commit
561c809fa5
5 changed files with 88 additions and 1 deletions
40
cmd/neofs-node/config/profiler/config.go
Normal file
40
cmd/neofs-node/config/profiler/config.go
Normal file
|
@ -0,0 +1,40 @@
|
|||
package profilerconfig
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/nspcc-dev/neofs-node/cmd/neofs-node/config"
|
||||
)
|
||||
|
||||
const (
|
||||
subsection = "profiler"
|
||||
|
||||
ShutdownTimeoutDefault = 30 * time.Second
|
||||
AddressDefault = ""
|
||||
)
|
||||
|
||||
// ShutdownTimeout returns value of "shutdown_timeout" config parameter
|
||||
// from "profiler" section.
|
||||
//
|
||||
// Returns ShutdownTimeoutDefault if value is not set.
|
||||
func ShutdownTimeout(c *config.Config) time.Duration {
|
||||
v := config.DurationSafe(c.Sub(subsection), "shutdown_timeout")
|
||||
if v != 0 {
|
||||
return v
|
||||
}
|
||||
|
||||
return ShutdownTimeoutDefault
|
||||
}
|
||||
|
||||
// Address returns value of "address" config parameter
|
||||
// from "profiler" section.
|
||||
//
|
||||
// Returns AddressDefault if value is not set.
|
||||
func Address(c *config.Config) string {
|
||||
v := config.StringSafe(c.Sub(subsection), "address")
|
||||
if v != "" {
|
||||
return v
|
||||
}
|
||||
|
||||
return AddressDefault
|
||||
}
|
37
cmd/neofs-node/config/profiler/config_test.go
Normal file
37
cmd/neofs-node/config/profiler/config_test.go
Normal file
|
@ -0,0 +1,37 @@
|
|||
package profilerconfig_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/nspcc-dev/neofs-node/cmd/neofs-node/config"
|
||||
profilerconfig "github.com/nspcc-dev/neofs-node/cmd/neofs-node/config/profiler"
|
||||
configtest "github.com/nspcc-dev/neofs-node/cmd/neofs-node/config/test"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestProfilerSection(t *testing.T) {
|
||||
t.Run("defaults", func(t *testing.T) {
|
||||
to := profilerconfig.ShutdownTimeout(configtest.EmptyConfig())
|
||||
addr := profilerconfig.Address(configtest.EmptyConfig())
|
||||
|
||||
require.Equal(t, profilerconfig.ShutdownTimeoutDefault, to)
|
||||
require.Equal(t, profilerconfig.AddressDefault, addr)
|
||||
})
|
||||
|
||||
const path = "../../../../config/example/node"
|
||||
|
||||
var fileConfigTest = func(c *config.Config) {
|
||||
to := profilerconfig.ShutdownTimeout(c)
|
||||
addr := profilerconfig.Address(c)
|
||||
|
||||
require.Equal(t, 15*time.Second, to)
|
||||
require.Equal(t, "127.0.0.1:6060", addr)
|
||||
}
|
||||
|
||||
configtest.ForEachFileType(path, fileConfigTest)
|
||||
|
||||
t.Run("ENV", func(t *testing.T) {
|
||||
configtest.ForEnvFileType(path, fileConfigTest)
|
||||
})
|
||||
}
|
|
@ -1 +1,4 @@
|
|||
NEOFS_LOGGER_LEVEL=debug
|
||||
NEOFS_LOGGER_LEVEL=debug
|
||||
|
||||
NEOFS_PROFILER_ADDRESS=127.0.0.1:6060
|
||||
NEOFS_PROFILER_SHUTDOWN_TIMEOUT=15s
|
|
@ -1,5 +1,9 @@
|
|||
{
|
||||
"logger": {
|
||||
"level": "debug"
|
||||
},
|
||||
"profiler": {
|
||||
"address": "127.0.0.1:6060",
|
||||
"shutdown_timeout": "15s"
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,2 +1,5 @@
|
|||
logger:
|
||||
level: debug
|
||||
profiler:
|
||||
address: 127.0.0.1:6060
|
||||
shutdown_timeout: 15s
|
||||
|
|
Loading…
Reference in a new issue