forked from TrueCloudLab/frostfs-node
[#238] node: Read cfg from dir even if cfg file not set
Signed-off-by: Anton Nikiforov <an.nikiforov@yadro.com>
This commit is contained in:
parent
8d2f443868
commit
995db117d0
3 changed files with 34 additions and 15 deletions
|
@ -7,6 +7,7 @@ Changelog for FrostFS Node
|
||||||
### Changed
|
### Changed
|
||||||
### Fixed
|
### Fixed
|
||||||
- Take network settings into account during netmap contract update (#100)
|
- Take network settings into account during netmap contract update (#100)
|
||||||
|
- Read config files from dir even if config file not provided via `--config` for node (#238)
|
||||||
|
|
||||||
### Removed
|
### Removed
|
||||||
### Updated
|
### Updated
|
||||||
|
|
24
cmd/frostfs-node/config/configdir_test.go
Normal file
24
cmd/frostfs-node/config/configdir_test.go
Normal file
|
@ -0,0 +1,24 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/spf13/cast"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestConfigDir(t *testing.T) {
|
||||||
|
dir := t.TempDir()
|
||||||
|
|
||||||
|
cfgFileName0 := path.Join(dir, "cfg_00.json")
|
||||||
|
cfgFileName1 := path.Join(dir, "cfg_01.yml")
|
||||||
|
|
||||||
|
require.NoError(t, os.WriteFile(cfgFileName0, []byte(`{"storage":{"shard_pool_size":15}}`), 0777))
|
||||||
|
require.NoError(t, os.WriteFile(cfgFileName1, []byte("logger:\n level: debug"), 0777))
|
||||||
|
|
||||||
|
c := New(Prm{}, WithConfigDir(dir))
|
||||||
|
require.Equal(t, "debug", cast.ToString(c.Sub("logger").Value("level")))
|
||||||
|
require.EqualValues(t, 15, cast.ToUint32(c.Sub("storage").Value("shard_pool_size")))
|
||||||
|
}
|
|
@ -1,6 +1,7 @@
|
||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
|
|
||||||
|
@ -20,7 +21,7 @@ func ReadConfigDir(v *viper.Viper, configDir string) error {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
ext := path.Ext(entry.Name())
|
ext := path.Ext(entry.Name())
|
||||||
if ext != ".yaml" && ext != ".yml" {
|
if ext != ".yaml" && ext != ".yml" && ext != ".json" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -33,22 +34,15 @@ func ReadConfigDir(v *viper.Viper, configDir string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
// mergeConfig reads config file and merge its content with current viper.
|
// mergeConfig reads config file and merge its content with current viper.
|
||||||
func mergeConfig(v *viper.Viper, fileName string) (err error) {
|
func mergeConfig(v *viper.Viper, fileName string) error {
|
||||||
var cfgFile *os.File
|
cv := viper.New()
|
||||||
cfgFile, err = os.Open(fileName)
|
cv.SetConfigFile(fileName)
|
||||||
|
err := cv.ReadInConfig()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("failed to read config: %w", err)
|
||||||
}
|
}
|
||||||
|
if err = v.MergeConfigMap(cv.AllSettings()); err != nil {
|
||||||
defer func() {
|
return fmt.Errorf("failed to merge config: %w", err)
|
||||||
errClose := cfgFile.Close()
|
|
||||||
if err == nil {
|
|
||||||
err = errClose
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
if err = v.MergeConfig(cfgFile); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|
Loading…
Reference in a new issue