annotate Orthanc/Core/FileStorage/FilesystemStorage.cpp @ 175:dd1ad819ca33

sync
author Sebastien Jodogne <s.jodogne@gmail.com>
date Thu, 04 Jan 2018 07:34:54 +0100
parents 5dc54316d68b
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
1 /**
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
2 * Orthanc - A Lightweight, RESTful DICOM Store
122
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 78
diff changeset
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
4 * Department, University Hospital of Liege, Belgium
175
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
5 * Copyright (C) 2017-2018 Osimis S.A., Belgium
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
6 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
7 * This program is free software: you can redistribute it and/or
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
8 * modify it under the terms of the GNU General Public License as
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
9 * published by the Free Software Foundation, either version 3 of the
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
10 * License, or (at your option) any later version.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
11 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
12 * In addition, as a special exception, the copyright holders of this
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
13 * program give permission to link the code of its release with the
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
15 * that use the same license as the "OpenSSL" library), and distribute
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
16 * the linked executables. You must obey the GNU General Public License
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
17 * in all respects for all of the code used other than "OpenSSL". If you
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
18 * modify file(s) with this exception, you may extend this exception to
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
19 * your version of the file(s), but you are not obligated to do so. If
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
20 * you do not wish to do so, delete this exception statement from your
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
21 * version. If you delete this exception statement from all source files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
22 * in the program, then also delete it here.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
23 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
24 * This program is distributed in the hope that it will be useful, but
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
25 * WITHOUT ANY WARRANTY; without even the implied warranty of
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
27 * General Public License for more details.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
28 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
29 * You should have received a copy of the GNU General Public License
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
31 **/
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
32
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
33
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
34 #include "../PrecompiledHeaders.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
35 #include "FilesystemStorage.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
36
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
37 // http://stackoverflow.com/questions/1576272/storing-large-number-of-files-in-file-system
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
38 // http://stackoverflow.com/questions/446358/storing-a-large-number-of-images
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
39
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
40 #include "../Logging.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
41 #include "../OrthancException.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
42 #include "../Toolbox.h"
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 145
diff changeset
43 #include "../SystemToolbox.h"
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
44
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
45 #include <boost/filesystem/fstream.hpp>
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
46
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
47
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
48 static std::string ToString(const boost::filesystem::path& p)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
49 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
50 #if BOOST_HAS_FILESYSTEM_V3 == 1
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
51 return p.filename().string();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
52 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
53 return p.filename();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
54 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
55 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
56
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
57
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
58 namespace Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
59 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
60 boost::filesystem::path FilesystemStorage::GetPath(const std::string& uuid) const
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
61 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
62 namespace fs = boost::filesystem;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
63
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
64 if (!Toolbox::IsUuid(uuid))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
65 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
66 throw OrthancException(ErrorCode_ParameterOutOfRange);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
67 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
68
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
69 fs::path path = root_;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
70
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
71 path /= std::string(&uuid[0], &uuid[2]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
72 path /= std::string(&uuid[2], &uuid[4]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
73 path /= uuid;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
74
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
75 #if BOOST_HAS_FILESYSTEM_V3 == 1
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
76 path.make_preferred();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
77 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
79 return path;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
80 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
81
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
82 FilesystemStorage::FilesystemStorage(std::string root)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
83 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
84 //root_ = boost::filesystem::absolute(root).string();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
85 root_ = root;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
86
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 145
diff changeset
87 SystemToolbox::MakeDirectory(root);
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
88 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
89
145
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
90
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
91
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
92 static const char* GetDescriptionInternal(FileContentType content)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
93 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
94 // This function is for logging only (internal use), a more
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
95 // fully-featured version is available in ServerEnumerations.cpp
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
96 switch (content)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
97 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
98 case FileContentType_Unknown:
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
99 return "Unknown";
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
100
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
101 case FileContentType_Dicom:
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
102 return "DICOM";
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
103
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
104 case FileContentType_DicomAsJson:
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
105 return "JSON summary of DICOM";
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
106
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
107 default:
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
108 return "User-defined";
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
109 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
110 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
111
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
112
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
113 void FilesystemStorage::Create(const std::string& uuid,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
114 const void* content,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
115 size_t size,
145
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
116 FileContentType type)
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
117 {
145
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
118 LOG(INFO) << "Creating attachment \"" << uuid << "\" of \"" << GetDescriptionInternal(type)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
119 << "\" type (size: " << (size / (1024 * 1024) + 1) << "MB)";
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
120
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
121 boost::filesystem::path path;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
122
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
123 path = GetPath(uuid);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
124
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
125 if (boost::filesystem::exists(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
126 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
127 // Extremely unlikely case: This Uuid has already been created
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
128 // in the past.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
129 throw OrthancException(ErrorCode_InternalError);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
130 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
131
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
132 if (boost::filesystem::exists(path.parent_path()))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
133 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
134 if (!boost::filesystem::is_directory(path.parent_path()))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
135 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
136 throw OrthancException(ErrorCode_DirectoryOverFile);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
137 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
138 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
139 else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
140 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
141 if (!boost::filesystem::create_directories(path.parent_path()))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
142 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
143 throw OrthancException(ErrorCode_FileStorageCannotWrite);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
144 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
145 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
146
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 145
diff changeset
147 SystemToolbox::WriteFile(content, size, path.string());
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
148 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
149
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
151 void FilesystemStorage::Read(std::string& content,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
152 const std::string& uuid,
145
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
153 FileContentType type)
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
154 {
145
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
155 LOG(INFO) << "Reading attachment \"" << uuid << "\" of \"" << GetDescriptionInternal(type)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
156 << "\" content type";
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
157
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
158 content.clear();
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 145
diff changeset
159 SystemToolbox::ReadFile(content, GetPath(uuid).string());
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
160 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
161
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
162
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
163 uintmax_t FilesystemStorage::GetSize(const std::string& uuid) const
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
164 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
165 boost::filesystem::path path = GetPath(uuid);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
166 return boost::filesystem::file_size(path);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
167 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
168
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
169
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
170
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
171 void FilesystemStorage::ListAllFiles(std::set<std::string>& result) const
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
172 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
173 namespace fs = boost::filesystem;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
174
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
175 result.clear();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
176
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
177 if (fs::exists(root_) && fs::is_directory(root_))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
178 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
179 for (fs::recursive_directory_iterator current(root_), end; current != end ; ++current)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
180 {
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 145
diff changeset
181 if (SystemToolbox::IsRegularFile(current->path().string()))
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
182 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
183 try
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
184 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
185 fs::path d = current->path();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
186 std::string uuid = ToString(d);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
187 if (Toolbox::IsUuid(uuid))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
188 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
189 fs::path p0 = d.parent_path().parent_path().parent_path();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
190 std::string p1 = ToString(d.parent_path().parent_path());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
191 std::string p2 = ToString(d.parent_path());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
192 if (p1.length() == 2 &&
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
193 p2.length() == 2 &&
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
194 p1 == uuid.substr(0, 2) &&
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
195 p2 == uuid.substr(2, 2) &&
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
196 p0 == root_)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
197 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
198 result.insert(uuid);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
199 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
200 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
201 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
202 catch (fs::filesystem_error)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
203 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
204 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
205 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
206 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
207 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
208 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
209
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
210
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
211 void FilesystemStorage::Clear()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
212 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
213 namespace fs = boost::filesystem;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
214 typedef std::set<std::string> List;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
215
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
216 List result;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
217 ListAllFiles(result);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
218
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
219 for (List::const_iterator it = result.begin(); it != result.end(); ++it)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
220 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
221 Remove(*it, FileContentType_Unknown /*ignored in this class*/);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
222 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
223 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
224
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
225
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
226 void FilesystemStorage::Remove(const std::string& uuid,
145
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
227 FileContentType type)
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
228 {
145
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 125
diff changeset
229 LOG(INFO) << "Deleting attachment \"" << uuid << "\" of type " << static_cast<int>(type);
78
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
230
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
231 namespace fs = boost::filesystem;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
232
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
233 fs::path p = GetPath(uuid);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
234
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
235 try
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
236 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
237 fs::remove(p);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
238 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
239 catch (...)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
240 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
241 // Ignore the error
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
242 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
243
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
244 // Remove the two parent directories, ignoring the error code if
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
245 // these directories are not empty
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
246
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
247 try
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
248 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
249 #if BOOST_HAS_FILESYSTEM_V3 == 1
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
250 boost::system::error_code err;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
251 fs::remove(p.parent_path(), err);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
252 fs::remove(p.parent_path().parent_path(), err);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
253 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
254 fs::remove(p.parent_path());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
255 fs::remove(p.parent_path().parent_path());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
256 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
257 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
258 catch (...)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
259 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
260 // Ignore the error
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
261 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
262 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
263
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
264
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
265 uintmax_t FilesystemStorage::GetCapacity() const
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
266 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
267 return boost::filesystem::space(root_).capacity;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
268 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
269
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
270 uintmax_t FilesystemStorage::GetAvailableSpace() const
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
271 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
272 return boost::filesystem::space(root_).available;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
273 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
274 }