优化健康检查代码

internal/tasks/health_check_task.go

@@ -72,12 +72,18 @@ func (this *HealthCheckTask) Loop() error {
 	for _, cluster := range clusters {
 		var clusterId = int64(cluster.Id)
 
+		if !cluster.IsOn {
+			this.stopClusterTask(clusterId)
+			continue
+		}
+
 		// 检查当前集群上是否有服务，如果尚没有部署服务，则直接跳过
 		countServers, err := models.SharedServerDAO.CountAllEnabledServersWithNodeClusterId(nil, clusterId)
 		if err != nil {
 			return err
 		}
 		if countServers == 0 {
+			this.stopClusterTask(clusterId)
 			continue
 		}
 
@@ -86,8 +92,16 @@ func (this *HealthCheckTask) Loop() error {
 			err = json.Unmarshal(cluster.HealthCheck, config)
 			if err != nil {
 				this.logErr("HealthCheckTask", err.Error())
+				this.stopClusterTask(clusterId)
 				continue
 			}
+			if !config.IsOn {
+				this.stopClusterTask(clusterId)
+				continue
+			}
+		} else {
+			this.stopClusterTask(clusterId)
+			continue
 		}
 
 		task, ok := this.tasksMap[clusterId]
@@ -112,3 +126,11 @@ func (this *HealthCheckTask) Loop() error {
 
 	return nil
 }
+
+func (this *HealthCheckTask) stopClusterTask(clusterId int64) {
+	var task = this.tasksMap[clusterId]
+	if task != nil {
+		task.Stop()
+		delete(this.tasksMap, clusterId)
+	}
+}