fix some problems in Quick Journal - now message containers are using Kaha maps instead of lists

git-svn-id: https://svn.apache.org/repos/asf/activemq/trunk@504501 13f79535-47bb-0310-9956-ffa450edef68

activemq-core/src/main/java/org/apache/activemq/broker/region/cursors/FilePendingMessageCursor.java

@@ -22,12 +22,12 @@ import java.util.concurrent.atomic.AtomicLong;
 import org.apache.activemq.broker.region.Destination;
 import org.apache.activemq.broker.region.MessageReference;
 import org.apache.activemq.command.Message;
+import org.apache.activemq.kaha.CommandMarshaller;
 import org.apache.activemq.kaha.ListContainer;
 import org.apache.activemq.kaha.Store;
 import org.apache.activemq.memory.UsageListener;
 import org.apache.activemq.memory.UsageManager;
 import org.apache.activemq.openwire.OpenWireFormat;
-import org.apache.activemq.store.kahadaptor.CommandMarshaller;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 

activemq-core/src/main/java/org/apache/activemq/kaha/CommandMarshaller.java

@@ -15,7 +15,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package org.apache.activemq.store.kahadaptor;
+package org.apache.activemq.kaha;
 
 import java.io.DataInput;
 import java.io.DataOutput;

activemq-core/src/main/java/org/apache/activemq/store/kahadaptor/KahaPersistenceAdapter.java

@@ -20,21 +20,20 @@ import java.util.HashSet;
 import java.util.Iterator;
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
-
 import org.apache.activemq.broker.ConnectionContext;
 import org.apache.activemq.command.ActiveMQDestination;
 import org.apache.activemq.command.ActiveMQQueue;
 import org.apache.activemq.command.ActiveMQTopic;
-import org.apache.activemq.command.Command;
 import org.apache.activemq.command.Message;
 import org.apache.activemq.command.MessageId;
+import org.apache.activemq.kaha.CommandMarshaller;
 import org.apache.activemq.kaha.ListContainer;
 import org.apache.activemq.kaha.MapContainer;
 import org.apache.activemq.kaha.Marshaller;
+import org.apache.activemq.kaha.MessageIdMarshaller;
 import org.apache.activemq.kaha.MessageMarshaller;
 import org.apache.activemq.kaha.Store;
 import org.apache.activemq.kaha.StoreFactory;
-import org.apache.activemq.kaha.StringMarshaller;
 import org.apache.activemq.memory.UsageManager;
 import org.apache.activemq.openwire.OpenWireFormat;
 import org.apache.activemq.store.MessageStore;
