com.taotao.boot.monitor.collect.task.XxlJobCollectTask 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
The 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.task;

import com.taotao.boot.common.utils.context.ContextUtils;
import com.taotao.boot.common.utils.log.LogUtils;
import com.taotao.boot.common.utils.reflect.ClassUtils;
import com.taotao.boot.common.utils.reflect.ReflectionUtils;
import com.taotao.boot.monitor.annotation.FieldReport;
import com.taotao.boot.monitor.collect.AbstractCollectTask;
import com.taotao.boot.monitor.collect.CollectInfo;
import com.taotao.boot.monitor.properties.CollectTaskProperties;
import com.xxl.job.core.executor.impl.XxlJobSpringExecutor;
import com.xxl.job.core.handler.annotation.XxlJob;

import java.util.Objects;

/**
 * XxlJobCollectTask
 *
 * @author shuigedeng
 * @version 2021.9
 * @since 2021-09-10 19:18:16
 */
public class XxlJobCollectTask extends AbstractCollectTask {

    private static final String TASK_NAME = "ttc.monitor.collect.xxljob";

    private final CollectTaskProperties properties;
    private final boolean classExist;

    public XxlJobCollectTask(CollectTaskProperties properties) {
        this.properties = properties;
        this.classExist = ClassUtils.isExist("com.xxl.job.core.executor.impl.XxlJobSpringExecutor");
    }

    @Override
    public int getTimeSpan() {
        return properties.getXxljobTimeSpan();
    }

    @Override
    public boolean getEnabled() {
        return properties.isXxljobEnabled() && classExist;
    }

    @Override
    public String getDesc() {
        return this.getClass().getName();
    }

    @Override
    public String getName() {
        return TASK_NAME;
    }

    @Override
    protected CollectInfo getData() {
        try {
            XxlJobSpringExecutor xxlJobSpringExecutor = ContextUtils.getBean(XxlJobSpringExecutor.class, true);
            if (Objects.isNull(xxlJobSpringExecutor)) {
                return null;
            }

            JobInfo data = new JobInfo();
            data.count = ContextUtils.getApplicationContext().getBeanNamesForAnnotation(XxlJob.class).length;

            Object jobThreadRepository =
                    ReflectionUtils.tryGetFieldValue(xxlJobSpringExecutor, "jobThreadRepository", null);
            if (Objects.nonNull(jobThreadRepository)) {
                data.jobThreadRepository = (Integer) ReflectionUtils.callMethod(jobThreadRepository, "size", null);
            }

            Object jobHandlerRepository =
                    ReflectionUtils.tryGetFieldValue(xxlJobSpringExecutor, "jobHandlerRepository", null);
            if (Objects.nonNull(jobThreadRepository)) {
                data.jobHandlerRepository = (Integer) ReflectionUtils.callMethod(jobHandlerRepository, "size", null);
            }

            Object adminBizList = ReflectionUtils.tryGetFieldValue(xxlJobSpringExecutor, "adminBizList", null);
            if (Objects.nonNull(adminBizList)) {
                data.adminBizList = (Integer) ReflectionUtils.callMethod(adminBizList, "size", null);
            }

            data.logRetentionDays = ReflectionUtils.tryGetValue(xxlJobSpringExecutor, "logRetentionDays");

            return data;
        } catch (Exception e) {
            if (LogUtils.isErrorEnabled()) {
                LogUtils.error(e);
            }
        }
        return null;
    }

    private static class JobInfo implements CollectInfo {

        @FieldReport(name = TASK_NAME + ".count", desc = "xxljob任务数量")
        private Integer count = 0;

        @FieldReport(name = TASK_NAME + ".job.thread.repository", desc = "xxljob thread repository数量")
        private Integer jobThreadRepository = 0;

        @FieldReport(name = TASK_NAME + ".job.handler.repository", desc = "xxljob handler repository 数量")
        private Integer jobHandlerRepository = 0;

        @FieldReport(name = TASK_NAME + ".admin.biz.list", desc = "xxljob admin biz 数量")
        private Integer adminBizList = 0;

        @FieldReport(name = TASK_NAME + ".log.retention.days", desc = "xxljob日志保留天数")
        private Integer logRetentionDays = 0;
    }
}