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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/*
*
* Copyright (c) 2006 The Apache Software Foundation
*
* Licensed 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.qpid.server.handler;
import org.apache.log4j.Logger;
import org.apache.qpid.AMQException;
import org.apache.qpid.configuration.Configured;
import org.apache.qpid.framing.AMQFrame;
import org.apache.qpid.framing.QueueDeclareBody;
import org.apache.qpid.framing.QueueDeclareOkBody;
import org.apache.qpid.server.exchange.ExchangeRegistry;
import org.apache.qpid.server.exchange.Exchange;
import org.apache.qpid.server.protocol.AMQMethodEvent;
import org.apache.qpid.server.protocol.AMQProtocolSession;
import org.apache.qpid.server.queue.AMQQueue;
import org.apache.qpid.server.queue.QueueRegistry;
import org.apache.qpid.server.state.AMQStateManager;
import org.apache.qpid.server.state.StateAwareMethodListener;
import org.apache.qpid.server.configuration.Configurator;
import org.apache.qpid.server.store.MessageStore;
import org.apache.qpid.server.registry.ApplicationRegistry;
import java.text.MessageFormat;
import java.util.concurrent.atomic.AtomicInteger;
public class QueueDeclareHandler implements StateAwareMethodListener<QueueDeclareBody>
{
private static final Logger _log = Logger.getLogger(QueueDeclareHandler.class);
private static final QueueDeclareHandler _instance = new QueueDeclareHandler();
public static QueueDeclareHandler getInstance()
{
return _instance;
}
@Configured(path = "queue.auto_register", defaultValue = "false")
public boolean autoRegister;
private final AtomicInteger _counter = new AtomicInteger();
private final MessageStore _store;
protected QueueDeclareHandler()
{
Configurator.configure(this);
_store = ApplicationRegistry.getInstance().getMessageStore();
}
public void methodReceived(AMQStateManager stateManager, QueueRegistry queueRegistry,
ExchangeRegistry exchangeRegistry, AMQProtocolSession protocolSession,
AMQMethodEvent<QueueDeclareBody> evt) throws AMQException
{
QueueDeclareBody body = evt.getMethod();
// if we aren't given a queue name, we create one which we return to the client
if (body.queue == null)
{
body.queue = createName();
}
//TODO: do we need to check that the queue already exists with exactly the same "configuration"?
synchronized (queueRegistry)
{
AMQQueue queue;
if ((queue = queueRegistry.getQueue(body.queue)) == null)
{
queue = createQueue(body, queueRegistry, protocolSession);
if (queue.isDurable() && !queue.isAutoDelete())
{
_store.createQueue(queue);
}
queueRegistry.registerQueue(queue);
if (autoRegister)
{
Exchange defaultExchange = exchangeRegistry.getExchange("amq.direct");
defaultExchange.registerQueue(body.queue, queue, null);
queue.bind(body.queue, defaultExchange);
_log.info("Queue " + body.queue + " bound to default exchange");
}
}
//set this as the default queue on the channel:
protocolSession.getChannel(evt.getChannelId()).setDefaultQueue(queue);
}
if (!body.nowait)
{
AMQFrame response = QueueDeclareOkBody.createAMQFrame(evt.getChannelId(), body.queue, 0L, 0L);
_log.info("Queue " + body.queue + " declared successfully");
protocolSession.writeFrame(response);
}
}
protected String createName()
{
return "tmp_" + pad(_counter.incrementAndGet());
}
protected static String pad(int value)
{
return MessageFormat.format("{0,number,0000000000000}", value);
}
protected AMQQueue createQueue(QueueDeclareBody body, QueueRegistry registry, AMQProtocolSession session)
throws AMQException
{
String owner = body.exclusive ? session.getContextKey() : null;
return new AMQQueue(body.queue, body.durable, owner, body.autoDelete, registry);
}
}
|
