]>
Commit | Line | Data |
---|---|---|
c21ad0f5 | 1 | |
2 | /* | |
1ca48d39 | 3 | * $Id: StoreEntryStream.h,v 1.5 2007/08/14 10:09:23 serassio Exp $ |
c21ad0f5 | 4 | * |
5 | * | |
6 | * SQUID Web Proxy Cache http://www.squid-cache.org/ | |
7 | * ---------------------------------------------------------- | |
8 | * | |
9 | * Squid is the result of efforts by numerous individuals from | |
10 | * the Internet community; see the CONTRIBUTORS file for full | |
11 | * details. Many organizations have provided support for Squid's | |
12 | * development; see the SPONSORS file for full details. Squid is | |
13 | * Copyrighted (C) 2001 by the Regents of the University of | |
14 | * California; see the COPYRIGHT file for full details. Squid | |
15 | * incorporates software developed and/or copyrighted by other | |
16 | * sources; see the CREDITS file for full details. | |
17 | * | |
18 | * This program is free software; you can redistribute it and/or modify | |
19 | * it under the terms of the GNU General Public License as published by | |
20 | * the Free Software Foundation; either version 2 of the License, or | |
21 | * (at your option) any later version. | |
22 | * | |
23 | * This program is distributed in the hope that it will be useful, | |
24 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
25 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
26 | * GNU 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, write to the Free Software | |
30 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA. | |
31 | * | |
32 | */ | |
33 | ||
34 | #ifndef SQUID_STORE_ENTRY_STREAM_H | |
35 | #define SQUID_STORE_ENTRY_STREAM_H | |
36 | ||
37 | #include "Store.h" | |
38 | ||
39 | #include <ostream> | |
40 | ||
41 | /* | |
42 | * This class provides a streambuf interface for writing | |
43 | * to StoreEntries. Typical use is via a StoreEntryStream | |
44 | * rather than direct manipulation | |
45 | */ | |
46 | ||
47 | class StoreEntryStreamBuf : public std::streambuf | |
48 | { | |
49 | ||
50 | public: | |
4e87141f | 51 | StoreEntryStreamBuf(StoreEntry *anEntry) : theEntry(anEntry) |
c21ad0f5 | 52 | { |
53 | ||
4e87141f | 54 | theEntry->lock(); |
55 | theEntry->buffer(); | |
c21ad0f5 | 56 | } |
57 | ||
58 | ~StoreEntryStreamBuf() | |
59 | { | |
4e87141f | 60 | theEntry->unlock(); |
c21ad0f5 | 61 | } |
62 | ||
63 | protected: | |
64 | /* flush the current buffer and the character that is overflowing | |
65 | * to the store entry. | |
66 | */ | |
480f366a | 67 | virtual int_type overflow(int_type aChar = traits_type::eof()) |
c21ad0f5 | 68 | { |
69 | std::streamsize pending(pptr() - pbase()); | |
70 | ||
71 | if (pending && sync ()) | |
72 | return traits_type::eof(); | |
73 | ||
74 | if (aChar != traits_type::eof()) { | |
75 | char chars[1] = {aChar}; | |
76 | ||
77 | if (aChar != traits_type::eof()) | |
4e87141f | 78 | theEntry->append(chars, 1); |
c21ad0f5 | 79 | } |
80 | ||
81 | pbump (-pending); // Reset pptr(). | |
82 | return aChar; | |
83 | } | |
84 | ||
85 | /* push the buffer to the store */ | |
86 | virtual int sync() | |
87 | { | |
88 | std::streamsize pending(pptr() - pbase()); | |
89 | ||
90 | if (pending) | |
4e87141f | 91 | theEntry->append(pbase(), pending); |
c21ad0f5 | 92 | |
4e87141f | 93 | theEntry->flush(); |
c21ad0f5 | 94 | |
95 | return 0; | |
96 | } | |
97 | ||
98 | /* write multiple characters to the store entry | |
99 | * - this is an optimisation method. | |
100 | */ | |
101 | virtual std::streamsize xsputn(const char * chars, std::streamsize number) | |
102 | { | |
103 | if (number) | |
4e87141f | 104 | theEntry->append(chars, number); |
c21ad0f5 | 105 | |
106 | return number; | |
107 | } | |
108 | ||
109 | private: | |
4e87141f | 110 | StoreEntry *theEntry; |
c21ad0f5 | 111 | |
112 | }; | |
113 | ||
114 | class StoreEntryStream : public std::ostream | |
115 | { | |
116 | ||
117 | public: | |
4e87141f | 118 | /* create a stream for writing text etc into theEntry */ |
1ca48d39 | 119 | // See http://www.codecomments.com/archive292-2005-2-396222.html |
120 | StoreEntryStream(StoreEntry *entry): std::ostream(0), theBuffer(entry) { | |
121 | rdbuf(&theBuffer); // set the buffer to now-initialized theBuffer | |
122 | clear(); //clear badbit set by calling init(0) | |
123 | } | |
c21ad0f5 | 124 | |
125 | private: | |
4e87141f | 126 | StoreEntryStreamBuf theBuffer; |
c21ad0f5 | 127 | }; |
128 | ||
129 | #endif /* SQUID_STORE_ENTRY_STREAM_H */ |