From 9eac94d7e747b09705619026877df496b9b51750 Mon Sep 17 00:00:00 2001 From: Thomas Roehl Date: Mon, 20 Dec 2021 17:44:10 +0100 Subject: [PATCH] Update entry points --- metric-collector.go | 23 ++++++----------------- 1 file changed, 6 insertions(+), 17 deletions(-) diff --git a/metric-collector.go b/metric-collector.go index 680ba68..e9b5a00 100644 --- a/metric-collector.go +++ b/metric-collector.go @@ -15,6 +15,7 @@ import ( "time" mr "github.com/ClusterCockpit/cc-metric-collector/internal/metricRouter" lp "github.com/ClusterCockpit/cc-metric-collector/internal/ccMetric" + mct "github.com/ClusterCockpit/cc-metric-collector/internal/multiChanTicker" ) // List of provided collectors. Which collector should be run can be @@ -76,13 +77,10 @@ type RuntimeConfig struct { ConfigFile CentralConfigFile Router mr.MetricRouter - RouterDone chan bool CollectManager collectors.CollectorManager - CollectManagerDone chan bool SinkManager sinks.SinkManager - SinkManagerDone chan bool ReceiveManager receivers.ReceiveManager - ReceiveManagerDone chan bool + Ticker mct.MultiChanTicker Channels []chan lp.CCMetric Sync sync.WaitGroup @@ -252,40 +250,30 @@ func main() { // log.Print("Error setting up logging system to ", rcfg.CliArgs["logfile"], " on ", rcfg.Hostname) // return // } - + rcfg.Ticker = mct.NewTicker(rcfg.Interval) if len(rcfg.ConfigFile.RouterConfigFile) > 0 { - rcfg.RouterDone = make(chan bool) - rcfg.Router, err = mr.New(rcfg.RouterDone, &rcfg.Sync) + rcfg.Router, err = mr.New(rcfg.Ticker, &rcfg.Sync, rcfg.ConfigFile.RouterConfigFile) if err != nil { log.Print(err.Error()) return } - rcfg.Router.ReadConfig(rcfg.ConfigFile.RouterConfigFile) } if len(rcfg.ConfigFile.SinkConfigFile) > 0 { - rcfg.SinkManagerDone = make(chan bool) rcfg.SinkManager, err = sinks.New(&rcfg.Sync, rcfg.ConfigFile.SinkConfigFile) if err != nil { log.Print(err.Error()) return } -// rcfg.SinkManager.ReadConfig(rcfg.ConfigFile.SinkConfigFile) RouterToSinksChannel := make(chan lp.CCMetric) rcfg.SinkManager.AddInput(RouterToSinksChannel) rcfg.Router.AddOutput(RouterToSinksChannel) } if len(rcfg.ConfigFile.CollectorConfigFile) > 0 { -// rcfg.CollectManagerDone = make(chan bool) - rcfg.CollectManager, err = collectors.New(rcfg.Interval, rcfg.Duration, &rcfg.Sync, rcfg.ConfigFile.CollectorConfigFile) + rcfg.CollectManager, err = collectors.New(rcfg.Ticker, rcfg.Duration, &rcfg.Sync, rcfg.ConfigFile.CollectorConfigFile) if err != nil { log.Print(err.Error()) return } -// rcfg.CollectManager.ReadConfig(rcfg.ConfigFile.CollectorConfigFile) -// if err != nil { -// log.Print(err.Error()) -// return -// } CollectToRouterChannel := make(chan lp.CCMetric) rcfg.CollectManager.AddOutput(CollectToRouterChannel) rcfg.Router.AddInput(CollectToRouterChannel) @@ -306,6 +294,7 @@ func main() { rcfg.Router.Start() rcfg.SinkManager.Start() rcfg.CollectManager.Start() + if use_recv { rcfg.ReceiveManager.Start() }