/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.activemq.bugs;

import java.util.HashSet;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import javax.jms.BytesMessage;
import javax.jms.Connection;
import javax.jms.DeliveryMode;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageProducer;
import javax.jms.Session;
import junit.framework.Test;
import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.activemq.ActiveMQPrefetchPolicy;
import org.apache.activemq.TestSupport;
import org.apache.activemq.broker.BrokerService;
import org.apache.activemq.broker.TransportConnector;
import org.apache.activemq.broker.region.DestinationStatistics;
import org.apache.activemq.broker.region.policy.FilePendingQueueMessageStoragePolicy;
import org.apache.activemq.broker.region.policy.PolicyEntry;
import org.apache.activemq.broker.region.policy.PolicyMap;
import org.apache.activemq.command.ActiveMQDestination;
import org.apache.activemq.command.ActiveMQQueue;
import org.apache.activemq.store.kahadb.plist.PListStoreImpl;
import org.apache.activemq.util.DefaultTestAppender;
import org.apache.log4j.Level;
import org.apache.log4j.LogManager;
import org.apache.log4j.spi.LoggingEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class AMQ4221Test extends TestSupport {
    private static final Logger LOG = LoggerFactory.getLogger(AMQ4221Test.class);
    public int PAYLOAD_SIZE_BYTES = 4 * 1024;
    public int NUM_TO_SEND = 60000;
    public int NUM_CONCURRENT_PRODUCERS = 20;
    public int QUEUE_COUNT = 1;
    public int TMP_JOURNAL_MAX_FILE_SIZE = 10 * 1024 * 1024;

    public int DLQ_PURGE_INTERVAL = 30000;

    public int MESSAGE_TIME_TO_LIVE = 20000;
    public int EXPIRE_SWEEP_PERIOD = 200;
    public int TMP_JOURNAL_GC_PERIOD = 50;
    public int RECEIVE_POLL_PERIOD = 4000;
    private int RECEIVE_BATCH = 5000;

    final byte[] payload = new byte[PAYLOAD_SIZE_BYTES];
    final AtomicInteger counter = new AtomicInteger(0);
    final HashSet<Throwable> exceptions = new HashSet<Throwable>();
    BrokerService brokerService;
    private String brokerUrlString;
    ExecutorService executorService = Executors.newCachedThreadPool();
    final AtomicBoolean done = new AtomicBoolean(false);

    public static Test suite() {
        return suite(AMQ4221Test.class);
    }

    @Override
    public void setUp() throws Exception {

        LogManager.getRootLogger().addAppender(new DefaultTestAppender() {

            @Override
            public void doAppend(LoggingEvent event) {
                if (event.getLevel().isGreaterOrEqual(Level.ERROR)) {
                    System.err.println("exit on error: " + event.getMessage());
                    done.set(true);
                    new Thread() {
                        public void run() {
                            System.exit(787);
                        }
                    }.start();
                }
            }
        });

        done.set(false);
        brokerService = new BrokerService();
        brokerService.setDeleteAllMessagesOnStartup(true);
        brokerService.setDestinations(new ActiveMQDestination[]{new ActiveMQQueue("ActiveMQ.DLQ")});

        PolicyEntry defaultPolicy = new PolicyEntry();
        defaultPolicy.setPendingQueuePolicy(new FilePendingQueueMessageStoragePolicy());
        defaultPolicy.setExpireMessagesPeriod(EXPIRE_SWEEP_PERIOD);
        defaultPolicy.setProducerFlowControl(false);
        defaultPolicy.setMemoryLimit(50 * 1024 * 1024);

        brokerService.getSystemUsage().getMemoryUsage().setLimit(50 * 1024 * 1024);


        PolicyMap destinationPolicyMap = new PolicyMap();
        destinationPolicyMap.setDefaultEntry(defaultPolicy);
        brokerService.setDestinationPolicy(destinationPolicyMap);


        PListStoreImpl tempDataStore = new PListStoreImpl();
        tempDataStore.setDirectory(brokerService.getTmpDataDirectory());
        tempDataStore.setJournalMaxFileLength(TMP_JOURNAL_MAX_FILE_SIZE);
        tempDataStore.setCleanupInterval(TMP_JOURNAL_GC_PERIOD);
        tempDataStore.setIndexPageSize(200);
        tempDataStore.setIndexEnablePageCaching(false);

        brokerService.setTempDataStore(tempDataStore);
        brokerService.setAdvisorySupport(false);
        TransportConnector tcp = brokerService.addConnector("tcp://localhost:0");
        brokerService.start();
        brokerUrlString = tcp.getPublishableConnectString();
    }

    @Override
    public void tearDown() throws Exception {
        brokerService.stop();
        brokerService.waitUntilStopped();
        executorService.shutdownNow();
    }

    public void testProduceConsumeExpireHalf() throws Exception {

        final org.apache.activemq.broker.region.Queue dlq =
                (org.apache.activemq.broker.region.Queue) getDestination(brokerService, new ActiveMQQueue("ActiveMQ.DLQ"));

        if (DLQ_PURGE_INTERVAL > 0) {
            executorService.execute(new Runnable() {
                @Override
                public void run() {
                    while (!done.get()) {
                        try {
                            Thread.sleep(DLQ_PURGE_INTERVAL);
                            LOG.info("Purge DLQ, current size: " + dlq.getDestinationStatistics().getMessages().getCount());
                            dlq.purge();
                        } catch (InterruptedException allDone) {
                        } catch (Throwable e) {
                            e.printStackTrace();
                            exceptions.add(e);
                        }
                    }
                }
            });

        }

        final CountDownLatch latch = new CountDownLatch(QUEUE_COUNT);
        for (int i = 0; i < QUEUE_COUNT; i++) {
            final int id = i;
            executorService.execute(new Runnable() {
                @Override
                public void run() {
                    try {
                        doProduceConsumeExpireHalf(id, latch);
                    } catch (Throwable e) {
                        e.printStackTrace();
                        exceptions.add(e);
                    }
                }
            });
        }

        while (!done.get()) {
            done.set(latch.await(5, TimeUnit.SECONDS));
        }
        executorService.shutdown();
        executorService.awaitTermination(5, TimeUnit.MINUTES);

        assertTrue("no exceptions:" + exceptions, exceptions.isEmpty());

    }

    public void doProduceConsumeExpireHalf(int id, CountDownLatch latch) throws Exception {

        final ActiveMQQueue queue = new ActiveMQQueue("Q" + id);

        final ActiveMQConnectionFactory factory = new ActiveMQConnectionFactory(brokerUrlString);
        ActiveMQPrefetchPolicy prefecthPolicy = new ActiveMQPrefetchPolicy();
        prefecthPolicy.setAll(0);
        factory.setPrefetchPolicy(prefecthPolicy);
        Connection connection = factory.createConnection();
        connection.start();
        final MessageConsumer consumer = connection.createSession(false, Session.AUTO_ACKNOWLEDGE).createConsumer(queue, "on = 'true'");

        executorService.execute(new Runnable() {
            @Override
            public void run() {
                try {
                    while (!done.get()) {
                        Thread.sleep(RECEIVE_POLL_PERIOD);
                        for (int i = 0; i < RECEIVE_BATCH && !done.get(); i++) {

                            Message message = consumer.receive(1000);
                            if (message != null) {
                                counter.incrementAndGet();
                                if (counter.get() > 0 && counter.get() % 500 == 0) {
                                    LOG.info("received: " + counter.get() + ", " + message.getJMSDestination().toString());
                                }
                            }
                        }
                    }
                } catch (JMSException ignored) {

                } catch (Exception e) {
                    e.printStackTrace();
                    exceptions.add(e);
                }
            }
        });

        final AtomicInteger accumulator = new AtomicInteger(0);
        final CountDownLatch producersDone = new CountDownLatch(NUM_CONCURRENT_PRODUCERS);

        for (int i = 0; i < NUM_CONCURRENT_PRODUCERS; i++) {
            executorService.execute(new Runnable() {
                @Override
                public void run() {
                    try {
                        Connection sendConnection = factory.createConnection();
                        sendConnection.start();
                        Session sendSession = sendConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
                        MessageProducer producer = sendSession.createProducer(queue);
                        producer.setTimeToLive(MESSAGE_TIME_TO_LIVE);
                        producer.setDeliveryMode(DeliveryMode.NON_PERSISTENT);

                        while (accumulator.incrementAndGet() < NUM_TO_SEND && !done.get()) {
                            BytesMessage message = sendSession.createBytesMessage();
                            message.writeBytes(payload);
                            message.setStringProperty("on", String.valueOf(accumulator.get() % 2 == 0));
                            producer.send(message);

                        }
                        producersDone.countDown();
                    } catch (Exception e) {
                        e.printStackTrace();
                        exceptions.add(e);
                    }
                }
            });
        }

        producersDone.await(10, TimeUnit.MINUTES);

        final DestinationStatistics view = getDestinationStatistics(brokerService, queue);
        LOG.info("total expired so far " + view.getExpired().getCount() + ", " + queue.getQueueName());
        latch.countDown();
    }
}
