Change API of raw_istream to be more similar to raw_ostream.

This commit is contained in:
Peter Johnson
2016-09-02 20:53:45 -07:00
parent 94c2b65798
commit c2ae897b02
6 changed files with 43 additions and 17 deletions

View File

@@ -16,21 +16,37 @@ class raw_istream {
public:
raw_istream() = default;
virtual ~raw_istream() = default;
virtual bool read(void* data, std::size_t len) = 0;
raw_istream& read(void* data, std::size_t len) {
read_impl(data, len);
return *this;
}
virtual void close() = 0;
bool has_error() const { return m_error; }
void clear_error() { m_error = false; }
raw_istream(const raw_istream&) = delete;
raw_istream& operator=(const raw_istream&) = delete;
protected:
void error_detected() { m_error = true; }
private:
virtual void read_impl(void* data, std::size_t len) = 0;
bool m_error = false;
};
class raw_mem_istream : public raw_istream {
public:
raw_mem_istream(const char* mem, std::size_t len) : m_cur(mem), m_left(len) {}
virtual ~raw_mem_istream() = default;
virtual bool read(void* data, std::size_t len);
virtual void close() {}
void close() override;
private:
void read_impl(void* data, std::size_t len) override;
const char* m_cur;
std::size_t m_left;
};

View File

@@ -9,19 +9,21 @@
#define WPIUTIL_SUPPORT_RAW_SOCKET_ISTREAM_H_
#include "support/raw_istream.h"
#include "tcpsockets/NetworkStream.h"
namespace wpi {
class NetworkStream;
class raw_socket_istream : public raw_istream {
public:
raw_socket_istream(NetworkStream& stream, int timeout = 0)
: m_stream(stream), m_timeout(timeout) {}
virtual ~raw_socket_istream() = default;
virtual bool read(void* data, std::size_t len);
virtual void close();
void close() override;
private:
void read_impl(void* data, std::size_t len) override;
NetworkStream& m_stream;
int m_timeout;
};