Refactor network layer to use streams

Summary:
The network layer now has a `Session` that handles all things that should be
done before the `Execute` method is called on sessions. Also, all sessions
now communicate using streams instead of holding the input buffer and writing
to the `Socket`. This design will allow implementation of a SSL middleware.

Reviewers: buda, dgleich

Reviewed By: buda

Subscribers: pullbot

Differential Revision: https://phabricator.memgraph.io/D1314
This commit is contained in:
Matej Ferencevic
2018-03-23 16:32:17 +01:00
parent b2f3bf9709
commit f1a8d7cd3d
27 changed files with 778 additions and 521 deletions

View File

@@ -0,0 +1,71 @@
#include "glog/logging.h"
#include "communication/buffer.hpp"
namespace communication {
Buffer::Buffer()
: data_(kBufferInitialSize, 0), read_end_(*this), write_end_(*this) {}
Buffer::ReadEnd::ReadEnd(Buffer &buffer) : buffer_(buffer) {}
uint8_t *Buffer::ReadEnd::data() { return buffer_.data(); }
size_t Buffer::ReadEnd::size() const { return buffer_.size(); }
void Buffer::ReadEnd::Shift(size_t len) { buffer_.Shift(len); }
void Buffer::ReadEnd::Resize(size_t len) { buffer_.Resize(len); }
void Buffer::ReadEnd::Clear() { buffer_.Clear(); }
Buffer::WriteEnd::WriteEnd(Buffer &buffer) : buffer_(buffer) {}
io::network::StreamBuffer Buffer::WriteEnd::Allocate() {
return buffer_.Allocate();
}
void Buffer::WriteEnd::Written(size_t len) { buffer_.Written(len); }
void Buffer::WriteEnd::Resize(size_t len) { buffer_.Resize(len); }
void Buffer::WriteEnd::Clear() { buffer_.Clear(); }
Buffer::ReadEnd &Buffer::read_end() { return read_end_; }
Buffer::WriteEnd &Buffer::write_end() { return write_end_; }
uint8_t *Buffer::data() { return data_.data(); }
size_t Buffer::size() const { return have_; }
void Buffer::Shift(size_t len) {
DCHECK(len <= have_) << "Tried to shift more data than the buffer has!";
if (len == have_) {
have_ = 0;
} else {
memmove(data_.data(), data_.data() + len, have_ - len);
have_ -= len;
}
}
io::network::StreamBuffer Buffer::Allocate() {
DCHECK(data_.size() > have_) << "The buffer thinks that there is more data "
"in the buffer than there is underlying "
"storage space!";
return {data_.data() + have_, data_.size() - have_};
}
void Buffer::Written(size_t len) {
have_ += len;
DCHECK(have_ <= data_.size()) << "Written more than storage has space!";
}
void Buffer::Resize(size_t len) {
if (len < data_.size()) return;
data_.resize(len, 0);
}
void Buffer::Clear() { have_ = 0; }
} // namespace communication