p17860254
/
categraf
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add args: interval

This commit is contained in:
Ulric Qin 2022-07-11 20:40:30 +08:00
parent 7dd967a79d
commit 3443d71efe
3 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
agent/metrics_reader.go
View File

@ -51,11 +51,6 @@ func (r *InputReader) startInput(inputName string) {
interval = time.Duration(r.input.GetInterval())
}
// get output quickly
if config.Config.TestMode {
interval = time.Second * 3
}
for {
select {
case <-r.quitChan:

6
config/config.go
View File

@ -61,7 +61,7 @@ type ConfigType struct {
var Config *ConfigType
func InitConfig(configDir string, debugMode bool, testMode bool) error {
func InitConfig(configDir string, debugMode, testMode bool, interval int64) error {
configFile := path.Join(configDir, "config.toml")
if !file.IsExist(configFile) {
return fmt.Errorf("configuration file(%s) not found", configFile)
@ -77,6 +77,10 @@ func InitConfig(configDir string, debugMode bool, testMode bool) error {
return fmt.Errorf("failed to load configs of dir: %s", configDir)
}
if interval > 0 {
Config.Global.Interval = Duration(time.Duration(interval) * time.Second)
}
if err := Config.fillIP(); err != nil {
return err
}

3
main.go
View File

@ -25,6 +25,7 @@ var (
configDir = flag.String("configs", osx.GetEnv("CATEGRAF_CONFIGS", "conf"), "Specify configuration directory.(env:CATEGRAF_CONFIGS)")
debugMode = flag.Bool("debug", false, "Is debug mode?")
testMode = flag.Bool("test", false, "Is test mode? print metrics to stdout")
interval = flag.Int64("interval", 0, "Global interval(unit:Second)")
showVersion = flag.Bool("version", false, "Show version.")
inputFilters = flag.String("inputs", "", "e.g. cpu:mem:system")
@ -59,7 +60,7 @@ func main() {
printEnv()
// init configs
if err := config.InitConfig(*configDir, *debugMode, *testMode); err != nil {
if err := config.InitConfig(*configDir, *debugMode, *testMode, *interval); err != nil {
log.Fatalln("F! failed to init config:", err)
}