package net.pterodactylus.rhynodge.engine;
+import static com.google.common.collect.Maps.newTreeMap;
+import static java.lang.String.format;
+import static java.util.Optional.empty;
+import static java.util.Optional.of;
+
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
+import java.util.Optional;
import java.util.SortedMap;
import net.pterodactylus.rhynodge.Filter;
import net.pterodactylus.rhynodge.states.FailedState;
import net.pterodactylus.rhynodge.states.StateManager;
+import com.google.common.util.concurrent.AbstractExecutionThreadService;
import org.apache.commons.lang3.tuple.Pair;
import org.apache.log4j.Logger;
-import com.google.common.collect.Maps;
-import com.google.common.util.concurrent.AbstractExecutionThreadService;
-
/**
* Rhynodge main engine.
*
@Override
public void run() {
while (isRunning()) {
-
- /* delay if we have no reactions. */
- synchronized (reactions) {
- if (reactions.isEmpty()) {
- logger.debug("Sleeping while no Reactions available.");
- try {
- reactions.wait();
- } catch (InterruptedException ie1) {
- /* ignore, we’re looping anyway. */
- }
- continue;
- }
+ Optional<NextReaction> nextReaction = getNextReaction();
+ if (!nextReaction.isPresent()) {
+ continue;
}
- /* find next reaction. */
- SortedMap<Long, Pair<String, Reaction>> nextReactions = Maps.newTreeMap();
- String reactionName;
- Reaction nextReaction;
- synchronized (reactions) {
- for (Entry<String, Reaction> reactionEntry : reactions.entrySet()) {
- net.pterodactylus.rhynodge.State state = stateManager.loadLastState(reactionEntry.getKey());
- long stateTime = (state != null) ? state.time() : 0;
- nextReactions.put(stateTime + reactionEntry.getValue().updateInterval(), Pair.of(reactionEntry.getKey(), reactionEntry.getValue()));
- }
- reactionName = nextReactions.get(nextReactions.firstKey()).getLeft();
- nextReaction = nextReactions.get(nextReactions.firstKey()).getRight();
- }
- logger.debug(String.format("Next Reaction: %s.", reactionName));
+ String reactionName = nextReaction.get().getKey();
+ logger.debug(format("Next Reaction: %s.", reactionName));
/* wait until the next reaction has to run. */
- net.pterodactylus.rhynodge.State lastState = stateManager.loadLastState(reactionName);
- long lastStateTime = (lastState != null) ? lastState.time() : 0;
- int lastStateFailCount = (lastState != null) ? lastState.failCount() : 0;
- long waitTime = (lastStateTime + nextReaction.updateInterval()) - System.currentTimeMillis();
- logger.debug(String.format("Time to wait for next Reaction: %d millseconds.", waitTime));
+ Optional<net.pterodactylus.rhynodge.State> lastState = stateManager.loadLastState(reactionName);
+ int lastStateFailCount = lastState.isPresent() ? lastState.get().failCount() : 0;
+ long waitTime = nextReaction.get().getNextTime() - System.currentTimeMillis();
+ logger.debug(format("Time to wait for next Reaction: %d millseconds.", waitTime));
if (waitTime > 0) {
- synchronized (reactions) {
- try {
- logger.info(String.format("Waiting until %tc.", lastStateTime + nextReaction.updateInterval()));
- reactions.wait(waitTime);
- } catch (InterruptedException ie1) {
- /* we’re looping! */
- }
- }
+ waitForNextReactionToStart(nextReaction, waitTime);
/* re-start loop to check for new reactions. */
continue;
}
- /* run reaction. */
- logger.info(String.format("Running Query for %s...", reactionName));
- Query query = nextReaction.query();
- net.pterodactylus.rhynodge.State state;
- try {
- logger.debug("Querying system...");
- state = query.state();
- if (state == null) {
- state = FailedState.INSTANCE;
- }
- logger.debug("System queried.");
- } catch (Throwable t1) {
- logger.warn("Querying system failed!", t1);
- state = new AbstractState(t1) {
- /* no further state. */
- };
- }
- logger.debug(String.format("State is %s.", state));
+ net.pterodactylus.rhynodge.State state = runReaction(nextReaction, reactionName);
+ logger.debug(format("State is %s.", state));
/* convert states. */
- for (Filter filter : nextReaction.filters()) {
+ for (Filter filter : nextReaction.get().getReaction().filters()) {
if (state.success()) {
- net.pterodactylus.rhynodge.State newState = filter.filter(state);
- logger.debug(String.format("Old state is %s, new state is %s.", state, newState));
- state = newState;
+ state = filter.filter(state);
}
}
if (!state.success()) {
state.setFailCount(lastStateFailCount + 1);
}
- net.pterodactylus.rhynodge.State lastSuccessfulState = stateManager.loadLastSuccessfulState(reactionName);
- stateManager.saveState(reactionName, state);
+ Optional<net.pterodactylus.rhynodge.State> lastSuccessfulState = stateManager.loadLastSuccessfulState(reactionName);
- /* only run trigger if we have collected two successful states. */
- Trigger trigger = nextReaction.trigger();
+ /* merge states. */
boolean triggerHit = false;
- if ((lastSuccessfulState != null) && lastSuccessfulState.success() && state.success()) {
- logger.debug("Checking Trigger for changes...");
- triggerHit = trigger.triggers(state, lastSuccessfulState);
+ Trigger trigger = nextReaction.get().getReaction().trigger();
+ if (lastSuccessfulState.isPresent() && lastSuccessfulState.get().success() && state.success()) {
+ net.pterodactylus.rhynodge.State newState = trigger.mergeStates(lastSuccessfulState.get(), state);
+
+ /* save new state. */
+ stateManager.saveState(reactionName, newState);
+
+ triggerHit = trigger.triggers();
+ } else {
+ /* save first or error state. */
+ stateManager.saveState(reactionName, state);
}
/* run action if trigger was hit. */
- logger.debug(String.format("Trigger was hit: %s.", triggerHit));
+ logger.debug(format("Trigger was hit: %s.", triggerHit));
if (triggerHit) {
logger.info("Executing Action...");
- nextReaction.action().execute(trigger.output(nextReaction));
+ nextReaction.get().getReaction().action().execute(trigger.output(nextReaction.get().getReaction()));
}
}
}
+ private net.pterodactylus.rhynodge.State runReaction(Optional<NextReaction> nextReaction, String reactionName) {
+ logger.info(format("Running Query for %s...", reactionName));
+ Query query = nextReaction.get().getReaction().query();
+ net.pterodactylus.rhynodge.State state;
+ try {
+ logger.debug("Querying system...");
+ state = query.state();
+ if (state == null) {
+ state = FailedState.INSTANCE;
+ }
+ logger.debug("System queried.");
+ } catch (Throwable t1) {
+ logger.warn("Querying system failed!", t1);
+ state = new AbstractState(t1) {
+ };
+ }
+ return state;
+ }
+
+ private void waitForNextReactionToStart(Optional<NextReaction> nextReaction, long waitTime) {
+ synchronized (reactions) {
+ try {
+ logger.info(format("Waiting until %tc.", nextReaction.get().getNextTime()));
+ reactions.wait(waitTime);
+ } catch (InterruptedException ie1) {
+ /* we’re looping! */
+ }
+ }
+ }
+
+ private Optional<NextReaction> getNextReaction() {
+ while (isRunning()) {
+ synchronized (reactions) {
+ if (reactions.isEmpty()) {
+ logger.debug("Sleeping while no Reactions available.");
+ try {
+ reactions.wait();
+ } catch (InterruptedException ie1) {
+ /* ignore, we’re looping anyway. */
+ }
+ continue;
+ }
+ }
+
+ /* find next reaction. */
+ SortedMap<Long, Pair<String, Reaction>> nextReactions = newTreeMap();
+ synchronized (reactions) {
+ for (Entry<String, Reaction> reactionEntry : reactions.entrySet()) {
+ Optional<net.pterodactylus.rhynodge.State> state = stateManager.loadLastState(reactionEntry.getKey());
+ long stateTime = state.isPresent() ? state.get().time() : 0;
+ nextReactions.put(stateTime + reactionEntry.getValue().updateInterval(), Pair.of(reactionEntry.getKey(), reactionEntry.getValue()));
+ }
+ Pair<String, Reaction> keyReaction = nextReactions.get(nextReactions.firstKey());
+ return of(new NextReaction(keyReaction.getKey(), keyReaction.getValue(), nextReactions.firstKey()));
+ }
+ }
+ return empty();
+ }
+
+ private static class NextReaction {
+
+ private final String key;
+ private final Reaction reaction;
+ private final long nextTime;
+
+ private NextReaction(String key, Reaction reaction, long nextTime) {
+ this.key = key;
+ this.reaction = reaction;
+ this.nextTime = nextTime;
+ }
+
+ public String getKey() {
+ return key;
+ }
+
+ public Reaction getReaction() {
+ return reaction;
+ }
+
+ public long getNextTime() {
+ return nextTime;
+ }
+
+ }
+
}
projects / rhynodge.git / blobdiff