com.taotao.boot.monitor.collect.HealthCheckProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of taotao-boot-starter-monitor Show documentation
taotao-boot-starter-monitor
There is a newer version: 2025.01
Show newest version
/*
 * Copyright (c) 2020-2030, Shuigedeng ([email protected] & https://blog.taotaocloud.top/).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.taotao.boot.monitor.collect;

import com.taotao.boot.common.constant.StarterName;
import com.taotao.boot.common.utils.lang.StringUtils;
import com.taotao.boot.common.utils.log.LogUtils;
import com.taotao.boot.monitor.Monitor;
import com.taotao.boot.monitor.collect.task.AsyncThreadPoolCollectTask;
import com.taotao.boot.monitor.collect.task.CpuCollectTask;
import com.taotao.boot.monitor.collect.task.DataSourceCollectTask;
import com.taotao.boot.monitor.collect.task.DoubtApiCollectTask;
import com.taotao.boot.monitor.collect.task.ElkCollectTask;
import com.taotao.boot.monitor.collect.task.HttpPoolCollectTask;
import com.taotao.boot.monitor.collect.task.LogStatisticCollectTask;
import com.taotao.boot.monitor.collect.task.MemoryCollectTask;
import com.taotao.boot.monitor.collect.task.MonitorThreadPoolCollectTask;
import com.taotao.boot.monitor.collect.task.MybatisCollectTask;
import com.taotao.boot.monitor.collect.task.NacosCollectTask;
import com.taotao.boot.monitor.collect.task.NetworkCollectTask;
import com.taotao.boot.monitor.collect.task.ThreadCollectTask;
import com.taotao.boot.monitor.collect.task.UnCatchExceptionCollectTask;
import com.taotao.boot.monitor.collect.task.WebServerCollectTask;
import com.taotao.boot.monitor.collect.task.XxlJobCollectTask;
import com.taotao.boot.monitor.enums.WarnTypeEnum;
import com.taotao.boot.monitor.model.Report;
import com.taotao.boot.monitor.properties.CollectTaskProperties;
import com.taotao.boot.monitor.properties.MonitorProperties;
import com.taotao.boot.monitor.strategy.DefaultWarnStrategy;
import com.taotao.boot.monitor.strategy.WarnStrategy;

import java.util.ArrayList;
import java.util.List;

/**
 * HealthCheckProvider
 *
 * @author shuigedeng
 * @version 2021.9
 * @since 2021-09-10 17:42:25
 */
public class HealthCheckProvider implements AutoCloseable {

    private List checkTasks = new ArrayList<>();
    private boolean close;

    private Monitor monitor;
    private WarnStrategy strategy;
    private MonitorProperties monitorProperties;
    private CollectTaskProperties collectTaskProperties;

    public void registerCollectTask(AbstractCollectTask task) {
        checkTasks.add(task);
    }

