| 1 | |
|
| 2 | |
|
| 3 | |
|
| 4 | |
|
| 5 | |
|
| 6 | |
|
| 7 | |
|
| 8 | |
|
| 9 | |
|
| 10 | |
|
| 11 | |
|
| 12 | |
|
| 13 | |
|
| 14 | |
package org.kuali.rice.ksb.messaging.exceptionhandling; |
| 15 | |
|
| 16 | |
import java.sql.Timestamp; |
| 17 | |
|
| 18 | |
import org.apache.commons.lang.StringUtils; |
| 19 | |
import org.apache.log4j.Logger; |
| 20 | |
import org.kuali.rice.core.config.ConfigContext; |
| 21 | |
import org.kuali.rice.ksb.messaging.PersistedMessage; |
| 22 | |
import org.kuali.rice.ksb.messaging.ServiceInfo; |
| 23 | |
import org.kuali.rice.ksb.service.KSBServiceLocator; |
| 24 | |
import org.kuali.rice.ksb.util.KSBConstants; |
| 25 | |
|
| 26 | |
|
| 27 | |
|
| 28 | |
|
| 29 | |
|
| 30 | |
|
| 31 | |
|
| 32 | 0 | public class DefaultMessageExceptionHandler implements MessageExceptionHandler { |
| 33 | |
|
| 34 | 0 | private static final Logger LOG = Logger.getLogger(DefaultMessageExceptionHandler.class); |
| 35 | |
|
| 36 | |
private static final long DEFAULT_TIME_INCREMENT = 60 * 60 * 1000; |
| 37 | |
|
| 38 | |
private static final int DEFAULT_MAX_RETRIES = 7; |
| 39 | |
|
| 40 | |
public void handleException(Throwable throwable, PersistedMessage message, Object service) throws Exception { |
| 41 | 0 | if (isInException(message)) { |
| 42 | 0 | placeInException(throwable, message); |
| 43 | |
} else { |
| 44 | 0 | requeue(throwable, message); |
| 45 | |
} |
| 46 | 0 | } |
| 47 | |
|
| 48 | |
public void handleExceptionLastDitchEffort(Throwable throwable, PersistedMessage message, Object service) throws Exception { |
| 49 | 0 | LOG.error("Complete failure when attempting to put message into exception routing! Message was: " + message, throwable); |
| 50 | 0 | } |
| 51 | |
|
| 52 | |
public boolean isInException(PersistedMessage message) { |
| 53 | 0 | ServiceInfo serviceInfo = message.getMethodCall().getServiceInfo(); |
| 54 | |
|
| 55 | 0 | if (getImmediateExceptionRouting()) { |
| 56 | 0 | return true; |
| 57 | |
} |
| 58 | |
|
| 59 | 0 | Integer globalMaxRetryAttempts = getGlobalMaxRetryAttempts(); |
| 60 | 0 | if (globalMaxRetryAttempts != null) { |
| 61 | 0 | LOG.info("Global Max Retry has been set, so is overriding other max retry attempts."); |
| 62 | 0 | LOG.info("Global Max Retry count = " + globalMaxRetryAttempts + "."); |
| 63 | 0 | return (message.getRetryCount().intValue() >= globalMaxRetryAttempts.intValue()); |
| 64 | |
} |
| 65 | |
|
| 66 | 0 | if (serviceInfo.getServiceDefinition().getRetryAttempts() > 0) { |
| 67 | 0 | LOG.info("Message set for retry exception handling. Message retry count = " + message.getRetryCount()); |
| 68 | 0 | if (message.getRetryCount() >= serviceInfo.getServiceDefinition().getRetryAttempts()) { |
| 69 | 0 | return true; |
| 70 | |
} |
| 71 | 0 | } else if (serviceInfo.getServiceDefinition().getMillisToLive() > 0) { |
| 72 | 0 | LOG.info("Message set for time to live exception handling. Message expiration date = " + message.getExpirationDate().getTime()); |
| 73 | 0 | if (System.currentTimeMillis() > message.getExpirationDate().getTime()) { |
| 74 | 0 | return true; |
| 75 | |
} |
| 76 | 0 | } else if (message.getRetryCount() >= this.getMaxRetryAttempts()) { |
| 77 | 0 | LOG.info("Message set for default exception handling. Comparing retry count = " + message.getRetryCount() + " against default max count."); |
| 78 | 0 | return true; |
| 79 | |
} |
| 80 | 0 | return false; |
| 81 | |
} |
| 82 | |
|
| 83 | |
protected void requeue(Throwable throwable, PersistedMessage message) throws Exception { |
| 84 | 0 | Integer retryCount = message.getRetryCount(); |
| 85 | 0 | message.setQueueStatus(KSBConstants.ROUTE_QUEUE_QUEUED); |
| 86 | 0 | long addMilliseconds = Math.round(getTimeIncrement() * Math.pow(2, retryCount)); |
| 87 | 0 | Timestamp currentTime = message.getQueueDate(); |
| 88 | 0 | Timestamp newTime = new Timestamp(currentTime.getTime() + addMilliseconds); |
| 89 | 0 | message.setQueueStatus(KSBConstants.ROUTE_QUEUE_QUEUED); |
| 90 | 0 | message.setRetryCount(new Integer(retryCount + 1)); |
| 91 | 0 | message.setQueueDate(newTime); |
| 92 | 0 | scheduleExecution(throwable, message); |
| 93 | 0 | } |
| 94 | |
|
| 95 | |
protected void placeInException(Throwable throwable, PersistedMessage message) throws Exception { |
| 96 | 0 | message.setQueueStatus(KSBConstants.ROUTE_QUEUE_EXCEPTION); |
| 97 | 0 | message.setQueueDate(new Timestamp(System.currentTimeMillis())); |
| 98 | 0 | KSBServiceLocator.getRouteQueueService().save(message); |
| 99 | 0 | } |
| 100 | |
|
| 101 | |
protected void scheduleExecution(Throwable throwable, PersistedMessage message) throws Exception { |
| 102 | 0 | KSBServiceLocator.getExceptionRoutingService().scheduleExecution(throwable, message, null); |
| 103 | 0 | } |
| 104 | |
|
| 105 | |
public Integer getMaxRetryAttempts() { |
| 106 | |
try { |
| 107 | 0 | return new Integer(ConfigContext.getCurrentContextConfig().getProperty(KSBConstants.ROUTE_QUEUE_MAX_RETRY_ATTEMPTS_KEY)); |
| 108 | 0 | } catch (NumberFormatException e) { |
| 109 | 0 | LOG.error("Constant '" + KSBConstants.ROUTE_QUEUE_MAX_RETRY_ATTEMPTS_KEY + "' is not a number and is being " + "used as a default for exception messages. " + DEFAULT_MAX_RETRIES + " will be used as a retry limit until this number is fixed"); |
| 110 | 0 | return DEFAULT_MAX_RETRIES; |
| 111 | |
} |
| 112 | |
} |
| 113 | |
|
| 114 | |
public Integer getGlobalMaxRetryAttempts() { |
| 115 | 0 | String globalMax = ConfigContext.getCurrentContextConfig().getProperty(KSBConstants.ROUTE_QUEUE_MAX_RETRY_ATTEMPTS_OVERRIDE_KEY); |
| 116 | 0 | if (StringUtils.isBlank(globalMax)) { |
| 117 | 0 | return null; |
| 118 | |
} |
| 119 | |
try { |
| 120 | 0 | Integer globalMaxRetries = new Integer(globalMax); |
| 121 | 0 | if (globalMaxRetries >= 0) { |
| 122 | 0 | return globalMaxRetries; |
| 123 | |
} |
| 124 | 0 | } catch (NumberFormatException e) { |
| 125 | 0 | LOG.error("Constant '" + KSBConstants.ROUTE_QUEUE_MAX_RETRY_ATTEMPTS_OVERRIDE_KEY + "' is not a number and is being " + "used as a default for exception messages. " + DEFAULT_MAX_RETRIES + " will be used as a retry limit until this number is fixed"); |
| 126 | 0 | } |
| 127 | 0 | return null; |
| 128 | |
} |
| 129 | |
|
| 130 | |
public Long getTimeIncrement() { |
| 131 | |
try { |
| 132 | 0 | return new Long(ConfigContext.getCurrentContextConfig().getProperty(KSBConstants.ROUTE_QUEUE_TIME_INCREMENT_KEY)); |
| 133 | 0 | } catch (NumberFormatException e) { |
| 134 | 0 | LOG.error("Constant '" + KSBConstants.ROUTE_QUEUE_TIME_INCREMENT_KEY + "' is not a number and will not be used " + "as the default time increment for exception routing. Default of " + DEFAULT_TIME_INCREMENT + " will be used."); |
| 135 | 0 | return DEFAULT_TIME_INCREMENT; |
| 136 | |
} |
| 137 | |
} |
| 138 | |
|
| 139 | |
public Boolean getImmediateExceptionRouting() { |
| 140 | 0 | return new Boolean(ConfigContext.getCurrentContextConfig().getProperty(KSBConstants.IMMEDIATE_EXCEPTION_ROUTING)); |
| 141 | |
} |
| 142 | |
} |