001/**
002 * Copyright 2005-2017 The Kuali Foundation
003 *
004 * Licensed under the Educational Community License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.opensource.org/licenses/ecl2.php
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package org.kuali.rice.ksb.messaging;
017
018import org.apache.log4j.Logger;
019import org.kuali.rice.core.api.config.property.ConfigContext;
020import org.kuali.rice.ksb.messaging.service.MessageQueueService;
021import org.kuali.rice.ksb.service.KSBServiceLocator;
022import org.kuali.rice.ksb.util.KSBConstants;
023import org.springframework.transaction.TransactionStatus;
024import org.springframework.transaction.support.TransactionCallback;
025
026/**
027 * Fetches messages from the db. Marks as 'R'. Gives messages to ThreadPool for execution
028 *
029 * @author Kuali Rice Team (rice.collab@kuali.org)
030 */
031public class MessageFetcher implements Runnable {
032
033    private static final Logger LOG = Logger.getLogger(MessageFetcher.class);
034
035    private Integer maxMessages;
036    private Long routeQueueId;
037
038    public MessageFetcher(Integer maxMessages) {
039        this.maxMessages = maxMessages;
040    }
041
042    public MessageFetcher(Long routeQueueId) {
043        this.routeQueueId = routeQueueId;
044    }
045
046    public void run() {
047        if (ConfigContext.getCurrentContextConfig().getBooleanProperty(KSBConstants.Config.MESSAGE_PERSISTENCE,
048                false)) {
049            try {
050                requeueDocument();
051                requeueMessages();
052            } catch (Throwable t) {
053                LOG.error("Failed to fetch messages.", t);
054            }
055        }
056    }
057
058    private void requeueMessages() {
059        if (this.routeQueueId == null) {
060            try {
061                for (PersistedMessageBO message : getRouteQueueService().getNextDocuments(maxMessages)) {
062                    message = markEnrouteAndSaveMessage(message);
063                    executeMessage(message);
064                }
065            } catch (Throwable t) {
066                LOG.error("Failed to fetch or process some messages during requeueMessages", t);
067            }
068        }
069    }
070
071    private void requeueDocument() {
072        try {
073            if (this.routeQueueId != null) {
074                PersistedMessageBO message = getRouteQueueService().findByRouteQueueId(this.routeQueueId);
075                message.setQueueStatus(KSBConstants.ROUTE_QUEUE_ROUTING);
076                message = getRouteQueueService().save(message);
077                executeMessage(message);
078            }
079        } catch (Throwable t) {
080            LOG.error("Failed to fetch or process some messages during requeueDocument", t);
081        }
082    }
083
084    private void executeMessage(PersistedMessageBO message) {
085        try {
086            KSBServiceLocator.getThreadPool().execute(new MessageServiceInvoker(message));
087        } catch (Throwable t) {
088            LOG.error("Failed to place message " + message + " in thread pool for execution", t);
089        }
090    }
091
092    private PersistedMessageBO markEnrouteAndSaveMessage(final PersistedMessageBO message) {
093        try {
094            return KSBServiceLocator.getTransactionTemplate().execute(new TransactionCallback<PersistedMessageBO>() {
095                public PersistedMessageBO doInTransaction(TransactionStatus status) {
096                    message.setQueueStatus(KSBConstants.ROUTE_QUEUE_ROUTING);
097                    return getRouteQueueService().save(message);
098                }
099            });
100        } catch (Throwable t) {
101            LOG.error("Caught error attempting to mark message " + message + " as R", t);
102        }
103        return message;
104    }
105
106    private MessageQueueService getRouteQueueService() {
107        return KSBServiceLocator.getMessageQueueService();
108    }
109
110}