Adding destination directive and wrapper, with appropriate unit test.

mikhail_beris · mikhail_beris · commit c44e28cbc927 · 2007-06-03T06:49:16.000Z
diff --git a/boost/network/message.hpp b/boost/network/message.hpp
@@ -54,6 +54,10 @@ namespace boost { namespace network {
             return _source;
         };
 
+        std::string & destination() const {
+            return _destination;
+        };
+
         private:
         
         friend struct detail::directive_base<tag> ;
@@ -62,6 +66,7 @@ namespace boost { namespace network {
         mutable headers_container_type _headers;
         mutable std::string _body;
         mutable std::string _source;
+        mutable std::string _destination;
     };
 
     typedef basic_message<> message; // default message type
diff --git a/boost/network/message/directives.hpp b/boost/network/message/directives.hpp
@@ -13,6 +13,7 @@
 #include <boost/network/message/directives/header.hpp>
 #include <boost/network/message/directives/body.hpp>
 #include <boost/network/message/directives/source.hpp>
+#include <boost/network/message/directives/destination.hpp>
 
 namespace boost { namespace network {
 
diff --git a/boost/network/message/directives/destination.hpp b/boost/network/message/directives/destination.hpp
@@ -0,0 +1,49 @@
+
+//          Copyright Dean Michael Berris 2007.
+// Distributed under the Boost Software License, Version 1.0.
+//    (See accompanying file LICENSE_1_0.txt or copy at
+//          http://www.boost.org/LICENSE_1_0.txt)
+
+#ifndef __NETWORK_MESSAGE_DIRECTIVES_DESTINATION_HPP__
+#define __NETWORK_MESSAGE_DIRECTIVES_DESTINATION_HPP__
+
+/** destination.hpp
+ *
+ * Defines the types involved and the semantics of adding
+ * destination information into message objects.
+ *
+ * WARNING: DO NOT INCLUDE THIS HEADER DIRECTLY. THIS REQUIRES
+ * TYPES TO BE DEFINED FROM EARLIER FILES THAT INCLUDE THIS
+ * HEADER.
+ */
+namespace boost { namespace network {
+    
+    namespace impl {
+        template <class Tag>
+            struct destination_directive : public detail::directive_base<Tag> {
+                typedef Tag tag;
+
+                explicit destination_directive ( std::string const & destination)
+                    : _destination(destination)
+                { };
+
+                void operator() (basic_message<tag> & msg) const {
+                    msg.destination() = _destination;
+                };
+
+                private:
+
+                std::string _destination;
+            };
+    }; // namespace impl
+
+    inline impl::destination_directive<tags::default_>
+        destination(std::string const & destination_) {
+            return impl::destination_directive<tags::default_>(destination_);
+        };
+
+}; // namespace network
+
+}; // namespace boost
+
+#endif // __NETWORK_MESSAGE_DIRECTIVES_DESTINATION_HPP__
diff --git a/boost/network/message/directives/source.hpp b/boost/network/message/directives/source.hpp
@@ -7,7 +7,7 @@
 #ifndef __NETWORK_MESSAGE_DIRECTIVES_SOURCE_HPP__
 #define __NETWORK_MESSAGE_DIRECTIVES_SOURCE_HPP__
 
-/** header.hpp
+/** source.hpp
  *
  * Defines the types involved and the semantics of adding
  * source information into message objects.
diff --git a/boost/network/message/wrappers.hpp b/boost/network/message/wrappers.hpp
@@ -14,5 +14,6 @@
 #include <boost/network/message/wrappers/headers.hpp>
 #include <boost/network/message/wrappers/body.hpp>
 #include <boost/network/message/wrappers/source.hpp>
+#include <boost/network/message/wrappers/destination.hpp>
 
 #endif // __NETWORK_MESSAGE_WRAPPERS_HPP__
diff --git a/boost/network/message/wrappers/destination.hpp b/boost/network/message/wrappers/destination.hpp
@@ -0,0 +1,39 @@
+
+//          Copyright Dean Michael Berris 2007.
+// Distributed under the Boost Software License, Version 1.0.
+//    (See accompanying file LICENSE_1_0.txt or copy at
+//          http://www.boost.org/LICENSE_1_0.txt)
+
+#ifndef __NETWORK_MESSAGE_WRAPPERS_DESTINATION_HPP__
+#define __NETWORK_MESSAGE_WRAPPERS_DESTINATION_HPP__
+
+namespace boost { namespace network {
+
+    namespace impl {
+        template <class Tag>
+            struct destination_wrapper : public detail::wrapper_base<Tag> {
+                typedef Tag tag;
+                typedef basic_message<tag> message_type;
+
+                explicit destination_wrapper(message_type & message_)
+                    : detail::wrapper_base<tag>(message_)
+                { };
+
+                operator std::string () const {
+                    return std::string(detail::wrapper_base<tag>::_message.destination());
+                };
+            };
+    }; // namespace impl
+
+    template <class Tag>
+        inline std::string
+        destination(basic_message<Tag> & message_) {
+            return impl::destination_wrapper<Tag>(message_);
+        };
+
+}; // namespace network
+
+}; // namespace boost
+
+#endif __NETWORK_MESSAGE_WRAPPERS_DESTINATION_HPP__
+
diff --git a/libs/network/test/message_test.cpp b/libs/network/test/message_test.cpp
@@ -48,4 +48,12 @@ BOOST_AUTO_TEST_CASE(source_directive_test) {
 	msg << source("Somewhere Out There") ;
 	
 	BOOST_CHECK_EQUAL( source(msg), "Somewhere Out There");
-}
+}
+
+BOOST_AUTO_TEST_CASE(destination_directive_test) {
+    using namespace boost::network;
+    message msg;
+    msg << destination("Somewhere Out There");
+
+    BOOST_CHECK_EQUAL( destination(msg), "Somewhere Out There");
+};

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@`
`7`	`7`	`#ifndef __NETWORK_MESSAGE_DIRECTIVES_SOURCE_HPP__`
`8`	`8`	`#define __NETWORK_MESSAGE_DIRECTIVES_SOURCE_HPP__`
`9`	`9`
`10`		`-/** header.hpp`
	`10`	`+/** source.hpp`
`11`	`11`	`*`
`12`	`12`	`* Defines the types involved and the semantics of adding`
`13`	`13`	`* source information into message objects.`