java/newclient/src/main/java/org/apache/qpid/nclient/message/TransientMessageStore.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

package org.apache.qpid.nclient.message;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import org.apache.qpid.nclient.core.AMQPException;

public class TransientMessageStore implements MessageStore {

	private Map<String,AMQPApplicationMessage> _messageMap = new ConcurrentHashMap<String,AMQPApplicationMessage>();
	
	public AMQPApplicationMessage getMessage(String identifier)
			throws AMQPException 
	{
		if (!_messageMap.containsKey(identifier))
		{
			throw new AMQPException("identifier not found " + identifier);
		}
		
		return _messageMap.get(identifier);		
	}

	public void removeMessage(String identifier) throws AMQPException
	{
		if (!_messageMap.containsKey(identifier))
		{
			throw new AMQPException("identifier not found " + identifier);
		}
		_messageMap.remove(identifier);
	}

	public void storeContentBodyChunk(String identifier, byte[] contentBody)
			throws AMQPException 
	{
		AMQPApplicationMessage msg = _messageMap.get(identifier);
		msg.addContent(contentBody);
	}

	public void storeMessageMetaData(String identifier,
			MessageHeaders messageHeaders) throws AMQPException 
	{
		AMQPApplicationMessage msg = _messageMap.get(identifier);
		msg.setMessageHeaders(messageHeaders);
	}

	public void storeMessage(String identifier,AMQPApplicationMessage msg)throws AMQPException
	{
		_messageMap.put(identifier, msg);
	}
}