Mercurial > hg > orthanc-transfers
diff Framework/PushMode/BucketPushQuery.cpp @ 0:95226b754d9e
initial release
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Mon, 17 Sep 2018 11:34:55 +0200 |
parents | |
children | 7e207ade2f1a |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Framework/PushMode/BucketPushQuery.cpp Mon Sep 17 11:34:55 2018 +0200 @@ -0,0 +1,84 @@ +/** + * Transfers accelerator plugin for Orthanc + * Copyright (C) 2018 Osimis, Belgium + * + * This program is free software: you can redistribute it and/or + * modify it under the terms of the GNU Affero General Public License + * as published by the Free Software Foundation, either version 3 of + * the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + **/ + + +#include "BucketPushQuery.h" + +#include <Core/ChunkedBuffer.h> +#include <Core/Compression/GzipCompressor.h> + +#include <boost/lexical_cast.hpp> + + +namespace OrthancPlugins +{ + BucketPushQuery::BucketPushQuery(OrthancInstancesCache& cache, + const TransferBucket& bucket, + const std::string& peer, + const std::string& transactionUri, + size_t bucketIndex, + BucketCompression compression) : + cache_(cache), + bucket_(bucket), + peer_(peer), + compression_(compression) + { + uri_ = transactionUri + "/" + boost::lexical_cast<std::string>(bucketIndex); + } + + + void BucketPushQuery::ReadBody(std::string& body) const + { + Orthanc::ChunkedBuffer buffer; + + for (size_t j = 0; j < bucket_.GetChunksCount(); j++) + { + std::string chunk; + std::string md5; // unused + cache_.GetChunk(chunk, md5, bucket_, j); + buffer.AddChunk(chunk); + } + + switch (compression_) + { + case BucketCompression_None: + buffer.Flatten(body); + break; + + case BucketCompression_Gzip: + { + std::string raw; + buffer.Flatten(raw); + + Orthanc::GzipCompressor compressor; + Orthanc::IBufferCompressor::Compress(body, compressor, raw); + break; + } + + default: + throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); + } + } + + + void BucketPushQuery::HandleAnswer(const void* answer, + size_t size) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); + } +}