001    /**
002     * Copyright (c) 2000-2012 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.increment;
016    
017    import com.liferay.portal.kernel.concurrent.BatchablePipe;
018    import com.liferay.portal.kernel.concurrent.RejectedExecutionHandler;
019    import com.liferay.portal.kernel.concurrent.ThreadPoolExecutor;
020    import com.liferay.portal.kernel.log.Log;
021    import com.liferay.portal.kernel.log.LogFactoryUtil;
022    import com.liferay.portal.kernel.messaging.Message;
023    import com.liferay.portal.kernel.messaging.MessageRunnable;
024    
025    /**
026     * @author Shuyang Zhou
027     */
028    public class BufferedIncrementDiscardPolicy
029            implements RejectedExecutionHandler {
030    
031            @SuppressWarnings("rawtypes")
032            public void rejectedExecution(
033                    Runnable runnable, ThreadPoolExecutor threadPoolExecutor) {
034    
035                    MessageRunnable messageRunnable = (MessageRunnable)runnable;
036    
037                    Message message = messageRunnable.getMessage();
038    
039                    BatchablePipe<String, BufferedIncreasableEntry> batchablePipe =
040                            (BatchablePipe<String, BufferedIncreasableEntry>)
041                                    message.getPayload();
042    
043                    for (int i = 0; i < _discardNumber; i++) {
044                            BufferedIncreasableEntry bufferedIncreasableEntry =
045                                    (BufferedIncreasableEntry)batchablePipe.take();
046    
047                            if (bufferedIncreasableEntry == null) {
048                                    break;
049                            }
050                            else if (_log.isInfoEnabled()) {
051                                    _log.info(
052                                            "Discarding BufferedIncreasableEntry " +
053                                                    bufferedIncreasableEntry);
054                            }
055                    }
056            }
057    
058            public void setDiscardNumber(int discardNumber) {
059                    _discardNumber = discardNumber;
060            }
061    
062            private static Log _log = LogFactoryUtil.getLog(
063                    BufferedIncrementDiscardPolicy.class);
064    
065            private int _discardNumber = 1;
066    
067    }