1 files changed, 139 insertions, 0 deletions
diff --git a/cpp/lib/broker/BrokerMessage.h b/cpp/lib/broker/BrokerMessage.h
new file mode 100644
index 0000000000..59e146959d
--- /dev/null
+++ b/cpp/lib/broker/BrokerMessage.h
@@ -0,0 +1,139 @@
+/*
+ *
+ * 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.
+ *
+ */
+#ifndef _Message_
+#define _Message_
+
+#include <memory>
+#include <boost/shared_ptr.hpp>
+#include <ConnectionToken.h>
+#include <Content.h>
+#include <TxBuffer.h>
+#include <AMQContentBody.h>
+#include <AMQHeaderBody.h>
+#include <BasicHeaderProperties.h>
+#include <OutputHandler.h>
+
+namespace qpid {
+    namespace broker {
+
+        class MessageStore;
+        using qpid::framing::string;
+        /**
+         * Represents an AMQP message, i.e. a header body, a list of
+         * content bodies and some details about the publication
+         * request.
+         */
+        class Message{
+            const ConnectionToken* const publisher;
+            string exchange;
+            string routingKey;
+            const bool mandatory;
+            const bool immediate;
+            bool redelivered;
+            qpid::framing::AMQHeaderBody::shared_ptr header;
+            std::auto_ptr<Content> content;
+            u_int64_t size;
+            u_int64_t persistenceId;
+
+            void sendContent(qpid::framing::OutputHandler* out, 
+                             int channel, u_int32_t framesize);
+
+        public:
+            typedef boost::shared_ptr<Message> shared_ptr;
+
+            Message(const ConnectionToken* const publisher, 
+                    const string& exchange, const string& routingKey, 
+                    bool mandatory, bool immediate);
+            Message(qpid::framing::Buffer& buffer, bool headersOnly = false, u_int32_t contentChunkSize = 0);
+            Message();
+            ~Message();
+            void setHeader(qpid::framing::AMQHeaderBody::shared_ptr header);
+            void addContent(qpid::framing::AMQContentBody::shared_ptr data);
+            bool isComplete();
+            const ConnectionToken* const getPublisher();
+
+            void deliver(qpid::framing::OutputHandler* out, 
+                         int channel, 
+                         const string& consumerTag, 
+                         u_int64_t deliveryTag, 
+                         u_int32_t framesize);
+            void sendGetOk(qpid::framing::OutputHandler* out, 
+                           int channel, 
+                           u_int32_t messageCount,
+                           u_int64_t deliveryTag, 
+                           u_int32_t framesize);
+            void redeliver();
+
+            qpid::framing::BasicHeaderProperties* getHeaderProperties();
+            bool isPersistent();
+            const string& getRoutingKey() const { return routingKey; }
+            const string& getExchange() const { return exchange; }
+            u_int64_t contentSize() const { return size; }
+            u_int64_t getPersistenceId() const { return persistenceId; }
+            void setPersistenceId(u_int64_t _persistenceId) { persistenceId = _persistenceId; }
+
+            void decode(qpid::framing::Buffer& buffer, bool headersOnly = false, u_int32_t contentChunkSize = 0);
+            void decodeHeader(qpid::framing::Buffer& buffer);
+            void decodeContent(qpid::framing::Buffer& buffer, u_int32_t contentChunkSize = 0);
+
+            void encode(qpid::framing::Buffer& buffer);
+            void encodeHeader(qpid::framing::Buffer& buffer);
+            void encodeContent(qpid::framing::Buffer& buffer);
+            /**
+             * @returns the size of the buffer needed to encode this
+             * message in its entirety
+             */
+            u_int32_t encodedSize();
+            /**
+             * @returns the size of the buffer needed to encode the
+             * 'header' of this message (not just the header frame,
+             * but other meta data e.g.routing key and exchange)
+             */
+            u_int32_t encodedHeaderSize();
+            /**
+             * @returns the size of the buffer needed to encode the
+             * (possibly partial) content held by this message
+             */
+            u_int32_t encodedContentSize();
+            /**
+             * Releases the in-memory content data held by this
+             * message. Must pass in a store from which the data can
+             * be reloaded.
+             */
+            void releaseContent(MessageStore* store);
+            /**
+             * If headers have been received, returns the expected
+             * content size else returns 0.
+             */
+            u_int64_t expectedContentSize();
+            /**
+             * Sets the 'content' implementation of this message (the
+             * message controls the lifecycle of the content instance
+             * it uses).
+             */
+            void setContent(std::auto_ptr<Content>& content);
+        };
+
+    }
+}
+
+
+#endif