Increase writer interface

wichtounet · Oct 16, 2023 · a411a40 · a411a40
1 parent 1c9f82a
commit a411a40
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/include/writer.hpp b/include/writer.hpp
@@ -87,6 +87,10 @@ struct writer {
 
     virtual writer& operator<<(std::string_view value) = 0;
     virtual writer& operator<<(double value) = 0;
+    virtual writer& operator<<(size_t value) = 0;
+    virtual writer& operator<<(long value) = 0;
+    virtual writer& operator<<(int value) = 0;
+    virtual writer& operator<<(unsigned value) = 0;
 
     virtual writer& operator<<(const budget::money& m) = 0;
     virtual writer& operator<<(const budget::year& y) = 0;
@@ -142,6 +146,10 @@ struct console_writer : writer {
 
     writer& operator<<(std::string_view value) override;
     writer& operator<<(double value) override;
+    writer& operator<<(size_t value) override;
+    writer& operator<<(long value) override;
+    writer& operator<<(int value) override;
+    writer& operator<<(unsigned value) override;
 
     writer& operator<<(const budget::money& m) override;
     writer& operator<<(const budget::year& y) override;

diff --git a/src/console_writer.cpp b/src/console_writer.cpp
@@ -111,7 +111,22 @@ budget::writer& budget::console_writer::operator<<(std::string_view value) {
 
 budget::writer& budget::console_writer::operator<<(double value) {
     os << value;
-
+    return *this;
+}
+budget::writer& budget::console_writer::operator<<(size_t value) {
+    os << value;
+    return *this;
+}
+budget::writer& budget::console_writer::operator<<(long value) {
+    os << value;
+    return *this;
+}
+budget::writer& budget::console_writer::operator<<(int value) {
+    os << value;
+    return *this;
+}
+budget::writer& budget::console_writer::operator<<(unsigned value) {
+    os << value;
     return *this;
 }