28 lines
534 B
C++
28 lines
534 B
C++
#include "stream_buffer.h"
|
|
|
|
namespace firebuf {
|
|
|
|
StreamBuffer::StreamBuffer(int sock, size_t size) : Buffer(size), sock_(sock) {}
|
|
|
|
bool StreamBuffer::Refill() {
|
|
if (WriteMaxLen() == 0) {
|
|
LOG(ERROR) << "buffer full";
|
|
return false;
|
|
}
|
|
auto read_len = read(sock_, WritePtr(), WriteMaxLen());
|
|
if (read_len == -1) {
|
|
if (errno == EINTR) {
|
|
return true;
|
|
}
|
|
PLOG(ERROR) << "read()";
|
|
return false;
|
|
}
|
|
if (read_len == 0) {
|
|
return false;
|
|
}
|
|
Wrote(read_len);
|
|
return true;
|
|
}
|
|
|
|
} // namespace firebuf
|