From d9ed5bf6152fe743928f18c1c821197072d89964 Mon Sep 17 00:00:00 2001
From: Robert Godfrey <rgodfrey@apache.org>
Date: Sun, 9 Feb 2014 22:57:49 +0000
Subject: Implement update

git-svn-id: https://svn.apache.org/repos/asf/qpid/branches/java-broker-amqp-1-0-management@1566462 13f79535-47bb-0310-9956-ffa450edef68
---
 .../server/management/amqp/ManagementNode.java     | 48 ++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/java/broker-plugins/management-amqp/src/main/java/org/apache/qpid/server/management/amqp/ManagementNode.java b/java/broker-plugins/management-amqp/src/main/java/org/apache/qpid/server/management/amqp/ManagementNode.java
index 2958fed0a1..18795f58b8 100644
--- a/java/broker-plugins/management-amqp/src/main/java/org/apache/qpid/server/management/amqp/ManagementNode.java
+++ b/java/broker-plugins/management-amqp/src/main/java/org/apache/qpid/server/management/amqp/ManagementNode.java
@@ -422,6 +422,10 @@ class ManagementNode implements MessageSource<ManagementNodeConsumer,ManagementN
         {
             return performDeleteOperation(requestMessage, entity);
         }
+        else if(UPDATE_OPERATION.equals(operation))
+        {
+            return performUpdateOperation(requestMessage, entity);
+        }
         else
         {
             return createFailureResponse(requestMessage, NOT_IMPLEMENTED_STATUS_CODE, "Unable to perform the {0} operation",operation);
@@ -486,6 +490,50 @@ class ManagementNode implements MessageSource<ManagementNodeConsumer,ManagementN
         return InternalMessage.createMapMessage(_virtualHost.getMessageStore(),responseHeader, Collections.emptyMap());
     }
 
+
+    private InternalMessage performUpdateOperation(final InternalMessage requestMessage, final ConfiguredObject entity)
+    {
+        final InternalMessageHeader requestHeader = requestMessage.getMessageHeader();
+        final MutableMessageHeader responseHeader = new MutableMessageHeader();
+        responseHeader.setCorrelationId(requestHeader.getCorrelationId() == null
+                                                ? requestHeader.getMessageId()
+                                                : requestHeader.getCorrelationId());
+        responseHeader.setMessageId(UUID.randomUUID().toString());
+        responseHeader.setHeader(NAME_ATTRIBUTE, entity.getName());
+        responseHeader.setHeader(IDENTITY_ATTRIBUTE, entity.getId().toString());
+        final String type = getManagementClass(entity.getClass()).getName();
+        responseHeader.setHeader(TYPE_ATTRIBUTE, type);
+
+        Object messageBody = requestMessage.getMessageBody();
+        if(messageBody instanceof Map)
+        {
+            try
+            {
+                entity.setAttributes((Map)messageBody);
+                return performReadOperation(requestMessage, entity);
+            }
+            catch(RuntimeException e)
+            {
+                if (e instanceof AccessControlException || e.getCause() instanceof AMQSecurityException)
+                {
+                    return createFailureResponse(requestMessage, STATUS_CODE_FORBIDDEN, e.getMessage());
+                }
+                else
+                {
+                    throw e;
+                }
+            }
+        }
+        else
+        {
+            return createFailureResponse(requestMessage,
+                                             STATUS_CODE_BAD_REQUEST,
+                                             "The message body in the request was not of the correct type");
+        }
+
+
+    }
+
     private ConfiguredObject findSubject(final String name, final String id, final String type)
     {
         ConfiguredObject subject;
-- 
cgit v1.2.1