com.firefly.utils.log.file.FileLogTask Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of firefly-common Show documentation
There is a newer version: 5.0.2
package com.firefly.utils.log.file;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedTransferQueue;
import java.util.concurrent.TimeUnit;

import com.firefly.utils.VerifyUtils;
import com.firefly.utils.collection.Trie;
import com.firefly.utils.lang.AbstractLifeCycle;
import com.firefly.utils.log.Log;
import com.firefly.utils.log.LogFactory;
import com.firefly.utils.log.LogItem;
import com.firefly.utils.log.LogTask;
import com.firefly.utils.time.Millisecond100Clock;

public class FileLogTask extends AbstractLifeCycle implements LogTask {

	private BlockingQueue queue = new LinkedTransferQueue<>();
	private Thread thread = new Thread(this, "firefly log thread");
	private final Trie logTree;

	public FileLogTask(Trie logTree) {
		thread.setPriority(Thread.MIN_PRIORITY);
		this.logTree = logTree;
	}

	private long flushAllPerSecond(final long lastFlushedTime) {
		// flush all log buffer per one second
		long timeDifference = Millisecond100Clock.currentTimeMillis() - lastFlushedTime;
		if (timeDifference > 1000) {
			LogFactory.getInstance().flushAll();
			return Millisecond100Clock.currentTimeMillis();
		} else {
			return lastFlushedTime;
		}
	}

	@Override
	public void run() {
		long lastFlushedTime = Millisecond100Clock.currentTimeMillis();
		while (true) {
			try {
				for (LogItem logItem = null; (logItem = queue.poll(1000, TimeUnit.MILLISECONDS)) != null;) {
					Log log = LogFactory.getInstance().getLog(logItem.getName());
					if (log instanceof FileLog) {
						((FileLog) log).write(logItem);
					}
					lastFlushedTime = flushAllPerSecond(lastFlushedTime);
				}

				lastFlushedTime = flushAllPerSecond(lastFlushedTime);
			} catch (Throwable e) {
				e.printStackTrace();
				try {
					// avoid CPU exhausting
					Thread.sleep(2000L);
				} catch (InterruptedException e1) {
					e1.printStackTrace();
				}
			}

			if (!start && queue.isEmpty()) {
				for (String key : logTree.keySet()) {
					Log log = logTree.get(key);
					if (log instanceof FileLog) {
						((FileLog) log).close();
					}
				}
				break;
			}
		}
	}

	@Override
	public void add(LogItem logItem) {
		if (!start)
			return;

		if (VerifyUtils.isEmpty(logItem.getName()))
			throw new IllegalArgumentException("log name is empty");

		queue.offer(logItem);
	}

	@Override
	protected void init() {
		start = true;
		thread.start();
	}

	@Override
	protected void destroy() {
		start = false;
	}
}