blob: cb73b1e1d08e287f4967a27892b77a830a5d86b8 (
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
|
#pragma once
#include <IO/WriteBufferFromFileBase.h>
#include <Common/Throttler_fwd.h>
namespace DB
{
/** Use ready file descriptor. Does not open or close a file.
*/
class WriteBufferFromFileDescriptor : public WriteBufferFromFileBase
{
public:
explicit WriteBufferFromFileDescriptor(
int fd_ = -1,
size_t buf_size = DBMS_DEFAULT_BUFFER_SIZE,
char * existing_memory = nullptr,
ThrottlerPtr throttler_ = {},
size_t alignment = 0,
std::string file_name_ = "");
/** Could be used before initialization if needed 'fd' was not passed to constructor.
* It's not possible to change 'fd' during work.
*/
void setFD(int fd_)
{
fd = fd_;
}
~WriteBufferFromFileDescriptor() override;
int getFD() const
{
return fd;
}
void sync() override;
/// clang-tidy wants these methods to be const, but
/// they are not const semantically
off_t seek(off_t offset, int whence); // NOLINT
void truncate(off_t length); // NOLINT
/// Name or some description of file.
std::string getFileName() const override;
off_t size() const;
protected:
void nextImpl() override;
int fd;
ThrottlerPtr throttler;
/// If file has name contains filename, otherwise contains string "(fd=...)"
std::string file_name;
void finalizeImpl() override;
};
}
|
