Method loadStrategiesForever

controller/strategy.go:10–51

持续定时加载报警策略

()

//持续定时加载报警策略
func (c *Controller) loadStrategiesForever() {
	var wg sync.WaitGroup
	for {
		// 获取所有产品线
		products := mydb.GetProducts()
		// 更新产品线告警队列
		c.refreshQueue(products)
		for {
			if len(c.nodePool.Nodes) != 0 {
				break
			}
			lg.Warn("no inspector connected, do not generate task, retry after 1 seconds")
			time.Sleep(time.Second)
		}

		for _, product := range products {
			// 根据产品线 id 获取策略
			strategies := mydb.GetStrategies(product.ID)
			wg.Add(1)
			go func(strategies []*types.Strategy) {
				defer wg.Done()
				for _, strategy := range strategies {
					if strategy.Enable == false {
						lg.Info("strategy %s is not enabled, skipped it.", strategy.Name)
						continue
					}
					// 根据策略 id 获取 trigger
					triggers := mydb.GetTriggersByStrategyID(strategy.ID)
					//如果没有 trigger 则忽略
					if len(triggers) == 0 {
						lg.Warn("strategy %s has no trigger, skipped it.", strategy.Name)
						continue
					}
					// 生成 AlarmTask
					c.processSingleStrategy(strategy, triggers)
				}
			}(strategies)
		}
		wg.Wait()
		time.Sleep(time.Second * time.Duration(GlobalConfig.LOAD_STRATEGIES_INTERVAL))
	}
}

func (c *Controller) processSingleStrategy(strategy *types.Strategy, triggers map[string]*types.Trigger) {
	globalHosts := make([]*types.Host, 0)