Create a new TeeSink abstraction

obsidiansystems · Jun 17, 2020 · 270005f · 270005f
1 parent 09e6e57
commit 270005f
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/libutil/serialise.hh b/src/libutil/serialise.hh
@@ -181,6 +181,24 @@ struct TeeSource : Source
     }
 };
 
+struct TeeSink : Sink
+{
+    Sink & orig;
+    ref<std::string> data;
+    TeeSink(Sink & orig)
+        : orig(orig), data(make_ref<std::string>()) { }
+    void operator () (const unsigned char * data, size_t len) {
+        this->data->append((const char *) data, len);
+        Sink::operator()(data, len);
+    }
+    void operator () (const std::string & s)
+    {
+        *data += s;
+        Sink::operator()(s);
+    }
+};
+
+
 /* A reader that consumes the original Source until 'size'. */
 struct SizedSource : Source
 {