Force close connection on 0-10 when message sent to closing virtual host

git-svn-id: https://svn.apache.org/repos/asf/qpid/branches/QPID-6262-JavaBrokerNIO@1644374 13f79535-47bb-0310-9956-ffa450edef68

4 files changed, 87 insertions, 40 deletions

diff --git a/qpid/java/broker-core/src/main/java/org/apache/qpid/server/exchange/AbstractExchange.java b/qpid/java/broker-core/src/main/java/org/apache/qpid/server/exchange/AbstractExchange.java
index 2e7f3eee7f..6587bc76b2 100644
--- a/qpid/java/broker-core/src/main/java/org/apache/qpid/server/exchange/AbstractExchange.java
+++ b/qpid/java/broker-core/src/main/java/org/apache/qpid/server/exchange/AbstractExchange.java
@@ -64,12 +64,12 @@ import org.apache.qpid.server.queue.BaseQueue;
 import org.apache.qpid.server.store.StorableMessageMetaData;
 import org.apache.qpid.server.txn.ServerTransaction;
 import org.apache.qpid.server.util.Action;
-import org.apache.qpid.server.util.ConnectionScopedRuntimeException;
 import org.apache.qpid.server.util.StateChangeListener;
 import org.apache.qpid.server.virtualhost.ExchangeIsAlternateException;
 import org.apache.qpid.server.virtualhost.RequiredExchangeException;
 import org.apache.qpid.server.virtualhost.ReservedExchangeNameException;
 import org.apache.qpid.server.virtualhost.VirtualHostImpl;
+import org.apache.qpid.server.virtualhost.VirtualHostUnavailableException;
 
 public abstract class AbstractExchange<T extends AbstractExchange<T>>
         extends AbstractConfiguredObject<T>
@@ -510,7 +510,7 @@ public abstract class AbstractExchange<T extends AbstractExchange<T>>
     {
         if (_virtualHost.getState() != State.ACTIVE)
         {
-            throw new ConnectionScopedRuntimeException("Virtualhost state " + _virtualHost.getState() + " prevents the message from being sent");
+            throw new VirtualHostUnavailableException(this._virtualHost);
         }
 
         List<? extends BaseQueue> queues = route(message, routingAddress, instanceProperties);
@@ -910,4 +910,5 @@ public abstract class AbstractExchange<T extends AbstractExchange<T>>
             this.getVirtualHost().getDurableConfigurationStore().update(false, asObjectRecord());
         }
     }
+
 }
diff --git a/qpid/java/broker-core/src/main/java/org/apache/qpid/server/queue/AbstractQueue.java b/qpid/java/broker-core/src/main/java/org/apache/qpid/server/queue/AbstractQueue.java
index 8328baca3f..f941112437 100644
--- a/qpid/java/broker-core/src/main/java/org/apache/qpid/server/queue/AbstractQueue.java
+++ b/qpid/java/broker-core/src/main/java/org/apache/qpid/server/queue/AbstractQueue.java
@@ -50,6 +50,7 @@ import org.apache.qpid.server.connection.SessionPrincipal;
 import org.apache.qpid.server.consumer.ConsumerImpl;
 import org.apache.qpid.server.consumer.ConsumerTarget;
 import org.apache.qpid.server.exchange.ExchangeImpl;
+import org.apache.qpid.server.virtualhost.VirtualHostUnavailableException;
 import org.apache.qpid.server.filter.FilterManager;
 import org.apache.qpid.server.logging.EventLogger;
 import org.apache.qpid.server.logging.LogMessage;
