diff options
Diffstat (limited to 'cpp/src/qpid/broker/MessageDelivery.cpp')
-rw-r--r-- | cpp/src/qpid/broker/MessageDelivery.cpp | 140 |
1 files changed, 140 insertions, 0 deletions
diff --git a/cpp/src/qpid/broker/MessageDelivery.cpp b/cpp/src/qpid/broker/MessageDelivery.cpp new file mode 100644 index 0000000000..09ab8ec465 --- /dev/null +++ b/cpp/src/qpid/broker/MessageDelivery.cpp @@ -0,0 +1,140 @@ +/* + * + * 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. + * + */ +#include "MessageDelivery.h" + +#include "DeliveryToken.h" +#include "Message.h" +#include "BrokerQueue.h" +#include "qpid/framing/ChannelAdapter.h" +#include "qpid/framing/BasicDeliverBody.h" +#include "qpid/framing/BasicGetOkBody.h" +#include "qpid/framing/MessageTransferBody.h" + + +using namespace boost; +using namespace qpid::broker; +using namespace qpid::framing; + +namespace qpid{ +namespace broker{ + +struct BaseToken : DeliveryToken +{ + virtual ~BaseToken() {} + virtual void sendMethod(Message::shared_ptr msg, ChannelAdapter& channel, DeliveryId id) = 0; +}; + +struct BasicGetToken : BaseToken +{ + typedef boost::shared_ptr<BasicGetToken> shared_ptr; + + Queue::shared_ptr queue; + + BasicGetToken(Queue::shared_ptr q) : queue(q) {} + + void sendMethod(Message::shared_ptr msg, ChannelAdapter& channel, DeliveryId id) + { + channel.send(BasicGetOkBody( + channel.getVersion(), id.getValue(), msg->getRedelivered(), msg->getExchangeName(), + msg->getRoutingKey(), queue->getMessageCount())); + + } +}; + +struct BasicConsumeToken : BaseToken +{ + typedef boost::shared_ptr<BasicConsumeToken> shared_ptr; + + const string consumer; + + BasicConsumeToken(const string c) : consumer(c) {} + + void sendMethod(Message::shared_ptr msg, ChannelAdapter& channel, DeliveryId id) + { + channel.send(BasicDeliverBody( + channel.getVersion(), consumer, id.getValue(), + msg->getRedelivered(), msg->getExchangeName(), msg->getRoutingKey())); + } + +}; + +struct MessageDeliveryToken : BaseToken +{ + const std::string destination; + const u_int8_t confirmMode; + const u_int8_t acquireMode; + + MessageDeliveryToken(const std::string& d, u_int8_t c, u_int8_t a) : + destination(d), confirmMode(c), acquireMode(a) {} + + void sendMethod(Message::shared_ptr msg, ChannelAdapter& channel, DeliveryId /*id*/) + { + //TODO; need to figure out how the acquire mode gets + //communicated (this is just a temporary solution) + channel.send(MessageTransferBody(channel.getVersion(), 0, destination, confirmMode, acquireMode)); + + //may need to set the redelivered flag: + if (msg->getRedelivered()){ + msg->getProperties<DeliveryProperties>()->setRedelivered(true); + } + } +}; + +} +} + +DeliveryToken::shared_ptr MessageDelivery::getBasicGetToken(Queue::shared_ptr queue) +{ + return DeliveryToken::shared_ptr(new BasicGetToken(queue)); +} + +DeliveryToken::shared_ptr MessageDelivery::getBasicConsumeToken(const string& consumer) +{ + return DeliveryToken::shared_ptr(new BasicConsumeToken(consumer)); +} + +DeliveryToken::shared_ptr MessageDelivery::getMessageDeliveryToken(const std::string& destination, + u_int8_t confirmMode, u_int8_t acquireMode) +{ + return DeliveryToken::shared_ptr(new MessageDeliveryToken(destination, confirmMode, acquireMode)); +} + +void MessageDelivery::deliver(Message::shared_ptr msg, + framing::ChannelAdapter& channel, + DeliveryId id, + DeliveryToken::shared_ptr token, + uint16_t framesize) +{ + //currently a message published from one class and delivered to + //another may well have the wrong headers; however we will only + //have one content class for 0-10 proper + + //send method + boost::shared_ptr<BaseToken> t = dynamic_pointer_cast<BaseToken>(token); + t->sendMethod(msg, channel, id); + + boost::shared_ptr<FrameHandler> handler = channel.getHandlers().out; + //send header + msg->sendHeader(*handler, channel.getId(), framesize); + + //send content + msg->sendContent(*handler, channel.getId(), framesize); +} |