-
Notifications
You must be signed in to change notification settings - Fork 378
/
ResponseLog.cc
99 lines (78 loc) · 2.29 KB
/
ResponseLog.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*
grive: an GPL program to sync a local directory with Google Drive
Copyright (C) 2012 Wan Wai Ho
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation version 2
of the License.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include "ResponseLog.hh"
#include "util/log/Log.hh"
#include "util/DateTime.hh"
#include <cassert>
namespace gr { namespace http {
ResponseLog::ResponseLog(
const std::string& prefix,
const std::string& suffix,
DataStream *next ) :
m_enabled ( true ),
m_next ( next )
{
Reset( prefix, suffix, next ) ;
}
ResponseLog::ResponseLog( DataStream *next ) :
m_enabled ( false ),
m_next ( next )
{
assert( m_next != 0 ) ;
}
std::size_t ResponseLog::Write( const char *data, std::size_t count )
{
if ( m_enabled )
{
assert( m_log.rdbuf() != 0 ) ;
m_log.rdbuf()->sputn( data, count ) ;
}
return m_next->Write( data, count ) ;
}
std::size_t ResponseLog::Read( char *data, std::size_t count )
{
assert( m_next != 0 ) ;
return m_next->Read( data, count ) ;
}
std::string ResponseLog::Filename( const std::string& prefix, const std::string& suffix )
{
return prefix + DateTime::Now().Format( "%F.%H%M%S" ) + suffix ;
}
void ResponseLog::Enable( bool enable )
{
m_enabled = enable ;
}
void ResponseLog::Reset( const std::string& prefix, const std::string& suffix, DataStream *next )
{
assert( next != 0 ) ;
if ( m_log.is_open() )
m_log.close() ;
const std::string fname = Filename(prefix, suffix) ;
// reset previous stream state. don't care if file can be opened
// successfully previously
m_log.clear() ;
// re-open the file
m_log.open( fname.c_str() ) ;
if ( m_log )
{
Trace( "logging HTTP response: %1%", fname ) ;
m_enabled = true ;
}
else
Trace( "cannot open log file %1%", fname ) ;
m_next = next ;
}
}} // end of namespace