prerna.reactor.insights.GlobalInsightCountUpdater Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation

SEMOSS

The newest version!

package prerna.reactor.insights;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import prerna.auth.utils.SecurityInsightUtils;
import prerna.util.Constants;

public class GlobalInsightCountUpdater {
	
	private static final Logger classLogger = LogManager.getLogger(GlobalInsightCountUpdater.class);

	/*
	 * Creating a class to manage updating the insight count
	 * This is necessary since we will get version conflicts
	 * if you run 2 insights at the same time
	 */

	private static GlobalInsightCountUpdater singleton;

	private BlockingQueue queue;
	private CountUpdater updater;

	private GlobalInsightCountUpdater() {
		queue = new ArrayBlockingQueue(2_000);
		updater = new CountUpdater(queue);

		new Thread(updater).start();
	}

	public static GlobalInsightCountUpdater getInstance() {
		if(singleton == null) {
			singleton = new GlobalInsightCountUpdater();
		}
		return singleton;
	}

	public void addToQueue(String engineId, String id) {
		try {
			queue.add(new String[]{engineId, id});
		} catch(Exception e) {
			classLogger.error(Constants.STACKTRACE, e);
		}
	}

}

class CountUpdater implements Runnable {
	
	private static final Logger classLogger = LogManager.getLogger(CountUpdater.class);

	protected BlockingQueue queue = null;

	public CountUpdater(BlockingQueue queue) {
		this.queue = queue;
	}

	public void run() {
		try {
			String[] update = null;
			while( (update = queue.take()) != null) {
				SecurityInsightUtils.updateExecutionCount(update[0], update[1]);
			}
		} catch (InterruptedException e) {
			classLogger.error(Constants.STACKTRACE, e);
		}
	}
}