Mercurial > hg > orthanc
annotate Core/FileStorage/CompressedFileStorageAccessor.cpp @ 799:777b6b694da6
rename folder
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 06 May 2014 12:55:41 +0200 |
parents | 01d8611c4a60 |
children | a811bdf8b8eb |
rev | line source |
---|---|
221 | 1 /** |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
689 | 3 * Copyright (C) 2012-2014 Medical Physics Department, CHU of Liege, |
221 | 4 * Belgium |
5 * | |
6 * This program is free software: you can redistribute it and/or | |
7 * modify it under the terms of the GNU General Public License as | |
8 * published by the Free Software Foundation, either version 3 of the | |
9 * License, or (at your option) any later version. | |
10 * | |
11 * In addition, as a special exception, the copyright holders of this | |
12 * program give permission to link the code of its release with the | |
13 * OpenSSL project's "OpenSSL" library (or with modified versions of it | |
14 * that use the same license as the "OpenSSL" library), and distribute | |
15 * the linked executables. You must obey the GNU General Public License | |
16 * in all respects for all of the code used other than "OpenSSL". If you | |
17 * modify file(s) with this exception, you may extend this exception to | |
18 * your version of the file(s), but you are not obligated to do so. If | |
19 * you do not wish to do so, delete this exception statement from your | |
20 * version. If you delete this exception statement from all source files | |
21 * in the program, then also delete it here. | |
22 * | |
23 * This program is distributed in the hope that it will be useful, but | |
24 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
26 * General Public License for more details. | |
27 * | |
28 * You should have received a copy of the GNU General Public License | |
29 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
30 **/ | |
31 | |
32 | |
33 #include "CompressedFileStorageAccessor.h" | |
34 | |
35 #include "../OrthancException.h" | |
36 #include "FileStorageAccessor.h" | |
37 #include "../HttpServer/BufferHttpSender.h" | |
38 | |
39 namespace Orthanc | |
40 { | |
232
5368bbe813cf
refactoring of attachments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
221
diff
changeset
|
41 FileInfo CompressedFileStorageAccessor::WriteInternal(const void* data, |
5368bbe813cf
refactoring of attachments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
221
diff
changeset
|
42 size_t size, |
233 | 43 FileContentType type) |
221 | 44 { |
693
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
45 std::string md5; |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
46 |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
47 if (storeMD5_) |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
48 { |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
49 Toolbox::ComputeMD5(md5, data, size); |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
50 } |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
51 |
221 | 52 switch (compressionType_) |
53 { | |
54 case CompressionType_None: | |
232
5368bbe813cf
refactoring of attachments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
221
diff
changeset
|
55 { |
5368bbe813cf
refactoring of attachments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
221
diff
changeset
|
56 std::string uuid = storage_.Create(data, size); |
693
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
57 return FileInfo(uuid, type, size, md5); |
232
5368bbe813cf
refactoring of attachments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
221
diff
changeset
|
58 } |
221 | 59 |
60 case CompressionType_Zlib: | |
61 { | |
62 std::string compressed; | |
63 zlib_.Compress(compressed, data, size); | |
693
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
64 |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
65 std::string compressedMD5; |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
66 |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
67 if (storeMD5_) |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
68 { |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
69 Toolbox::ComputeMD5(compressedMD5, compressed); |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
70 } |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
71 |
232
5368bbe813cf
refactoring of attachments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
221
diff
changeset
|
72 std::string uuid = storage_.Create(compressed); |
693
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
73 return FileInfo(uuid, type, size, md5, |
01d8611c4a60
md5 for attached files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
74 CompressionType_Zlib, compressed.size(), compressedMD5); |
221 | 75 } |
76 | |
77 default: | |
78 throw OrthancException(ErrorCode_NotImplemented); | |
79 } | |
80 } | |
81 | |
82 CompressedFileStorageAccessor::CompressedFileStorageAccessor(FileStorage& storage) : | |
83 storage_(storage) | |
84 { | |
232
5368bbe813cf
refactoring of attachments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
221
diff
changeset
|
85 compressionType_ = CompressionType_None; |
221 | 86 } |
87 | |
88 void CompressedFileStorageAccessor::Read(std::string& content, | |
89 const std::string& uuid) | |
90 { | |
91 switch (compressionType_) | |
92 { | |
93 case CompressionType_None: | |
94 storage_.ReadFile(content, uuid); | |
95 break; | |
96 | |
97 case CompressionType_Zlib: | |
98 { | |
99 std::string compressed; | |
100 storage_.ReadFile(compressed, uuid); | |
101 zlib_.Uncompress(content, compressed); | |
102 break; | |
103 } | |
104 | |
105 default: | |
106 throw OrthancException(ErrorCode_NotImplemented); | |
107 } | |
108 } | |
109 | |
110 HttpFileSender* CompressedFileStorageAccessor::ConstructHttpFileSender(const std::string& uuid) | |
111 { | |
112 switch (compressionType_) | |
113 { | |
114 case CompressionType_None: | |
115 { | |
116 FileStorageAccessor uncompressedAccessor(storage_); | |
117 return uncompressedAccessor.ConstructHttpFileSender(uuid); | |
118 } | |
119 | |
120 case CompressionType_Zlib: | |
121 { | |
122 std::string compressed; | |
123 storage_.ReadFile(compressed, uuid); | |
124 | |
125 std::auto_ptr<BufferHttpSender> sender(new BufferHttpSender); | |
126 zlib_.Uncompress(sender->GetBuffer(), compressed); | |
127 | |
128 return sender.release(); | |
129 } | |
130 | |
131 default: | |
132 throw OrthancException(ErrorCode_NotImplemented); | |
133 } | |
134 } | |
135 } |