@@ -2521,7 +2522,7 @@ public abstract class AbstractQueue<X extends AbstractQueue<X>>
     {
         if (_virtualHost.getState() != State.ACTIVE)
         {
-            throw new ConnectionScopedRuntimeException("Virtualhost state " + _virtualHost.getState() + " prevents the message from being sent");
+            throw new VirtualHostUnavailableException(this._virtualHost);
         }
 
         if(!message.isReferenced(this))
diff --git a/qpid/java/broker-core/src/main/java/org/apache/qpid/server/virtualhost/VirtualHostUnavailableException.java b/qpid/java/broker-core/src/main/java/org/apache/qpid/server/virtualhost/VirtualHostUnavailableException.java
new file mode 100644
index 0000000000..a0bab0b0a6
--- /dev/null
+++ b/qpid/java/broker-core/src/main/java/org/apache/qpid/server/virtualhost/VirtualHostUnavailableException.java
@@ -0,0 +1,33 @@
+/*
+ *
+ * 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.virtualhost;
+
+import org.apache.qpid.server.util.ConnectionScopedRuntimeException;
+
+public class VirtualHostUnavailableException extends ConnectionScopedRuntimeException
+{
+    public VirtualHostUnavailableException(VirtualHostImpl<?, ?, ?> host)
+    {
+        super("Virtualhost state "
+              + host.getState()
+              + " prevents the message from being sent");
+    }
+}
diff --git a/qpid/java/broker-plugins/amqp-0-10-protocol/src/main/java/org/apache/qpid/server/protocol/v0_10/ServerSessionDelegate.java b/qpid/java/broker-plugins/amqp-0-10-protocol/src/main/java/org/apache/qpid/server/protocol/v0_10/ServerSessionDelegate.java
index 288a4f946c..a117ddb0c6 100644
--- a/qpid/java/broker-plugins/amqp-0-10-protocol/src/main/java/org/apache/qpid/server/protocol/v0_10/ServerSessionDelegate.java
+++ b/qpid/java/broker-plugins/amqp-0-10-protocol/src/main/java/org/apache/qpid/server/protocol/v0_10/ServerSessionDelegate.java
@@ -37,6 +37,7 @@ import org.apache.qpid.exchange.ExchangeDefaults;
 import org.apache.qpid.protocol.AMQConstant;
 import org.apache.qpid.server.consumer.ConsumerImpl;
 import org.apache.qpid.server.exchange.ExchangeImpl;
+import org.apache.qpid.server.virtualhost.VirtualHostUnavailableException;
 import org.apache.qpid.server.filter.AMQInvalidArgumentException;
 import org.apache.qpid.server.filter.FilterManager;
 import org.apache.qpid.server.filter.FilterManagerFactory;
@@ -381,58 +382,69 @@ public class ServerSessionDelegate extends SessionDelegate
                     new MessageTransferMessage(storeMessage, serverSession.getReference());
             MessageReference<MessageTransferMessage> reference = message.newReference();
 
-            final InstanceProperties instanceProperties = new InstanceProperties()
+            try
             {
-                @Override
-                public Object getProperty(final Property prop)
+                final InstanceProperties instanceProperties = new InstanceProperties()
                 {
-                    switch (prop)
+                    @Override
+                    public Object getProperty(final Property prop)
                     {
-                        case EXPIRATION:
-                            return message.getExpiration();
-                        case IMMEDIATE:
-                            return message.isImmediate();
-                        case MANDATORY:
-                            return (delvProps == null || !delvProps.getDiscardUnroutable())
-                                   && xfr.getAcceptMode() == MessageAcceptMode.EXPLICIT;
-                        case PERSISTENT:
-                            return message.isPersistent();
-                        case REDELIVERED:
-                            return delvProps.getRedelivered();
+                        switch (prop)
+                        {
+                            case EXPIRATION:
+                                return message.getExpiration();
+                            case IMMEDIATE:
+                                return message.isImmediate();
+                            case MANDATORY:
+                                return (delvProps == null || !delvProps.getDiscardUnroutable())
+                                       && xfr.getAcceptMode() == MessageAcceptMode.EXPLICIT;
+                            case PERSISTENT:
+                                return message.isPersistent();
+                            case REDELIVERED:
+                                return delvProps.getRedelivered();
+                        }
+                        return null;
                     }
-                    return null;
-                }
-            };
+                };
 
-            int enqueues = serverSession.enqueue(message, instanceProperties, destination);
+                int enqueues = serverSession.enqueue(message, instanceProperties, destination);
 
-            if (enqueues == 0)
-            {
-                if ((delvProps == null || !delvProps.getDiscardUnroutable())
-                    && xfr.getAcceptMode() == MessageAcceptMode.EXPLICIT)
+                if (enqueues == 0)
+                {
+                    if ((delvProps == null || !delvProps.getDiscardUnroutable())
+                        && xfr.getAcceptMode() == MessageAcceptMode.EXPLICIT)
+                    {
+                        RangeSet rejects = RangeSetFactory.createRangeSet();
+                        rejects.add(xfr.getId());
+                        MessageReject reject = new MessageReject(rejects, MessageRejectCode.UNROUTABLE, "Unroutable");
+                        ssn.invoke(reject);
+                    }
+                    else
+                    {
+                        virtualHost.getEventLogger().message(ExchangeMessages.DISCARDMSG(destination.getName(),
+                                                                                         messageMetaData.getRoutingKey()));
+                    }
+                }
+
+                if (serverSession.isTransactional())
                 {
-                    RangeSet rejects = RangeSetFactory.createRangeSet();
-                    rejects.add(xfr.getId());
-                    MessageReject reject = new MessageReject(rejects, MessageRejectCode.UNROUTABLE, "Unroutable");
-                    ssn.invoke(reject);
+                    serverSession.processed(xfr);
                 }
                 else
                 {
-                    virtualHost.getEventLogger().message(ExchangeMessages.DISCARDMSG(destination.getName(),
-                                                                                     messageMetaData.getRoutingKey()));
+                    serverSession.recordFuture(StoreFuture.IMMEDIATE_FUTURE,
+                                               new CommandProcessedAction(serverSession, xfr));
                 }
             }
-
-            if (serverSession.isTransactional())
+            catch (VirtualHostUnavailableException e)
             {
-                serverSession.processed(xfr);
+                getServerConnection(serverSession).close(AMQConstant.CONNECTION_FORCED, e.getMessage());
             }
-            else
+            finally
             {
-                serverSession.recordFuture(StoreFuture.IMMEDIATE_FUTURE,
-                                           new CommandProcessedAction(serverSession, xfr));
+                reference.release();
             }
-            reference.release();
+
         }
     }
 
@@ -549,7 +561,7 @@ public class ServerSessionDelegate extends SessionDelegate
         {
             try
             {
-                ((ServerSession)session).endDtx(method.getXid(), method.getFail(), method.getSuspend());
+                ((ServerSession) session).endDtx(method.getXid(), method.getFail(), method.getSuspend());
             }
             catch (TimeoutDtxException e)
             {