lordgamez commented on a change in pull request #1221:
URL: https://github.com/apache/nifi-minifi-cpp/pull/1221#discussion_r783098509
##########
File path: extensions/azure/storage/AzureDataLakeStorage.cpp
##########
@@ -61,4 +61,29 @@ bool AzureDataLakeStorage::deleteFile(const
storage::DeleteAzureDataLakeStorageP
}
}
+std::optional<uint64_t> AzureDataLakeStorage::fetchFile(const
FetchAzureDataLakeStorageParameters& params, io::BaseStream& stream) {
+ try {
+ auto fetch_res = data_lake_storage_client_->fetchFile(params);
+
+ std::array<uint8_t, 4096> buffer;
+ size_t write_size = 0;
+ if (fetch_res.FileSize < 0) return 0;
+ while (write_size < gsl::narrow<uint64_t>(fetch_res.FileSize)) {
+ const auto next_write_size =
(std::min)(gsl::narrow<size_t>(fetch_res.FileSize) - write_size, buffer.size());
+ if (!fetch_res.Body->Read(buffer.data(),
gsl::narrow<std::streamsize>(next_write_size))) {
+ return -1;
+ }
+ const auto ret = stream.write(buffer.data(), next_write_size);
+ if (io::isError(ret)) {
+ return -1;
+ }
+ write_size += next_write_size;
+ }
Review comment:
I see your point, updated in 0f01db34bb1cc4576c7ed709dd1b9ef71801b43d
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
