Java tutorial
/** * 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.store.kahadb; import java.io.File; import java.io.IOException; import java.io.RandomAccessFile; import java.util.ArrayList; import junit.framework.Test; import org.apache.activemq.broker.BrokerService; import org.apache.activemq.broker.RecoveryBrokerTest; import org.apache.activemq.broker.StubConnection; import org.apache.activemq.command.ActiveMQDestination; import org.apache.activemq.command.ActiveMQQueue; import org.apache.activemq.command.ConnectionInfo; import org.apache.activemq.command.ConsumerInfo; import org.apache.activemq.command.Message; import org.apache.activemq.command.MessageAck; import org.apache.activemq.command.ProducerInfo; import org.apache.activemq.command.SessionInfo; import org.apache.commons.io.FileUtils; /** * Used to verify that recovery works correctly against * * */ public class KahaDBStoreRecoveryBrokerTest extends RecoveryBrokerTest { public static final String KAHADB_DIR_BASE = "target/activemq-data/kahadb"; public static String kahaDbDirectoryName; enum CorruptionType { None, FailToLoad, LoadInvalid, LoadCorrupt, LoadOrderIndex0 }; public CorruptionType failTest = CorruptionType.None; @Override protected void setUp() throws Exception { kahaDbDirectoryName = KAHADB_DIR_BASE + "/" + System.currentTimeMillis(); super.setUp(); } @Override protected void tearDown() throws Exception { super.tearDown(); try { File kahaDbDir = new File(kahaDbDirectoryName); FileUtils.deleteDirectory(kahaDbDir); } catch (IOException e) { } } @Override protected BrokerService createBroker() throws Exception { BrokerService broker = new BrokerService(); KahaDBStore kaha = new KahaDBStore(); kaha.setDirectory(new File(kahaDbDirectoryName)); kaha.deleteAllMessages(); kaha.setCheckForCorruptJournalFiles(failTest == CorruptionType.LoadOrderIndex0); broker.setPersistenceAdapter(kaha); return broker; } @Override @SuppressWarnings("resource") protected BrokerService createRestartedBroker() throws Exception { // corrupting index File index = new File(kahaDbDirectoryName + "/db.data"); RandomAccessFile raf = new RandomAccessFile(index, "rw"); switch (failTest) { case FailToLoad: index.delete(); raf = new RandomAccessFile(index, "rw"); raf.seek(index.length()); raf.writeBytes("corrupt"); break; case LoadInvalid: // page size 0 raf.seek(0); raf.writeBytes("corrupt and cannot load metadata"); break; case LoadCorrupt: // loadable but invalid metadata // location of order index low priority index for first destination... raf.seek(8 * 1024 + 57); raf.writeLong(Integer.MAX_VALUE - 10); break; case LoadOrderIndex0: // loadable but invalid metadata // location of order index default priority index size // so looks like there are no ids in the order index // picked up by setCheckForCorruptJournalFiles raf.seek(12 * 1024 + 21); raf.writeShort(0); raf.writeChar(0); raf.writeLong(-1); break; default: } raf.close(); // starting broker BrokerService broker = new BrokerService(); KahaDBStore kaha = new KahaDBStore(); kaha.setCheckForCorruptJournalFiles(failTest == CorruptionType.LoadOrderIndex0); // uncomment if you want to test archiving //kaha.setArchiveCorruptedIndex(true); kaha.setDirectory(new File(kahaDbDirectoryName)); broker.setPersistenceAdapter(kaha); return broker; } public static Test suite() { return suite(KahaDBStoreRecoveryBrokerTest.class); } public static void main(String[] args) { junit.textui.TestRunner.run(suite()); } public void initCombosForTestLargeQueuePersistentMessagesNotLostOnRestart() { this.addCombinationValues("failTest", new CorruptionType[] { CorruptionType.FailToLoad, CorruptionType.LoadInvalid, CorruptionType.LoadCorrupt, CorruptionType.LoadOrderIndex0 }); } public void testLargeQueuePersistentMessagesNotLostOnRestart() throws Exception { ActiveMQDestination destination = new ActiveMQQueue("TEST"); // Setup the producer and send the message. StubConnection connection = createConnection(); ConnectionInfo connectionInfo = createConnectionInfo(); SessionInfo sessionInfo = createSessionInfo(connectionInfo); ProducerInfo producerInfo = createProducerInfo(sessionInfo); connection.send(connectionInfo); connection.send(sessionInfo); connection.send(producerInfo); ArrayList<String> expected = new ArrayList<String>(); int MESSAGE_COUNT = 10000; for (int i = 0; i < MESSAGE_COUNT; i++) { Message message = createMessage(producerInfo, destination); message.setPersistent(true); connection.send(message); expected.add(message.getMessageId().toString()); } connection.request(closeConnectionInfo(connectionInfo)); // restart the broker. restartBroker(); // Setup the consumer and receive the message. connection = createConnection(); connectionInfo = createConnectionInfo(); sessionInfo = createSessionInfo(connectionInfo); connection.send(connectionInfo); connection.send(sessionInfo); ConsumerInfo consumerInfo = createConsumerInfo(sessionInfo, destination); connection.send(consumerInfo); producerInfo = createProducerInfo(sessionInfo); connection.send(producerInfo); for (int i = 0; i < MESSAGE_COUNT / 2; i++) { Message m = receiveMessage(connection); assertNotNull("Should have received message " + expected.get(0) + " by now!", m); assertEquals(expected.remove(0), m.getMessageId().toString()); MessageAck ack = createAck(consumerInfo, m, 1, MessageAck.STANDARD_ACK_TYPE); connection.send(ack); } connection.request(closeConnectionInfo(connectionInfo)); // restart the broker. restartBroker(); // Setup the consumer and receive the message. connection = createConnection(); connectionInfo = createConnectionInfo(); sessionInfo = createSessionInfo(connectionInfo); connection.send(connectionInfo); connection.send(sessionInfo); consumerInfo = createConsumerInfo(sessionInfo, destination); connection.send(consumerInfo); for (int i = 0; i < MESSAGE_COUNT / 2; i++) { Message m = receiveMessage(connection); assertNotNull("Should have received message " + expected.get(i) + " by now!", m); assertEquals(expected.get(i), m.getMessageId().toString()); MessageAck ack = createAck(consumerInfo, m, 1, MessageAck.STANDARD_ACK_TYPE); connection.send(ack); } connection.request(closeConnectionInfo(connectionInfo)); } }