wbtiger
/
foundationdb
mirror of https://github.com/apple/foundationdb.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'andrew/flatbuffers-fixes2' into flatbuffers-fixes2

This commit is contained in:
mpilman 2019-07-11 23:06:13 -07:00
parent 74f992b2cc c96854fce2
commit 4870362dbb
2 changed files with 10 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fdbclient/HTTP.actor.cpp
View File

@ -334,7 +334,7 @@ namespace HTTP {
}
// Write headers to a packet buffer chain
PacketBuffer* pFirst = PacketBuffer::create();
PacketBuffer* pFirst = PacketBuffer::create();
PacketBuffer *pLast = writeRequestHeader(verb, resource, headers, pFirst);
// Prepend headers to content packer buffer chain
pContent->prependWriteBuffer(pFirst, pLast);

47
fdbrpc/ReplicationPolicy.h
View File

@ -70,13 +70,6 @@ struct IReplicationPolicy : public ReferenceCounted<IReplicationPolicy> {
return keys;
}
virtual void attributeKeys(std::set<std::string>*) const = 0;
// For flatbuffers, IReplicationPolicy is just encoded as a string using
// |serializeReplicationPolicy|. |writer| is a member of IReplicationPolicy
// so that this string outlives all calls to
// dynamic_size_traits<Reference<IReplicationPolicy>>::save
mutable BinaryWriter writer{ IncludeVersion() };
mutable bool alreadyWritten = false;
};
template <class Archive>
@ -288,34 +281,18 @@ private:
public:
static size_t size(const T& value) {
bool present = true;
if (value.getPtr() == nullptr) {
present = false;
BinaryWriter writer{ IncludeVersion() };
writer << present;
return writer.getLength();
}
if (!value->alreadyWritten) {
value->writer = BinaryWriter{ IncludeVersion() };
value->writer << present;
serializeReplicationPolicy(value->writer, const_cast<Reference<IReplicationPolicy>&>(value));
value->alreadyWritten = true;
}
return value->writer.getLength();
// size gets called multiple times. If this becomes a performance problem, we can perform the
// serialization once and cache the result as a mutable member of IReplicationPolicy
BinaryWriter writer{ IncludeVersion() };
::save(writer, value);
return writer.getLength();
}
// Guaranteed to be called only once during serialization
static void save(uint8_t* out, const T& value) {
if (value.getPtr() == nullptr) {
bool present = false;
BinaryWriter writer{ IncludeVersion() };
writer << present;
memcpy(out, writer.getData(), writer.getLength());
} else {
ASSERT(value->alreadyWritten)
memcpy(out, value->writer.getData(), value->writer.getLength());
value->alreadyWritten = false;
}
BinaryWriter writer{ IncludeVersion() };
::save(writer, value);
memcpy(out, writer.getData(), writer.getLength());
}
// Context is an arbitrary type that is plumbed by reference throughout the
@ -324,13 +301,7 @@ public:
static void load(const uint8_t* buf, size_t sz, Reference<IReplicationPolicy>& value, Context&) {
StringRef str(buf, sz);
BinaryReader reader(str, IncludeVersion());
bool present = false;
reader >> present;
if (present) {
serializeReplicationPolicy(reader, value);
} else {
value.clear();
}
::load(reader, value);
}
};