iostream sinks/sources

Author: me

src/msgpack.h

@@ -274,6 +274,11 @@ namespace msgpackcpp
     template<class... Args>
     void deserialize(source_base& in, std::tuple<Args...>& tpl);
 
+//----------------------------------------------------------------------------------------------------------------
+
+    void serialize_from_value(sink_base& out, const value& jv);
+    auto deserialize_to_value(source_base& in) -> value;
+    
 //----------------------------------------------------------------------------------------------------------------
 //----------------------------------------------------------------------------------------------------------------
 // DEFINITIONS

src/msgpack_sinks.h

@@ -62,19 +62,39 @@ namespace msgpackcpp
 
 //----------------------------------------------------------------------------------------------------------------
 
-    // inline auto sink(std::ostream& out)
-    // {
-    //     return [&](const char* bytes, size_t nbytes) {
-    //         out.write(bytes, nbytes);
-    //     };
-    // }
-
-    // inline auto source(std::istream& in)
-    // {
-    //     return [&](char* bytes, size_t nbytes) {
-    //         in.read(bytes, nbytes);
-    //         if (in.gcount() != (long)nbytes)
-    //             throw std::system_error(OUT_OF_DATA);
-    //     };
-    // }
+    struct ostream_sink : sink_base
+    {
+        std::ostream& out;
+
+        ostream_sink(std::ostream& out_) : out{out_}{}
+        void write(const char* data, size_t nbytes) override {out.write(data, nbytes);}
+    };
+
+    struct istream_source : source_base
+    {
+        std::istream& in;
+
+        istream_source(std::istream& in_) : in{in_}{}
+
+        void read(char* buf, size_t nbytes) override
+        {
+            in.read(buf, nbytes);
+            if (in.gcount() != (long)nbytes)
+                throw std::system_error(OUT_OF_DATA);
+        }
+
+        uint8_t peak() override
+        {
+            uint8_t b = static_cast<uint8_t>(in.peek());
+            if (!in || !in.good() || in.eof())
+                throw std::system_error(OUT_OF_DATA);
+            return b;
+        }
+    };
+
+    inline auto sink(std::ostream& out)  { return ostream_sink{out}; }
+    inline auto source(std::istream& in) { return istream_source{in}; }
+
+//----------------------------------------------------------------------------------------------------------------
+
 }