ma27/hydra
0
0
Fork 0
forked from lix-project/hydra
Code Pull requests Activity

hydra-queue-runner/nar-extractor: address review comments

Browse source 
* Turn `NarMemberData` in `MyFileHandle` into a reference.
* `expectedSize` now uses an initializer.
* Used `std::make_unique` instead of `std::unique_ptr(new ...)` as
  suggested by clangd.

(cherry picked from commit 06a30ec8df)
This commit is contained in:
Maximilian Bosch 2024-10-06 19:43:34 +02:00
parent 413029892b
commit 01997d5445
Signed by: ma27
SSH key fingerprint: SHA256:d7dmwHmpai66L6KIXA+wxzVbkPq0nGLrcHK3ZNroqZY
1 changed files with 10 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/hydra-queue-runner/nar-extractor.cc
View file

@ -11,29 +11,28 @@ struct Extractor : NARParseVisitor
{
class MyFileHandle : public FileHandle
{
NarMemberData * memberData;
NarMemberData & memberData;
uint64_t expectedSize;
std::unique_ptr<HashSink> hashSink;
public:
MyFileHandle(NarMemberData * memberData, uint64_t size) : memberData(memberData)
MyFileHandle(NarMemberData & memberData, uint64_t size) : memberData(memberData), expectedSize(size)
{
expectedSize = size;
hashSink = std::make_unique<HashSink>(HashType::SHA256);
}
void receiveContents(std::string_view data) override
{
*memberData->fileSize += data.size();
*memberData.fileSize += data.size();
(*hashSink)(data);
if (memberData->contents) {
memberData->contents->append(data);
if (memberData.contents) {
memberData.contents->append(data);
}
assert(memberData->fileSize <= expectedSize);
if (memberData->fileSize == expectedSize) {
assert(memberData.fileSize <= expectedSize);
if (memberData.fileSize == expectedSize) {
auto [hash, len] = hashSink->finish();
assert(memberData->fileSize == len);
memberData->sha256 = hash;
assert(memberData.fileSize == len);
memberData.sha256 = hash;
hashSink.reset();
}
}
@ -65,7 +64,7 @@ struct Extractor : NARParseVisitor
.contents = filesToKeep.count(path) ? std::optional("") : std::nullopt,
}).first->second;
return std::unique_ptr<MyFileHandle>(new MyFileHandle{memberData, size});
return std::make_unique<MyFileHandle>(*memberData, size);
}
void createSymlink(const Path & path, const std::string & target) override