@@ -78,7 +77,7 @@ public class KahaPersistenceAdapter implements PersistenceAdapter{
         Set<ActiveMQDestination> rc=new HashSet<ActiveMQDestination>();
         try{
             Store store=getStore();
-            for(Iterator i=store.getListContainerIds().iterator();i.hasNext();){
+            for(Iterator i=store.getMapContainerIds().iterator();i.hasNext();){
                 Object obj=i.next();
                 if(obj instanceof ActiveMQDestination){
                     rc.add((ActiveMQDestination) obj);

activemq-core/src/main/java/org/apache/activemq/store/kahadaptor/KahaReferenceStore.java

@@ -110,7 +110,7 @@ public class KahaReferenceStore implements ReferenceStore{
 
     public void addReferenceFileIdsInUse(Set<Integer> rc){
         for(StoreEntry entry=messageContainer.getFirst();entry!=null;entry=messageContainer.getNext(entry)){
-            ReferenceRecord msg=(ReferenceRecord)messageContainer.get(entry);
+            ReferenceRecord msg=(ReferenceRecord)messageContainer.getValue(entry);
             rc.add(msg.data.getFileId());
         }
     }

activemq-core/src/main/java/org/apache/activemq/store/kahadaptor/KahaReferenceStoreAdapter.java

@@ -32,6 +32,7 @@ import org.apache.activemq.command.MessageId;
 import org.apache.activemq.kaha.ListContainer;
 import org.apache.activemq.kaha.MapContainer;
 import org.apache.activemq.kaha.Marshaller;
+import org.apache.activemq.kaha.MessageIdMarshaller;
 import org.apache.activemq.kaha.MessageMarshaller;
 import org.apache.activemq.kaha.Store;
 import org.apache.activemq.store.MessageStore;

activemq-core/src/main/java/org/apache/activemq/store/kahadaptor/KahaTopicReferenceStore.java

@@ -100,7 +100,7 @@ public class KahaTopicReferenceStore extends KahaReferenceStore implements Topic
         for(StoreEntry entry=ackContainer.getFirst();entry!=null;entry=ackContainer.getNext(entry)){
             TopicSubAck subAck=(TopicSubAck)ackContainer.get(entry);
             if(subAck.getCount()>0){
-                ReferenceRecord rr=(ReferenceRecord)messageContainer.get(subAck.getMessageEntry());
+                ReferenceRecord rr=(ReferenceRecord)messageContainer.getValue(subAck.getMessageEntry());
                 rc.add(rr.data.getFileId());
             }
         }

activemq-core/src/main/java/org/apache/activemq/store/kahadaptor/MessageIdMarshaller.java

@@ -1,43 +0,0 @@
-/**
- *
- * 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.kahadaptor;
-
-import java.io.DataInput;
-import java.io.DataOutput;
-import java.io.IOException;
-import org.apache.activemq.command.MessageId;
-import org.apache.activemq.kaha.Marshaller;
-
-
-/**
- * Marshall an AtomicInteger
- * @version $Revision: 1.10 $
- */
-public class MessageIdMarshaller implements Marshaller<MessageId>{
-   
-
-    public void writePayload(MessageId mid,DataOutput dataOut) throws IOException{
-       dataOut.writeUTF(mid.toString());
-       
-    }
-
-    public MessageId readPayload(DataInput dataIn) throws IOException{
-        String str = dataIn.readUTF();
-        return new MessageId(str);
-    }
-}

activemq-core/src/main/java/org/apache/activemq/store/rapid/RapidPersistenceAdapter.java

@@ -48,6 +48,7 @@ import org.apache.activemq.command.JournalTrace;
 import org.apache.activemq.command.JournalTransaction;
 import org.apache.activemq.command.Message;
 import org.apache.activemq.command.MessageAck;
+import org.apache.activemq.kaha.CommandMarshaller;
 import org.apache.activemq.kaha.ListContainer;
 import org.apache.activemq.kaha.MapContainer;
 import org.apache.activemq.kaha.Store;
@@ -60,9 +61,6 @@ import org.apache.activemq.store.MessageStore;
 import org.apache.activemq.store.PersistenceAdapter;
 import org.apache.activemq.store.TopicMessageStore;
 import org.apache.activemq.store.TransactionStore;
-import org.apache.activemq.store.kahadaptor.AtomicIntegerMarshaller;
-import org.apache.activemq.store.kahadaptor.CommandMarshaller;
-import org.apache.activemq.store.kahadaptor.KahaTopicMessageStore;
 import org.apache.activemq.store.kahadaptor.TopicSubAckMarshaller;
 import org.apache.activemq.store.rapid.RapidTransactionStore.Tx;
 import org.apache.activemq.store.rapid.RapidTransactionStore.TxOperation;

activemq-core/src/test/java/org/apache/activemq/perf/SimpleTopicTest.java

@@ -1,20 +1,17 @@
 /**
  * 
- * Licensed to the Apache Software Foundation (ASF) under one or more
+ * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE
- * contributor license agreements.  See the NOTICE file distributed with
+ * file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file
- * this work for additional information regarding copyright ownership.
+ * to You under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the
- * The ASF licenses this file to You under the Apache License, Version 2.0
+ * License. You may obtain a copy of the License at
- * (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
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
- * distributed under the License is distributed on an "AS IS" BASIS,
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * specific language governing permissions and limitations under the License.
- * See the License for the specific language governing permissions and
- * limitations under the License.
  */
+
 package org.apache.activemq.perf;
 
 import javax.jms.Connection;
@@ -22,23 +19,24 @@ import javax.jms.ConnectionFactory;
 import javax.jms.Destination;
 import javax.jms.JMSException;
 import javax.jms.Session;
-
 import junit.framework.TestCase;
-
 import org.apache.activemq.ActiveMQConnectionFactory;
 import org.apache.activemq.broker.BrokerService;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
+
 /**
  * @version $Revision: 1.3 $
  */
 public class SimpleTopicTest extends TestCase{
-    private final Log log=LogFactory.getLog(getClass());
 
+    private final Log log=LogFactory.getLog(getClass());
     protected BrokerService broker;
-//    protected String bindAddress="tcp://localhost:61616?wireFormat.cacheEnabled=true&wireFormat.tightEncodingEnabled=true&jms.useAsyncSend=false";
+    // protected String
+    // bindAddress="tcp://localhost:61616?wireFormat.cacheEnabled=true&wireFormat.tightEncodingEnabled=true&jms.useAsyncSend=false";
     protected String bindAddress="tcp://localhost:61616?wireFormat.cacheEnabled=true&wireFormat.tightEncodingEnabled=true&jms.useAsyncSend=true";
-    //protected String bindAddress="tcp://localhost:61616?wireFormat.cacheEnabled=true&wireFormat.tightEncodingEnabled=false";
+    // protected String
+    // bindAddress="tcp://localhost:61616?wireFormat.cacheEnabled=true&wireFormat.tightEncodingEnabled=false";
     // protected String bindAddress="vm://localhost?marshal=true";
     // protected String bindAddress="vm://localhost";
     protected PerfProducer[] producers;
@@ -46,7 +44,7 @@ public class SimpleTopicTest extends TestCase{
     protected String DESTINATION_NAME=getClass().getName();
     protected int SAMPLE_COUNT=30;
     protected long SAMPLE_INTERVAL=2000;
-    protected int NUMBER_OF_CONSUMERS=10;
+    protected int NUMBER_OF_CONSUMERS=1;
     protected int NUMBER_OF_PRODUCERS=1;
     protected int PAYLOAD_SIZE=1024;
     protected byte[] array=null;
@@ -66,11 +64,9 @@ public class SimpleTopicTest extends TestCase{
         factory=createConnectionFactory();
         Connection con=factory.createConnection();
         Session session=con.createSession(false,Session.AUTO_ACKNOWLEDGE);
-        
         destination=createDestination(session,DESTINATION_NAME);
         log.info("Testing against destination: "+destination);
         log.info("Running "+NUMBER_OF_PRODUCERS+" producer(s) and "+NUMBER_OF_CONSUMERS+" consumer(s)");
-        
         con.close();
         producers=new PerfProducer[NUMBER_OF_PRODUCERS];
         consumers=new PerfConsumer[NUMBER_OF_CONSUMERS];
@@ -118,7 +114,8 @@ public class SimpleTopicTest extends TestCase{
         return answer;
     }
 
-    protected PerfProducer createProducer(ConnectionFactory fac,Destination dest,int number, byte[] payload) throws JMSException{
+    protected PerfProducer createProducer(ConnectionFactory fac,Destination dest,int number,byte[] payload)
+            throws JMSException{
         return new PerfProducer(fac,dest,payload);
     }
 
@@ -129,7 +126,6 @@ public class SimpleTopicTest extends TestCase{
     protected void configureBroker(BrokerService answer) throws Exception{
         answer.addConnector(bindAddress);
         answer.setDeleteAllMessagesOnStartup(true);
-        
     }
 
     protected ActiveMQConnectionFactory createConnectionFactory() throws Exception{
@@ -137,21 +133,18 @@ public class SimpleTopicTest extends TestCase{
     }
 
     public void testPerformance() throws JMSException,InterruptedException{
-    	
         for(int i=0;i<NUMBER_OF_CONSUMERS;i++){
             consumers[i].start();
         }
         for(int i=0;i<NUMBER_OF_PRODUCERS;i++){
             producers[i].start();
         }
-        
         log.info("Sampling performance "+SAMPLE_COUNT+" times at a "+SAMPLE_INTERVAL+" ms interval.");
         for(int i=0;i<SAMPLE_COUNT;i++){
             Thread.sleep(SAMPLE_INTERVAL);
             dumpProducerRate();
             dumpConsumerRate();
         }
-        
         for(int i=0;i<NUMBER_OF_PRODUCERS;i++){
             producers[i].stop();
         }
@@ -180,7 +173,9 @@ public class SimpleTopicTest extends TestCase{
             totalRate+=rate.getRate();
             totalCount+=rate.getTotalCount();
         }
+        if(consumers!=null&&consumers.length>0){
             int avgRate=totalRate/consumers.length;
             log.info("Avg consumer rate = "+avgRate+" msg/sec | Total rate = "+totalRate+", received = "+totalCount);
         }
     }
+}

activemq-core/src/test/java/org/apache/activemq/perf/SlowConsumer.java

@@ -25,7 +25,7 @@ import javax.jms.Message;
  * @version $Revision: 1.3 $
  */
 public class SlowConsumer extends PerfConsumer{
-    public SlowConsumer(ConnectionFactory fac,Destination dest,String consumerName,boolean slowConsumer)
+    public SlowConsumer(ConnectionFactory fac,Destination dest,String consumerName)
                     throws JMSException{
         super(fac,dest,consumerName);
     }
@@ -36,6 +36,7 @@ public class SlowConsumer extends PerfConsumer{
 
     public void onMessage(Message msg){
         super.onMessage(msg);
+        System.err.println("GOT A MSG " + msg);
         try{
             Thread.sleep(10000);
         }catch(InterruptedException e){

activemq-core/src/test/java/org/apache/activemq/perf/SlowConsumerTopicTest.java

@@ -1,52 +1,46 @@
 /**
  * 
- * Licensed to the Apache Software Foundation (ASF) under one or more
+ * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE
- * contributor license agreements.  See the NOTICE file distributed with
+ * file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file
- * this work for additional information regarding copyright ownership.
+ * to You under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the
- * The ASF licenses this file to You under the Apache License, Version 2.0
+ * License. You may obtain a copy of the License at
- * (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
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
- * distributed under the License is distributed on an "AS IS" BASIS,
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * specific language governing permissions and limitations under the License.
- * See the License for the specific language governing permissions and
- * limitations under the License.
  */
+
 package org.apache.activemq.perf;
 
-import javax.jms.BytesMessage;
-import javax.jms.Connection;
 import javax.jms.ConnectionFactory;
+import javax.jms.DeliveryMode;
 import javax.jms.Destination;
 import javax.jms.JMSException;
-import javax.jms.Session;
-import junit.framework.TestCase;
-
 import org.apache.activemq.ActiveMQConnectionFactory;
+import org.apache.activemq.ActiveMQPrefetchPolicy;
 import org.apache.activemq.broker.BrokerService;
 import org.apache.activemq.xbean.BrokerFactoryBean;
-import org.apache.commons.logging.Log;
-import org.apache.commons.logging.LogFactory;
 import org.springframework.core.io.ClassPathResource;
 import org.springframework.core.io.Resource;
+
 /**
  * @version $Revision: 1.3 $
  */
 public class SlowConsumerTopicTest extends SimpleTopicTest{
+
     protected PerfConsumer[] slowConsumers;
     protected int NUMBER_OF_SLOW_CONSUMERS=1;
 
-    
     protected void setUp() throws Exception{
+        NUMBER_OF_CONSUMERS=0;
+        PAYLOAD_SIZE=10 * 1024;
         super.setUp();
-        
         slowConsumers=new SlowConsumer[NUMBER_OF_SLOW_CONSUMERS];
         for(int i=0;i<NUMBER_OF_SLOW_CONSUMERS;i++){
-            consumers[i]=createSlowConsumer(factory,destination,i);
+            slowConsumers[i]=createSlowConsumer(factory,destination,i);
-            consumers[i].start();
+            slowConsumers[i].start();
         }
     }
 
@@ -54,6 +48,13 @@ public class SlowConsumerTopicTest extends SimpleTopicTest{
         return new SlowConsumer(fac,dest);
     }
 
+    protected PerfProducer createProducer(ConnectionFactory fac,Destination dest,int number,byte[] payload)
+            throws JMSException{
+        PerfProducer result=super.createProducer(fac,dest,number,payload);
+        result.setDeliveryMode(DeliveryMode.NON_PERSISTENT);
+        return result;
+    }
+
     protected BrokerService createBroker() throws Exception{
         Resource resource=new ClassPathResource("org/apache/activemq/perf/slowConsumerBroker.xml");
         BrokerFactoryBean factory=new BrokerFactoryBean(resource);
@@ -62,4 +63,12 @@ public class SlowConsumerTopicTest extends SimpleTopicTest{
         broker.start();
         return broker;
     }
+    
+    protected ActiveMQConnectionFactory createConnectionFactory() throws Exception{
+        ActiveMQConnectionFactory result = super.createConnectionFactory();
+        ActiveMQPrefetchPolicy policy = new ActiveMQPrefetchPolicy();
+        policy.setTopicPrefetch(1000);
+        result.setPrefetchPolicy(policy);
+        return result;
+    }
 }