com.hmsonline.cassandra.triggers.TriggerTask Maven / Gradle / Ivy

Go to download
package com.hmsonline.cassandra.triggers;

import java.util.List;
import java.util.Map;
import java.util.TimerTask;

import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang.exception.ExceptionUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class TriggerTask extends TimerTask {
    private static Logger logger = LoggerFactory.getLogger(TriggerTask.class);

    @Override
    public void run() {
        try {
            if (ConfigurationStore.getStore().isCommitLogEnabled()) {
                Map> triggerMap = null;
                logger.debug("Running triggers.");
                triggerMap = TriggerStore.getStore().getTriggers();
                List logEntries = DistributedCommitLog.getLog().getPending();
                for (LogEntry logEntry : logEntries) {

                    // Make sure its mine, or its old enough that I should pick
                    // it up to ensure processing by someone
                    if (DistributedCommitLog.getLog().isMine(logEntry) || DistributedCommitLog.getLog().isOld(logEntry)) {

                        // Make sure it hadn't error'd previously
                        if (!LogEntryStatus.ERROR.equals(logEntry.getStatus())) {
                            logger.debug("Processing Entry [" + logEntry.getUuid() + "]:[" + logEntry.getKeyspace()
                                    + "]:[" + logEntry.getColumnFamily() + "]");
                            String path = logEntry.getKeyspace() + ":" + logEntry.getColumnFamily();
                            List triggers = triggerMap.get(path);
                            if (CollectionUtils.isNotEmpty(triggers) && triggers.get(0) instanceof PausedTrigger) {
                                logger.debug("Paused triggers for: " + logEntry.getColumnFamily());
                            } else {
                                if (triggers != null) {
                                    for (Trigger trigger : triggers) {
                                        try {
                                            trigger.process(logEntry);
                                        } catch (Throwable t) {
                                            logEntry.setStatus(LogEntryStatus.ERROR);
                                            logEntry.getErrors().put(
                                                    trigger.getClass().getName(),
                                                    ExceptionUtils.getMessage(t) + " : "
                                                            + ExceptionUtils.getFullStackTrace(t));
                                        }
                                    }
                                }
                                if (LogEntryStatus.ERROR.equals(logEntry.getStatus())) {
                                    DistributedCommitLog.getLog().errorLogEntry(logEntry);
                                } else {
                                    // Provided all processed properly, remove
                                    // the logEntry
                                    DistributedCommitLog.getLog().removeLogEntry(logEntry);
                                }
                            }
                        }
                    }
                }
            } else {
                logger.debug("Skipping trigger execution because commit log is disabled.");
            }
        } catch (Throwable t) {
            logger.error("Could not execute triggers.", t);
        }
    }
}