    public HealthCheckProvider(
            CollectTaskProperties collectTaskProperties,
            MonitorProperties monitorProperties,
            WarnStrategy strategy,
            Monitor monitor) {
        this.strategy = strategy;
        this.close = false;
        this.collectTaskProperties = collectTaskProperties;
        this.monitorProperties = monitorProperties;
        this.monitor = monitor;

        registerCollectTask(new CpuCollectTask(collectTaskProperties));
        // registerCollectTask(new IOCollectTask(collectTaskProperties));
        registerCollectTask(new MemoryCollectTask(collectTaskProperties));
        registerCollectTask(new ThreadCollectTask(collectTaskProperties));
        registerCollectTask(new UnCatchExceptionCollectTask(collectTaskProperties));
        registerCollectTask(new MonitorThreadPoolCollectTask(collectTaskProperties, monitor.getMonitorThreadPoolExecutor()));
        registerCollectTask(new AsyncThreadPoolCollectTask(collectTaskProperties, monitor.getAsyncThreadPoolExecutor()));
        registerCollectTask(new MybatisCollectTask(collectTaskProperties));
        registerCollectTask(new DataSourceCollectTask(collectTaskProperties));
        registerCollectTask(new WebServerCollectTask(collectTaskProperties));
        // registerCollectTask(new JedisCollectTask(collectTaskProperties));
        registerCollectTask(new NetworkCollectTask(collectTaskProperties));
        registerCollectTask(new XxlJobCollectTask(collectTaskProperties));
        // registerCollectTask(new FileCollectTask());
        // registerCollectTask(new RocketMQCollectTask());
        registerCollectTask(new HttpPoolCollectTask(collectTaskProperties));
        // registerCollectTask(new CatCollectTask());
        // registerCollectTask(new ElasticSearchCollectTask());
        registerCollectTask(new ElkCollectTask(collectTaskProperties));
        registerCollectTask(new DoubtApiCollectTask(collectTaskProperties));
        registerCollectTask(new LogStatisticCollectTask(collectTaskProperties));
        registerCollectTask(new NacosCollectTask(collectTaskProperties));

        monitor.monitorSubmit("系统任务: MonitorCheckProvider 采集任务", () -> {
            while (!monitor.monitorIsShutdown() && !close) {
                try {
                    Report report = getReport(false);
                    String text = strategy.analyseText(report);
                    if (StringUtils.isEmpty(text)) {
                        return;
                    }

                    AbstractCollectTask.notifyMessage(WarnTypeEnum.ERROR, "健康检查", text);
                } catch (Exception e) {
                    LogUtils.warn(StarterName.MONITOR_STARTER, "run 循环采集出错", e);
                }

                try {
                    Thread.sleep(monitorProperties.getTimeSpan() * 1000L);
                } catch (Exception e) {
                    LogUtils.error(e);
                }
            }
        });
    }

    /**
     * getReport
     *
     * @param isAnalyse isAnalyse
     * @return {@link Report }
     * @since 2021-09-10 17:42:55
     */
    public Report getReport(boolean isAnalyse) {
        Report report = new Report().setDesc("健康检查报表").setName("ttc.monitor.report");

        for (AbstractCollectTask task : checkTasks) {
            if (task.getEnabled()) {
                try {
                    Report report2 = task.getReport();
                    if (report2 != null) {
                        report.put(
                                task.getName(), report2.setDesc(task.getDesc()).setName(task.getName()));
                    }
                } catch (Exception e) {
                    LogUtils.error(e, StarterName.MONITOR_STARTER + task.getName() + "采集获取报表出错");
                }
            }
        }

        if (isAnalyse) {
            report = strategy.analyse(report);
        }

        return report;
    }

    @Override
    public void close() {
        close = true;

        monitor.monitorShutdown();

        for (AbstractCollectTask task : checkTasks) {
            try {
                task.close();
            } catch (Exception exp) {
                LogUtils.warn(StarterName.MONITOR_STARTER, "close资源释放出错", exp);
            }
        }
    }

    public List getCheckTasks() {
        return checkTasks;
    }

    public void setCheckTasks(List checkTasks) {
        this.checkTasks = checkTasks;
    }

    public WarnStrategy getStrategy() {
        return strategy;
    }

    public void setStrategy(DefaultWarnStrategy strategy) {
        this.strategy = strategy;
    }

    public boolean isClose() {
        return close;
    }

    public void setClose(boolean close) {
        this.close = close;
    }

    public MonitorProperties getHealthProperties() {
        return monitorProperties;
    }

    public void setHealthProperties(MonitorProperties monitorProperties) {
        this.monitorProperties = monitorProperties;
    }

    public Monitor getMonitor() {
        return monitor;
    }

    public void setMonitor(Monitor monitor) {
        this.monitor = monitor;
    }

    public void setStrategy(WarnStrategy strategy) {
        this.strategy = strategy;
    }

    public CollectTaskProperties getCollectTaskProperties() {
        return collectTaskProperties;
    }

    public void setCollectTaskProperties(CollectTaskProperties collectTaskProperties) {
        this.collectTaskProperties = collectTaskProperties;
    }
}