30 lines
513 B
C++
30 lines
513 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
|