/*
* 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.qpid.AMQException;
import org.apache.qpid.framing.QueuePurgeBody;
import org.apache.qpid.framing.MethodRegistry;
import org.apache.qpid.framing.AMQMethodBody;
import org.apache.qpid.protocol.AMQConstant;
import org.apache.qpid.server.protocol.AMQProtocolSession;
import org.apache.qpid.server.queue.QueueRegistry;
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;
import org.apache.qpid.server.AMQChannel;
import org.apache.qpid.server.security.access.Permission;
public class QueuePurgeHandler implements StateAwareMethodListener<QueuePurgeBody>
{
private static final QueuePurgeHandler _instance = new QueuePurgeHandler();
public static QueuePurgeHandler getInstance()
{
return _instance;
}
private final boolean _failIfNotFound;
public QueuePurgeHandler()
{
this(true);
}
public QueuePurgeHandler(boolean failIfNotFound)
{
_failIfNotFound = failIfNotFound;
}
public void methodReceived(AMQStateManager stateManager, QueuePurgeBody body, int channelId) throws AMQException
{
AMQProtocolSession session = stateManager.getProtocolSession();
VirtualHost virtualHost = session.getVirtualHost();
QueueRegistry queueRegistry = virtualHost.getQueueRegistry();
AMQChannel channel = session.getChannel(channelId);
AMQQueue queue;
if(body.getQueue() == null)
{
if (channel == null)
{
throw body.getChannelNotFoundException(channelId);
}
//get the default queue on the channel:
queue = channel.getDefaultQueue();
if(queue == null)
{
if(_failIfNotFound)
{
throw body.getConnectionException(AMQConstant.NOT_ALLOWED,"No queue specified.");
}
}
}
else
{
queue = queueRegistry.getQueue(body.getQueue());
}
if(queue == null)
{
if(_failIfNotFound)
{
throw body.getChannelException(AMQConstant.NOT_FOUND, "Queue " + body.getQueue() + " does not exist.");
}
}
else
{
//Perform ACLs
virtualHost.getAccessManager().authorise(session, Permission.PURGE, body, queue);
long purged = queue.clearQueue(channel.getStoreContext());
if(!body.getNowait())
{
MethodRegistry methodRegistry = session.getMethodRegistry();
AMQMethodBody responseBody = methodRegistry.createQueuePurgeOkBody(purged);
session.writeFrame(responseBody.generateFrame(channelId));
}
}
}
}