summaryrefslogtreecommitdiff
path: root/lib/common/SelfFlushingStream.h
blob: 36e9a4d359837896e197edc4ebd8226f2e3b2edd (plain)
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
// --------------------------------------------------------------------------
//
// File
//		Name:    SelfFlushingStream.h
//		Purpose: A stream wrapper that always flushes the underlying
//			 stream, to ensure protocol safety.
//		Created: 2008/08/20
//
// --------------------------------------------------------------------------

#ifndef SELFFLUSHINGSTREAM__H
#define SELFFLUSHINGSTREAM__H

#include "IOStream.h"

// --------------------------------------------------------------------------
//
// Class
//		Name:    SelfFlushingStream
//		Purpose: A stream wrapper that always flushes the underlying
//			 stream, to ensure protocol safety.
//		Created: 2008/08/20
//
// --------------------------------------------------------------------------
class SelfFlushingStream : public IOStream
{
public:
	SelfFlushingStream(IOStream &rSource)
	: mrSource(rSource) { }

	SelfFlushingStream(const SelfFlushingStream &rToCopy)
	: mrSource(rToCopy.mrSource) { }
	
	~SelfFlushingStream()
	{
		Flush();
	}

private:
	// no copying from IOStream allowed
	SelfFlushingStream(const IOStream& rToCopy);
	
public:
	virtual int Read(void *pBuffer, int NBytes,
		int Timeout = IOStream::TimeOutInfinite)
	{
		return mrSource.Read(pBuffer, NBytes, Timeout);
	}
	virtual pos_type BytesLeftToRead()
	{
		return mrSource.BytesLeftToRead();
	}
	virtual void Write(const void *pBuffer, int NBytes)
	{
		mrSource.Write(pBuffer, NBytes);
	}
	virtual bool StreamDataLeft()
	{
		return mrSource.StreamDataLeft();
	}
	virtual bool StreamClosed()
	{
		return mrSource.StreamClosed();
	}

private:
	IOStream &mrSource;
};

#endif // SELFFLUSHINGSTREAM__H