Run custom plugins immediately on startup

Author: yguo0905

pkg/custompluginmonitor/plugin/plugin.go

@@ -55,41 +55,53 @@ func (p *Plugin) GetResultChan() <-chan cpmtypes.Result {
 func (p *Plugin) Run() {
 	runTicker := time.NewTicker(*p.config.PluginGlobalConfig.InvokeInterval)
 
-	for {
-		select {
-		case <-runTicker.C:
-			glog.Info("Start to run custom plugins")
+	runner := func() {
+		glog.Info("Start to run custom plugins")
 
-			for _, rule := range p.config.Rules {
-				p.syncChan <- struct{}{}
-				p.Add(1)
+		for _, rule := range p.config.Rules {
+			p.syncChan <- struct{}{}
+			p.Add(1)
 
-				go func(rule *cpmtypes.CustomRule) {
-					defer p.Done()
-					defer func() {
-						<-p.syncChan
-					}()
+			go func(rule *cpmtypes.CustomRule) {
+				defer p.Done()
+				defer func() {
+					<-p.syncChan
+				}()
 
-					start := time.Now()
-					exitStatus, message := p.run(*rule)
-					end := time.Now()
+				start := time.Now()
+				exitStatus, message := p.run(*rule)
+				end := time.Now()
 
-					glog.V(3).Infof("Rule: %+v. Start time: %v. End time: %v. Duration: %v", rule, start, end, end.Sub(start))
+				glog.V(3).Infof("Rule: %+v. Start time: %v. End time: %v. Duration: %v", rule, start, end, end.Sub(start))
 
-					result := cpmtypes.Result{
-						Rule:       rule,
-						ExitStatus: exitStatus,
-						Message:    message,
-					}
+				result := cpmtypes.Result{
+					Rule:       rule,
+					ExitStatus: exitStatus,
+					Message:    message,
+				}
+
+				p.resultChan <- result
+
+				glog.Infof("Add check result %+v for rule %+v", result, rule)
+			}(rule)
+		}
 
-					p.resultChan <- result
+		p.Wait()
+		glog.Info("Finish running custom plugins")
+	}
 
-					glog.Infof("Add check result %+v for rule %+v", result, rule)
-				}(rule)
-			}
+	select {
+	case <-p.tomb.Stopping():
+		glog.Info("Stopping plugin execution")
+		p.tomb.Done()
+	default:
+		runner()
+	}
 
-			p.Wait()
-			glog.Info("Finish running custom plugins")
+	for {
+		select {
+		case <-runTicker.C:
+			runner()
 		case <-p.tomb.Stopping():
 			glog.Info("Stopping plugin execution")
 			p.tomb.Done()