com.pamirs.pradar.PradarLogDaemon Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of module-pradar-core Show documentation
There is a newer version: 1.0.2
package com.pamirs.pradar;

import com.shulie.instrument.simulator.api.executors.ExecutorServiceFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;

/**
 * Pradar 定时检查，主要有下面的行为：  删除 .deleted 结尾的文件 
检测文件开关 
间隔一段时间输出一次索引 
日志文件被删除，则尝试重新创建之
 * 
定时强制输出文件内容 
 
 */
class PradarLogDaemon implements Runnable {

    private static final String SWITCH_ON = "true";
    private static final String SWITCH_OFF = "false";

    private static AtomicBoolean running = new AtomicBoolean(false);

    private static final CopyOnWriteArrayList watchedAppenders
            = new CopyOnWriteArrayList();

    // 文件开关
    private static final File configFile = new File(Pradar.PRADAR_LOG_DIR + "config.properties");

    private final static Logger LOGGER = LoggerFactory.getLogger(PradarLogDaemon.class);

    private static ScheduledFuture scheduledFuture;

    /**
     * 定期检测日志文件：如果被删除，则尝试重新创建之；强制刷新 appender
     */
    static final PradarAppender watch(PradarAppender appender) {
        watchedAppenders.addIfAbsent(appender);
        return appender;
    }

    static final boolean unwatch(PradarAppender appender) {
        return watchedAppenders.remove(appender);
    }

    @Override
    public void run() {
        // 定时清理
        cleanupFiles();

        // 如果被删除，则尝试重新创建之；强制刷新 appender
        flushAndReload();
    }

    private void cleanupFiles() {
        for (PradarAppender watchedAppender : watchedAppenders) {
            try {
                watchedAppender.cleanup();
            } catch (Throwable e) {
                LOGGER.error("fail to cleanup: {}", watchedAppender, e);
            }
        }
    }

    private void flushAndReload() {
        for (PradarAppender watchedAppender : watchedAppenders) {
            try {
                watchedAppender.reload();
            } catch (Throwable e) {
                LOGGER.error("fail to reload: {}", watchedAppender, e);
            }
        }
    }

    private String getSystemProperty(String propertyName) {
        String value = System.getProperty(propertyName);
        if (value == null) {
            value = System.getenv(propertyName);
        }
        return value;
    }

    static void start() {
        if (PradarSwitcher.isPradarLogDaemonEnabled() && running.compareAndSet(false, true)) {
            int logDaemonInterval = Pradar.PRADAR_LOG_DAEMON_INTERVAL;
            scheduledFuture = ExecutorServiceFactory.getFactory().scheduleAtFixedRate(new PradarLogDaemon(), logDaemonInterval, logDaemonInterval, TimeUnit.SECONDS);
        } else {
            LOGGER.warn("PradarLogDaemon start failed. cause by logDaemonSwitcher: {}, runningStatus: {}", PradarSwitcher.isPradarLogDaemonEnabled(), running.get());
        }
    }

    static void shutdown() {
        if (scheduledFuture != null && !scheduledFuture.isDone() && !scheduledFuture.isCancelled()) {
            try {
                scheduledFuture.cancel(true);
            } catch (Throwable e) {
                LOGGER.error("shutdown PradarLogDaemon failed: ", e);
            }
        }
    }

    static void flushAndWait() {
        for (PradarAppender watchedAppender : watchedAppenders) {
            try {
                if (watchedAppender instanceof AsyncAppender) {
                    ((AsyncAppender) watchedAppender).flushAndWait();
                } else {
                    watchedAppender.flush();
                }
            } catch (Throwable e) {
                LOGGER.error("fail to flush: {}", watchedAppender, e);
            }
        }
    }
}