comparison Core/FileBuffer.cpp @ 3357:c0aa5f1cf2f5

new class: FileBuffer
author Sebastien Jodogne <s.jodogne@gmail.com>
date Thu, 02 May 2019 09:22:36 +0200
parents
children 94f4a18a79cc
comparison
equal deleted inserted replaced
3356:f744730c294b 3357:c0aa5f1cf2f5
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
4 * Department, University Hospital of Liege, Belgium
5 * Copyright (C) 2017-2019 Osimis S.A., Belgium
6 *
7 * This program is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License as
9 * published by the Free Software Foundation, either version 3 of the
10 * License, or (at your option) any later version.
11 *
12 * In addition, as a special exception, the copyright holders of this
13 * program give permission to link the code of its release with the
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it
15 * that use the same license as the "OpenSSL" library), and distribute
16 * the linked executables. You must obey the GNU General Public License
17 * in all respects for all of the code used other than "OpenSSL". If you
18 * modify file(s) with this exception, you may extend this exception to
19 * your version of the file(s), but you are not obligated to do so. If
20 * you do not wish to do so, delete this exception statement from your
21 * version. If you delete this exception statement from all source files
22 * in the program, then also delete it here.
23 *
24 * This program is distributed in the hope that it will be useful, but
25 * WITHOUT ANY WARRANTY; without even the implied warranty of
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
27 * General Public License for more details.
28 *
29 * You should have received a copy of the GNU General Public License
30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
31 **/
32
33
34 #include "PrecompiledHeaders.h"
35 #include "FileBuffer.h"
36
37 #include "TemporaryFile.h"
38 #include "OrthancException.h"
39
40 #include <boost/filesystem/fstream.hpp>
41
42
43 namespace Orthanc
44 {
45 class FileBuffer::PImpl
46 {
47 private:
48 TemporaryFile file_;
49 boost::filesystem::ofstream stream_;
50 bool isWriting_;
51
52 public:
53 PImpl() :
54 isWriting_(true)
55 {
56 stream_.open(file_.GetPath(), std::ofstream::out | std::ofstream::binary);
57 if (!stream_.good())
58 {
59 throw OrthancException(ErrorCode_CannotWriteFile);
60 }
61 }
62
63 ~PImpl()
64 {
65 if (isWriting_)
66 {
67 stream_.close();
68 }
69 }
70
71 void Append(const char* buffer,
72 size_t size)
73 {
74 if (!isWriting_)
75 {
76 throw OrthancException(ErrorCode_BadSequenceOfCalls);
77 }
78
79 if (size > 0)
80 {
81 stream_.write(buffer, size);
82 if (!stream_.good())
83 {
84 stream_.close();
85 throw OrthancException(ErrorCode_FileStorageCannotWrite);
86 }
87 }
88 }
89
90 void Read(std::string& target)
91 {
92 if (isWriting_)
93 {
94 stream_.close();
95 isWriting_ = false;
96 }
97
98 file_.Read(target);
99 }
100 };
101
102
103 FileBuffer::FileBuffer() :
104 pimpl_(new PImpl)
105 {
106 }
107
108
109 void FileBuffer::Append(const char* buffer,
110 size_t size)
111 {
112 assert(pimpl_.get() != NULL);
113 pimpl_->Append(buffer, size);
114 }
115
116
117 void FileBuffer::Read(std::string& target)
118 {
119 assert(pimpl_.get() != NULL);
120 pimpl_->Read(target);
121 }
122 }