TUN-3476: Fix conversion to string and int slice
This commit is contained in:
parent
ed54d150fe
commit
f0cfad8efa
|
@ -263,8 +263,16 @@ func (c *configFileSettings) String(name string) (string, error) {
|
||||||
|
|
||||||
func (c *configFileSettings) StringSlice(name string) ([]string, error) {
|
func (c *configFileSettings) StringSlice(name string) ([]string, error) {
|
||||||
if raw, ok := c.Settings[name]; ok {
|
if raw, ok := c.Settings[name]; ok {
|
||||||
if v, ok := raw.([]string); ok {
|
if slice, ok := raw.([]interface{}); ok {
|
||||||
return v, nil
|
strSlice := make([]string, len(slice))
|
||||||
|
for i, v := range slice {
|
||||||
|
str, ok := v.(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("expected string, found %T for %v", i, v)
|
||||||
|
}
|
||||||
|
strSlice[i] = str
|
||||||
|
}
|
||||||
|
return strSlice, nil
|
||||||
}
|
}
|
||||||
return nil, fmt.Errorf("expected string slice found %T for %s", raw, name)
|
return nil, fmt.Errorf("expected string slice found %T for %s", raw, name)
|
||||||
}
|
}
|
||||||
|
@ -273,6 +281,17 @@ func (c *configFileSettings) StringSlice(name string) ([]string, error) {
|
||||||
|
|
||||||
func (c *configFileSettings) IntSlice(name string) ([]int, error) {
|
func (c *configFileSettings) IntSlice(name string) ([]int, error) {
|
||||||
if raw, ok := c.Settings[name]; ok {
|
if raw, ok := c.Settings[name]; ok {
|
||||||
|
if slice, ok := raw.([]interface{}); ok {
|
||||||
|
intSlice := make([]int, len(slice))
|
||||||
|
for i, v := range slice {
|
||||||
|
str, ok := v.(int)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("expected int, found %T for %v ", v, v)
|
||||||
|
}
|
||||||
|
intSlice[i] = str
|
||||||
|
}
|
||||||
|
return intSlice, nil
|
||||||
|
}
|
||||||
if v, ok := raw.([]int); ok {
|
if v, ok := raw.([]int); ok {
|
||||||
return v, nil
|
return v, nil
|
||||||
}
|
}
|
||||||
|
@ -322,7 +341,6 @@ func ReadConfigFile(c *cli.Context, log logger.Service) (*configFileSettings, er
|
||||||
if err := yaml.NewDecoder(file).Decode(&configuration); err != nil {
|
if err := yaml.NewDecoder(file).Decode(&configuration); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
configuration.sourceFile = configFile
|
configuration.sourceFile = configFile
|
||||||
return &configuration, nil
|
return &configuration, nil
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,76 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"gopkg.in/yaml.v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestConfigFileSettings(t *testing.T) {
|
||||||
|
var (
|
||||||
|
firstIngress = UnvalidatedIngressRule{
|
||||||
|
Hostname: "tunnel1.example.com",
|
||||||
|
Path: "/id",
|
||||||
|
Service: "https://localhost:8000",
|
||||||
|
}
|
||||||
|
secondIngress = UnvalidatedIngressRule{
|
||||||
|
Hostname: "*",
|
||||||
|
Path: "",
|
||||||
|
Service: "https://localhost:8001",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
rawYAML := `
|
||||||
|
tunnel: config-file-test
|
||||||
|
ingress:
|
||||||
|
- hostname: tunnel1.example.com
|
||||||
|
path: /id
|
||||||
|
service: https://localhost:8000
|
||||||
|
- hostname: "*"
|
||||||
|
service: https://localhost:8001
|
||||||
|
retries: 5
|
||||||
|
grace-period: 30s
|
||||||
|
percentage: 3.14
|
||||||
|
hostname: example.com
|
||||||
|
tag:
|
||||||
|
- test
|
||||||
|
- central-1
|
||||||
|
counters:
|
||||||
|
- 123
|
||||||
|
- 456
|
||||||
|
`
|
||||||
|
var config configFileSettings
|
||||||
|
err := yaml.Unmarshal([]byte(rawYAML), &config)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
assert.Equal(t, "config-file-test", config.TunnelID)
|
||||||
|
assert.Equal(t, firstIngress, config.Ingress[0])
|
||||||
|
assert.Equal(t, secondIngress, config.Ingress[1])
|
||||||
|
|
||||||
|
retries, err := config.Int("retries")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, 5, retries)
|
||||||
|
|
||||||
|
gracePeriod, err := config.Duration("grace-period")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, time.Second*30, gracePeriod)
|
||||||
|
|
||||||
|
percentage, err := config.Float64("percentage")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, 3.14, percentage)
|
||||||
|
|
||||||
|
hostname, err := config.String("hostname")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, "example.com", hostname)
|
||||||
|
|
||||||
|
tags, err := config.StringSlice("tag")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, "test", tags[0])
|
||||||
|
assert.Equal(t, "central-1", tags[1])
|
||||||
|
|
||||||
|
counters, err := config.IntSlice("counters")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, 123, counters[0])
|
||||||
|
assert.Equal(t, 456, counters[1])
|
||||||
|
}
|
|
@ -537,25 +537,24 @@ func forceSetFlag(c *cli.Context, name, value string) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func SetFlagsFromConfigFile(c *cli.Context) error {
|
func SetFlagsFromConfigFile(c *cli.Context) error {
|
||||||
|
const exitCode = 1
|
||||||
log, err := createLogger(c, false, false)
|
log, err := createLogger(c, false, false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return cliutil.PrintLoggerSetupError("error setting up logger", err)
|
return cliutil.PrintLoggerSetupError("error setting up logger", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
inputSource, err := config.ReadConfigFile(c, log)
|
inputSource, err := config.ReadConfigFile(c, log)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == config.ErrNoConfigFile {
|
if err == config.ErrNoConfigFile {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return err
|
return cli.Exit(err, exitCode)
|
||||||
}
|
}
|
||||||
targetFlags := c.Command.Flags
|
targetFlags := c.Command.Flags
|
||||||
if c.Command.Name == "" {
|
if c.Command.Name == "" {
|
||||||
targetFlags = c.App.Flags
|
targetFlags = c.App.Flags
|
||||||
}
|
}
|
||||||
if err := altsrc.ApplyInputSourceValues(c, inputSource, targetFlags); err != nil {
|
if err := altsrc.ApplyInputSourceValues(c, inputSource, targetFlags); err != nil {
|
||||||
log.Errorf("Cannot load configuration from %s: %v", inputSource.Source(), err)
|
return cli.Exit(err, exitCode)
|
||||||
return err
|
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue