| /* |
| * |
| * 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.qpid.server.handler; |
| |
| import org.apache.log4j.Logger; |
| |
| import org.apache.qpid.AMQException; |
| import org.apache.qpid.framing.AMQMethodBody; |
| import org.apache.qpid.framing.AMQShortString; |
| import org.apache.qpid.framing.BasicConsumeBody; |
| import org.apache.qpid.framing.MethodRegistry; |
| import org.apache.qpid.protocol.AMQConstant; |
| import org.apache.qpid.server.AMQChannel; |
| import org.apache.qpid.server.protocol.AMQProtocolSession; |
| import org.apache.qpid.server.protocol.AMQSessionModel; |
| import org.apache.qpid.server.queue.AMQQueue; |
| import org.apache.qpid.server.state.AMQStateManager; |
| import org.apache.qpid.server.state.StateAwareMethodListener; |
| import org.apache.qpid.server.virtualhost.VirtualHost; |
| |
| public class BasicConsumeMethodHandler implements StateAwareMethodListener<BasicConsumeBody> |
| { |
| private static final Logger _logger = Logger.getLogger(BasicConsumeMethodHandler.class); |
| |
| private static final BasicConsumeMethodHandler _instance = new BasicConsumeMethodHandler(); |
| |
| public static BasicConsumeMethodHandler getInstance() |
| { |
| return _instance; |
| } |
| |
| private BasicConsumeMethodHandler() |
| { |
| } |
| |
| public void methodReceived(AMQStateManager stateManager, BasicConsumeBody body, int channelId) throws AMQException |
| { |
| AMQProtocolSession protocolConnection = stateManager.getProtocolSession(); |
| |
| AMQChannel channel = protocolConnection.getChannel(channelId); |
| VirtualHost vHost = protocolConnection.getVirtualHost(); |
| |
| if (channel == null) |
| { |
| throw body.getChannelNotFoundException(channelId); |
| } |
| else |
| { |
| channel.sync(); |
| if (_logger.isDebugEnabled()) |
| { |
| _logger.debug("BasicConsume: from '" + body.getQueue() + |
| "' for:" + body.getConsumerTag() + |
| " nowait:" + body.getNowait() + |
| " args:" + body.getArguments()); |
| } |
| |
| AMQQueue queue = body.getQueue() == null ? channel.getDefaultQueue() : vHost.getQueueRegistry().getQueue(body.getQueue().intern()); |
| |
| if (queue == null) |
| { |
| if (_logger.isDebugEnabled()) |
| { |
| _logger.debug("No queue for '" + body.getQueue() + "'"); |
| } |
| if (body.getQueue() != null) |
| { |
| String msg = "No such queue, '" + body.getQueue() + "'"; |
| throw body.getChannelException(AMQConstant.NOT_FOUND, msg); |
| } |
| else |
| { |
| String msg = "No queue name provided, no default queue defined."; |
| throw body.getConnectionException(AMQConstant.NOT_ALLOWED, msg); |
| } |
| } |
| else |
| { |
| final AMQShortString consumerTagName; |
| |
| if (queue.isExclusive() && !queue.isDurable()) |
| { |
| AMQSessionModel session = queue.getExclusiveOwningSession(); |
| if (session == null || session.getConnectionModel() != protocolConnection) |
| { |
| throw body.getConnectionException(AMQConstant.NOT_ALLOWED, |
| "Queue " + queue.getNameShortString() + " is exclusive, but not created on this Connection."); |
| } |
| } |
| |
| if (body.getConsumerTag() != null) |
| { |
| consumerTagName = body.getConsumerTag().intern(); |
| } |
| else |
| { |
| consumerTagName = null; |
| } |
| |
| try |
| { |
| if(consumerTagName == null || channel.getSubscription(consumerTagName) == null) |
| { |
| |
| AMQShortString consumerTag = channel.subscribeToQueue(consumerTagName, queue, !body.getNoAck(), |
| body.getArguments(), body.getNoLocal(), body.getExclusive()); |
| if (!body.getNowait()) |
| { |
| MethodRegistry methodRegistry = protocolConnection.getMethodRegistry(); |
| AMQMethodBody responseBody = methodRegistry.createBasicConsumeOkBody(consumerTag); |
| protocolConnection.writeFrame(responseBody.generateFrame(channelId)); |
| |
| } |
| } |
| else |
| { |
| AMQShortString msg = new AMQShortString("Non-unique consumer tag, '" + body.getConsumerTag() + "'"); |
| |
| MethodRegistry methodRegistry = protocolConnection.getMethodRegistry(); |
| AMQMethodBody responseBody = methodRegistry.createConnectionCloseBody(AMQConstant.NOT_ALLOWED.getCode(), // replyCode |
| msg, // replytext |
| body.getClazz(), |
| body.getMethod()); |
| protocolConnection.writeFrame(responseBody.generateFrame(0)); |
| } |
| |
| } |
| catch (org.apache.qpid.AMQInvalidArgumentException ise) |
| { |
| _logger.debug("Closing connection due to invalid selector"); |
| |
| MethodRegistry methodRegistry = protocolConnection.getMethodRegistry(); |
| AMQMethodBody responseBody = methodRegistry.createChannelCloseBody(AMQConstant.ARGUMENT_INVALID.getCode(), |
| new AMQShortString(ise.getMessage()), |
| body.getClazz(), |
| body.getMethod()); |
| protocolConnection.writeFrame(responseBody.generateFrame(channelId)); |
| |
| |
| } |
| catch (AMQQueue.ExistingExclusiveSubscription e) |
| { |
| throw body.getChannelException(AMQConstant.ACCESS_REFUSED, |
| "Cannot subscribe to queue " |
| + queue.getNameShortString() |
| + " as it already has an existing exclusive consumer"); |
| } |
| catch (AMQQueue.ExistingSubscriptionPreventsExclusive e) |
| { |
| throw body.getChannelException(AMQConstant.ACCESS_REFUSED, |
| "Cannot subscribe to queue " |
| + queue.getNameShortString() |
| + " exclusively as it already has a consumer"); |
| } |
| |
| } |
| } |
| } |
| } |