bool ServerContext::AddAttachment(const std::string& resourceId,
FileContentType attachmentType,
const void* data,
size_t size)
{
LOG(INFO) << "Adding attachment " << EnumerationToString(attachmentType) << " to resource " << resourceId;
if (compressionEnabled_)
{
accessor_.SetCompressionForNextOperations(CompressionType_Zlib);
}
else
{
accessor_.SetCompressionForNextOperations(CompressionType_None);
}
FileInfo info = accessor_.Write(data, size, attachmentType);
StoreStatus status = index_.AddAttachment(info, resourceId);
if (status != StoreStatus_Success)
{
accessor_.Remove(info.GetUuid(), info.GetContentType());
return false;
}
else
{
return true;
}
}
void DatabaseWrapperBase::AddAttachment(int64_t id,
const FileInfo& attachment)
{
SQLite::Statement s(db_, SQLITE_FROM_HERE, "INSERT INTO AttachedFiles VALUES(?, ?, ?, ?, ?, ?, ?, ?)");
s.BindInt64(0, id);
s.BindInt(1, attachment.GetContentType());
s.BindString(2, attachment.GetUuid());
s.BindInt64(3, attachment.GetCompressedSize());
s.BindInt64(4, attachment.GetUncompressedSize());
s.BindInt(5, attachment.GetCompressionType());
s.BindString(6, attachment.GetUncompressedMD5());
s.BindString(7, attachment.GetCompressedMD5());
s.Run();
}
TEST(StorageAccessor, Compression)
{
FilesystemStorage s("UnitTestsStorage");
StorageAccessor accessor(s);
std::string data = "Hello world";
FileInfo info = accessor.Write(data, FileContentType_DicomAsJson, CompressionType_ZlibWithSize, true);
std::string r;
accessor.Read(r, info);
ASSERT_EQ(data, r);
ASSERT_EQ(CompressionType_ZlibWithSize, info.GetCompressionType());
ASSERT_EQ(11u, info.GetUncompressedSize());
ASSERT_EQ(FileContentType_DicomAsJson, info.GetContentType());
ASSERT_EQ("3e25960a79dbc69b674cd4ec67a72c62", info.GetUncompressedMD5());
ASSERT_NE(info.GetUncompressedMD5(), info.GetCompressedMD5());
}
void ServerContext::ReadFile(std::string& result,
const std::string& instancePublicId,
FileContentType content,
bool uncompressIfNeeded)
{
FileInfo attachment;
if (!index_.LookupAttachment(attachment, instancePublicId, content))
{
throw OrthancException(ErrorCode_InternalError);
}
if (uncompressIfNeeded)
{
accessor_.SetCompressionForNextOperations(attachment.GetCompressionType());
}
else
{
accessor_.SetCompressionForNextOperations(CompressionType_None);
}
accessor_.Read(result, attachment.GetUuid(), attachment.GetContentType());
}
void ServerContext::AnswerAttachment(RestApiOutput& output,
const std::string& instancePublicId,
FileContentType content)
{
FileInfo attachment;
if (!index_.LookupAttachment(attachment, instancePublicId, content))
{
throw OrthancException(ErrorCode_InternalError);
}
accessor_.SetCompressionForNextOperations(attachment.GetCompressionType());
std::auto_ptr<HttpFileSender> sender(accessor_.ConstructHttpFileSender(attachment.GetUuid(), attachment.GetContentType()));
sender->SetContentType(GetMimeType(content));
sender->SetDownloadFilename(instancePublicId + ".dcm");
output.AnswerFile(*sender);
}