mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-13 11:04:10 +00:00
66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
#pragma once
|
||
|
||
#include <DB/IO/IBufferAIO.h>
|
||
#include <DB/IO/ReadBuffer.h>
|
||
#include <DB/IO/BufferWithOwnMemory.h>
|
||
#include <statdaemons/AIO.h>
|
||
|
||
#include <limits>
|
||
|
||
#include <unistd.h>
|
||
#include <fcntl.h>
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Класс для асинхронной чтения данных.
|
||
* Все размеры и смещения должны быть кратны 512 байтам.
|
||
*/
|
||
class ReadBufferAIO : public IBufferAIO, public BufferWithOwnMemory<ReadBuffer>
|
||
{
|
||
public:
|
||
ReadBufferAIO(const std::string & filename_, size_t buffer_size_ = DBMS_DEFAULT_BUFFER_SIZE, int flags_ = -1, mode_t mode_ = 0666,
|
||
char * existing_memory_ = nullptr);
|
||
~ReadBufferAIO() override;
|
||
|
||
ReadBufferAIO(const ReadBufferAIO &) = delete;
|
||
ReadBufferAIO & operator=(const ReadBufferAIO &) = delete;
|
||
|
||
void setMaxBytes(size_t max_bytes_read_);
|
||
off_t seek(off_t off, int whence = SEEK_SET);
|
||
size_t getPositionInFile() const noexcept { return pos_in_file - (working_buffer.end() - pos); }
|
||
std::string getFileName() const noexcept override { return filename; }
|
||
int getFD() const noexcept override { return fd; }
|
||
|
||
private:
|
||
bool nextImpl() override;
|
||
void waitForCompletion() override;
|
||
void swapBuffers() noexcept override;
|
||
|
||
private:
|
||
/// Буфер для асинхронных операций чтения данных.
|
||
BufferWithOwnMemory<ReadBuffer> fill_buffer;
|
||
const std::string filename;
|
||
|
||
AIOContext aio_context;
|
||
iocb cb;
|
||
std::vector<iocb *> request_ptrs;
|
||
std::vector<io_event> events;
|
||
|
||
int fd = -1;
|
||
size_t max_bytes_read = std::numeric_limits<size_t>::max();
|
||
size_t total_bytes_read = 0;
|
||
off_t pos_in_file = 0;
|
||
|
||
/// Асинхронная операция чтения ещё не завершилась.
|
||
bool is_pending_read = false;
|
||
/// Было получено исключение.
|
||
bool got_exception = false;
|
||
/// Конец файла достигнут.
|
||
bool is_eof = false;
|
||
/// Был отправлен хоть один запрос на асинхронную операцию чтения.
|
||
bool is_started = false;
|
||
};
|
||
|
||
}
|