blob: a95b34879c614fb068cc3bc8a968d61d4837b534 (
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
#pragma once
#include "../Buffer.h"
#include "../Exception.h"
#include "Stream.h"
#include <condition_variable>
#include <list>
#include <mutex>
namespace cru::io {
class WriteAfterEofException : public Exception {
public:
using Exception::Exception;
~WriteAfterEofException() override = default;
};
struct BufferStreamOptions {
/**
* @brief The size of a single buffer allocated each time new space is needed.
* Use default value if <= 0.
*
* When current buffer is full and there is no space for following data, a new
* buffer will be allocated and appended to the buffer list. Note if sum size
* of all buffers reaches the total_buffer_limit, no more buffer will be
* allocated but wait.
*/
Index block_size = 0;
/**
* @brief Total size limit of saved data in buffer. Use default value if < 0.
* No limit if == 0.
*
* The size will be ceil(total_size_limit / block_size). When the buffer is
* filled, it will block and wait for user to read to get free space of buffer
* to continue read.
*/
Index total_size_limit = 0;
};
class BufferStream : public Stream {
public:
/**
* Actually I have no ideas about the best value for this. May change it later
* when I get some ideas.
*/
constexpr static Index kDefaultBlockSize = 1024;
/**
* Actually I have no ideas about the best value for this. May change it later
* when I get some ideas.
*/
constexpr static Index kDefaultTotalSizeLimit = 1024;
public:
BufferStream(const BufferStreamOptions& options);
~BufferStream() override;
bool CanSeek() override;
Index Seek(Index offset, SeekOrigin origin = SeekOrigin::Current) override;
bool CanRead() override;
Index Read(std::byte* buffer, Index offset, Index size) override;
bool CanWrite() = 0;
Index Write(const std::byte* buffer, Index offset, Index size) = 0;
virtual void Flush();
virtual void Close();
void SetEof();
private:
bool CheckClosed();
private:
Index block_size_;
Index total_size_limit_;
Index block_count_limit_;
std::list<Buffer> buffer_list_;
bool eof_;
std::mutex mutex_;
std::condition_variable condition_variable_;
};
} // namespace cru::io
|
