Apache
/
activemq
mirror of https://github.com/apache/activemq.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Do better reference count management of persistent messages when they are stored in a Memory store. 

git-svn-id: https://svn.apache.org/repos/asf/activemq/trunk@644447 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Hiram R. Chirino 2008-04-03 19:26:04 +00:00
parent d35f83bcf5
commit ce1d85d1ee
3 changed files with 23 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
activemq-core/src/main/java/org/apache/activemq/broker/region/cursors/AbstractStoreCursor.java
View File

@ -77,8 +77,10 @@ public abstract class AbstractStoreCursor extends AbstractPendingMessageCursor i
if (!isDuplicate(message.getMessageId())) {
if (!cached) {
message.setRegionDestination(regionDestination);
if( message.getMemoryUsage()==null ) {
message.setMemoryUsage(this.getSystemUsage().getMemoryUsage());
}
}
message.incrementReferenceCount();
batchList.put(message.getMessageId(), message);
} else {

6
activemq-core/src/main/java/org/apache/activemq/store/memory/MemoryMessageStore.java
View File

@ -58,6 +58,7 @@ public class MemoryMessageStore implements MessageStore {
synchronized (messageTable) {
messageTable.put(message.getMessageId(), message);
}
message.incrementReferenceCount();
}
// public void addMessageReference(ConnectionContext context,MessageId
@ -82,7 +83,10 @@ public class MemoryMessageStore implements MessageStore {
public void removeMessage(MessageId msgId) throws IOException {
synchronized (messageTable) {
messageTable.remove(msgId);
Message removed = messageTable.remove(msgId);
if( removed !=null ) {
removed.decrementReferenceCount();
}
if ((lastBatchId != null && lastBatchId.equals(msgId)) || messageTable.isEmpty()) {
lastBatchId = null;
}

15
activemq-core/src/main/java/org/apache/activemq/store/memory/MemoryTopicSub.java
View File

@ -34,16 +34,25 @@ class MemoryTopicSub {
private Map<MessageId, Message> map = new LinkedHashMap<MessageId, Message>();
private MessageId lastBatch;
synchronized void addMessage(MessageId id, Message message) {
void addMessage(MessageId id, Message message) {
synchronized(this) {
map.put(id, message);
}
message.incrementReferenceCount();
}
synchronized void removeMessage(MessageId id) {
map.remove(id);
void removeMessage(MessageId id) {
Message removed;
synchronized(this) {
removed = map.remove(id);
if ((lastBatch != null && lastBatch.equals(id)) || map.isEmpty()) {
resetBatching();
}
}
if( removed!=null ) {
removed.decrementReferenceCount();
}
}
synchronized int size() {
return map.size();