mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 10:52:30 +00:00
97f2a2213e
* Move some code outside dbms/src folder * Fix paths
143 lines
2.6 KiB
C++
143 lines
2.6 KiB
C++
#if defined(__linux__)
|
|
|
|
#include <boost/noncopyable.hpp>
|
|
#include <Common/Exception.h>
|
|
#include <sys/syscall.h>
|
|
#include <unistd.h>
|
|
|
|
#include <IO/AIO.h>
|
|
|
|
|
|
/** Small wrappers for asynchronous I/O.
|
|
*/
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int CANNOT_IOSETUP;
|
|
}
|
|
}
|
|
|
|
|
|
int io_setup(unsigned nr, aio_context_t * ctxp)
|
|
{
|
|
return syscall(__NR_io_setup, nr, ctxp);
|
|
}
|
|
|
|
int io_destroy(aio_context_t ctx)
|
|
{
|
|
return syscall(__NR_io_destroy, ctx);
|
|
}
|
|
|
|
int io_submit(aio_context_t ctx, long nr, struct iocb * iocbpp[]) // NOLINT
|
|
{
|
|
return syscall(__NR_io_submit, ctx, nr, iocbpp);
|
|
}
|
|
|
|
int io_getevents(aio_context_t ctx, long min_nr, long max_nr, io_event * events, struct timespec * timeout) // NOLINT
|
|
{
|
|
return syscall(__NR_io_getevents, ctx, min_nr, max_nr, events, timeout);
|
|
}
|
|
|
|
|
|
AIOContext::AIOContext(unsigned int nr_events)
|
|
{
|
|
ctx = 0;
|
|
if (io_setup(nr_events, &ctx) < 0)
|
|
DB::throwFromErrno("io_setup failed", DB::ErrorCodes::CANNOT_IOSETUP);
|
|
}
|
|
|
|
AIOContext::~AIOContext()
|
|
{
|
|
io_destroy(ctx);
|
|
}
|
|
|
|
#elif defined(__FreeBSD__)
|
|
|
|
# include <aio.h>
|
|
# include <boost/noncopyable.hpp>
|
|
# include <sys/event.h>
|
|
# include <sys/time.h>
|
|
# include <sys/types.h>
|
|
# include <Common/Exception.h>
|
|
|
|
# include <IO/AIO.h>
|
|
|
|
|
|
/** Small wrappers for asynchronous I/O.
|
|
*/
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int CANNOT_IOSETUP;
|
|
}
|
|
}
|
|
|
|
|
|
int io_setup(void)
|
|
{
|
|
return kqueue();
|
|
}
|
|
|
|
int io_destroy(int ctx)
|
|
{
|
|
return close(ctx);
|
|
}
|
|
|
|
int io_submit(int ctx, long nr, struct iocb * iocbpp[])
|
|
{
|
|
for (long i = 0; i < nr; i++)
|
|
{
|
|
struct aiocb * iocb = &iocbpp[i]->aio;
|
|
|
|
struct sigevent * se = &iocb->aio_sigevent;
|
|
se->sigev_notify_kqueue = ctx;
|
|
se->sigev_notify_kevent_flags = 0;
|
|
se->sigev_notify = SIGEV_KEVENT;
|
|
se->sigev_value.sival_ptr = iocbpp[i];
|
|
|
|
switch (iocb->aio_lio_opcode)
|
|
{
|
|
case LIO_READ:
|
|
{
|
|
int r = aio_read(iocb);
|
|
if (r < 0)
|
|
return r;
|
|
break;
|
|
}
|
|
case LIO_WRITE:
|
|
{
|
|
int r = aio_write(iocb);
|
|
if (r < 0)
|
|
return r;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return nr;
|
|
}
|
|
|
|
int io_getevents(int ctx, long, long max_nr, struct kevent * events, struct timespec * timeout)
|
|
{
|
|
return kevent(ctx, NULL, 0, events, max_nr, timeout);
|
|
}
|
|
|
|
|
|
AIOContext::AIOContext(unsigned int)
|
|
{
|
|
ctx = io_setup();
|
|
if (ctx < 0)
|
|
DB::throwFromErrno("io_setup failed", DB::ErrorCodes::CANNOT_IOSETUP);
|
|
}
|
|
|
|
AIOContext::~AIOContext()
|
|
{
|
|
io_destroy(ctx);
|
|
}
|
|
|
|
#endif
|