fix issue with execd restart_delay being ignored (#7867)
This commit is contained in:
parent
903a065a0d
commit
9f6b7092f2
|
|
@ -18,14 +18,14 @@ type Size int64
|
||||||
type Number float64
|
type Number float64
|
||||||
|
|
||||||
// UnmarshalTOML parses the duration from the TOML config file
|
// UnmarshalTOML parses the duration from the TOML config file
|
||||||
func (d Duration) UnmarshalTOML(b []byte) error {
|
func (d *Duration) UnmarshalTOML(b []byte) error {
|
||||||
var err error
|
var err error
|
||||||
b = bytes.Trim(b, `'`)
|
b = bytes.Trim(b, `'`)
|
||||||
|
|
||||||
// see if we can directly convert it
|
// see if we can directly convert it
|
||||||
dur, err := time.ParseDuration(string(b))
|
dur, err := time.ParseDuration(string(b))
|
||||||
if err == nil {
|
if err == nil {
|
||||||
d = Duration(dur)
|
*d = Duration(dur)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -33,7 +33,7 @@ func (d Duration) UnmarshalTOML(b []byte) error {
|
||||||
if uq, err := strconv.Unquote(string(b)); err == nil && len(uq) > 0 {
|
if uq, err := strconv.Unquote(string(b)); err == nil && len(uq) > 0 {
|
||||||
dur, err := time.ParseDuration(uq)
|
dur, err := time.ParseDuration(uq)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
d = Duration(dur)
|
*d = Duration(dur)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -42,27 +42,27 @@ func (d Duration) UnmarshalTOML(b []byte) error {
|
||||||
sI, err := strconv.ParseInt(string(b), 10, 64)
|
sI, err := strconv.ParseInt(string(b), 10, 64)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
dur := time.Second * time.Duration(sI)
|
dur := time.Second * time.Duration(sI)
|
||||||
d = Duration(dur)
|
*d = Duration(dur)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
// Second try parsing as float seconds
|
// Second try parsing as float seconds
|
||||||
sF, err := strconv.ParseFloat(string(b), 64)
|
sF, err := strconv.ParseFloat(string(b), 64)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
dur := time.Second * time.Duration(sF)
|
dur := time.Second * time.Duration(sF)
|
||||||
d = Duration(dur)
|
*d = Duration(dur)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s Size) UnmarshalTOML(b []byte) error {
|
func (s *Size) UnmarshalTOML(b []byte) error {
|
||||||
var err error
|
var err error
|
||||||
b = bytes.Trim(b, `'`)
|
b = bytes.Trim(b, `'`)
|
||||||
|
|
||||||
val, err := strconv.ParseInt(string(b), 10, 64)
|
val, err := strconv.ParseInt(string(b), 10, 64)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
s = Size(val)
|
*s = Size(val)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
uq, err := strconv.Unquote(string(b))
|
uq, err := strconv.Unquote(string(b))
|
||||||
|
|
@ -73,16 +73,16 @@ func (s Size) UnmarshalTOML(b []byte) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
s = Size(val)
|
*s = Size(val)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n Number) UnmarshalTOML(b []byte) error {
|
func (n *Number) UnmarshalTOML(b []byte) error {
|
||||||
value, err := strconv.ParseFloat(string(b), 64)
|
value, err := strconv.ParseFloat(string(b), 64)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
n = Number(value)
|
*n = Number(value)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -41,10 +41,10 @@ const sampleConfig = `
|
||||||
`
|
`
|
||||||
|
|
||||||
type Execd struct {
|
type Execd struct {
|
||||||
Command []string
|
Command []string `toml:"command"`
|
||||||
Signal string
|
Signal string `toml:"signal"`
|
||||||
RestartDelay config.Duration
|
RestartDelay config.Duration `toml:"restart_delay"`
|
||||||
Log telegraf.Logger
|
Log telegraf.Logger `toml:"-"`
|
||||||
|
|
||||||
process *process.Process
|
process *process.Process
|
||||||
acc telegraf.Accumulator
|
acc telegraf.Accumulator
|
||||||
|
|
|
||||||
|
|
@ -24,6 +24,24 @@ import (
|
||||||
"github.com/influxdata/telegraf"
|
"github.com/influxdata/telegraf"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func TestSettingConfigWorks(t *testing.T) {
|
||||||
|
cfg := `
|
||||||
|
[[inputs.execd]]
|
||||||
|
command = ["a", "b", "c"]
|
||||||
|
restart_delay = "1m"
|
||||||
|
signal = "SIGHUP"
|
||||||
|
`
|
||||||
|
conf := config.NewConfig()
|
||||||
|
require.NoError(t, conf.LoadConfigData([]byte(cfg)))
|
||||||
|
|
||||||
|
require.Len(t, conf.Inputs, 1)
|
||||||
|
inp, ok := conf.Inputs[0].Input.(*Execd)
|
||||||
|
require.True(t, ok)
|
||||||
|
require.EqualValues(t, []string{"a", "b", "c"}, inp.Command)
|
||||||
|
require.EqualValues(t, 1*time.Minute, inp.RestartDelay)
|
||||||
|
require.EqualValues(t, "SIGHUP", inp.Signal)
|
||||||
|
}
|
||||||
|
|
||||||
func TestExternalInputWorks(t *testing.T) {
|
func TestExternalInputWorks(t *testing.T) {
|
||||||
influxParser, err := parsers.NewInfluxParser()
|
influxParser, err := parsers.NewInfluxParser()
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue