blob: 9dc99dbe49f774e8032539b87b0bd87f59466c35 (
plain) (
blame)
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#pragma once
#include "zerocopy.h"
#include "zerocopy_output.h"
#include <util/generic/ptr.h>
class TBuffer;
/**
* @addtogroup Streams_Buffers
* @{
*/
/**
* Output stream that writes into a `TBuffer`.
*/
class TBufferOutput: public IZeroCopyOutput {
public:
class TImpl;
/**
* Constructs a stream that writes into an internal buffer.
*
* @param buflen Initial size of the internal buffer.
*/
TBufferOutput(size_t buflen = 1024);
/**
* Constructs a stream that writes into the provided buffer. It's up to the
* user to make sure that the buffer doesn't get destroyed while this stream
* is in use.
*
* @param buffer Buffer to write into.
*/
TBufferOutput(TBuffer& buffer);
TBufferOutput(TBufferOutput&&) noexcept;
TBufferOutput& operator=(TBufferOutput&&) noexcept;
~TBufferOutput() override;
/**
* @returns Buffer that this stream writes into.
*/
TBuffer& Buffer() const noexcept;
private:
size_t DoNext(void** ptr) override;
void DoUndo(size_t len) override;
void DoWrite(const void* buf, size_t len) override;
void DoWriteC(char c) override;
private:
THolder<TImpl> Impl_;
};
/**
* Input stream that reads from an external `TBuffer`.
*/
class TBufferInput: public IZeroCopyInputFastReadTo {
public:
/**
* Constructs a stream that reads from an external buffer. It's up to the
* user to make sure that the buffer doesn't get destroyed before this
* stream.
*
* @param buffer External buffer to read from.
*/
TBufferInput(const TBuffer& buffer);
~TBufferInput() override;
const TBuffer& Buffer() const noexcept;
void Rewind() noexcept;
protected:
size_t DoNext(const void** ptr, size_t len) override;
void DoUndo(size_t len) override;
private:
const TBuffer& Buf_;
size_t Readed_;
};
/**
* Input/output stream that works with a `TBuffer`.
*/
class TBufferStream: public TBufferOutput, public TBufferInput {
public:
/**
* Constructs a stream that works with an internal buffer.
*
* @param buflen Initial size of the internal buffer.
*/
inline TBufferStream(size_t buflen = 1024)
: TBufferOutput(buflen)
, TBufferInput(TBufferOutput::Buffer())
{
}
/**
* Constructs a stream that works with the provided buffer.
*
* @param buffer Buffer to work with.
*/
inline TBufferStream(TBuffer& buffer)
: TBufferOutput(buffer)
, TBufferInput(TBufferOutput::Buffer())
{
}
~TBufferStream() override = default;
using TBufferOutput::Buffer;
};
/** @} */
|
