Skip to content

Commit

Permalink
Make logging respect stream flushes
Browse files Browse the repository at this point in the history
also add override keyword and fix overflow() behavior
  • Loading branch information
sfan5 committed Oct 10, 2024
1 parent d7a2b23 commit 0adb43b
Showing 1 changed file with 22 additions and 15 deletions.
37 changes: 22 additions & 15 deletions src/util/stream.h
Original file line number Diff line number Diff line change
Expand Up @@ -23,48 +23,55 @@ with this program; if not, write to the Free Software Foundation, Inc.,
#include <string_view>
#include <functional>

template<int BufferLength, typename Emitter = std::function<void(std::string_view)> >
template<unsigned int BufferLength, typename Emitter = std::function<void(std::string_view)> >
class StringStreamBuffer : public std::streambuf {
public:
StringStreamBuffer(Emitter emitter) : m_emitter(emitter) {
buffer_index = 0;
}

int overflow(int c) {
push_back(c);
return c;
int overflow(int c) override {
if (c != traits_type::eof())
push_back(c);
return 0;
}

void push_back(char c) {
if (c == '\n' || c == '\r') {
if (buffer_index)
m_emitter(std::string_view(buffer, buffer_index));
buffer_index = 0;
// emit only complete lines, or if the buffer is full
if (c == '\n') {
sync();
} else {
buffer[buffer_index++] = c;
if (buffer_index >= BufferLength) {
m_emitter(std::string_view(buffer, buffer_index));
buffer_index = 0;
sync();
}
}
}

std::streamsize xsputn(const char *s, std::streamsize n) {
std::streamsize xsputn(const char *s, std::streamsize n) override {
for (std::streamsize i = 0; i < n; ++i)
push_back(s[i]);
return n;
}

int sync() override {
if (buffer_index)
m_emitter(std::string_view(buffer, buffer_index));
buffer_index = 0;
return 0;
}

private:
Emitter m_emitter;
unsigned int buffer_index;
char buffer[BufferLength];
int buffer_index;
};

class DummyStreamBuffer : public std::streambuf {
int overflow(int c) {
return c;
int overflow(int c) override {
return 0;
}
std::streamsize xsputn(const char *s, std::streamsize n) {
std::streamsize xsputn(const char *s, std::streamsize n) override {
return n;
}
};

0 comments on commit 0adb43b

Please sign in to comment.