uring.hpp

#pragma once
 
#if __has_include(<liburing.h>)
#define HAVE_URING
 
#include <coroio/base.hpp>
#include <coroio/socket.hpp>
#include <coroio/poller.hpp>
 
#include <liburing.h>
#include <assert.h>
#include <sys/eventfd.h>
#include <sys/epoll.h>
 
#include <system_error>
#include <iostream>
#include <tuple>
#include <vector>
#include <coroutine>
#include <queue>
 
namespace NNet {
 
class TUring: public TPollerBase {
public:
    using TSocket = NNet::TPollerDrivenSocket<TUring>;
    using TFileHandle = NNet::TPollerDrivenFileHandle<TUring>;
    TUring(int queueSize = 256);
    ~TUring();
    void Read(int fd, void* buf, int size, std::coroutine_handle<> handle);
    void Write(int fd, const void* buf, int size, std::coroutine_handle<> handle);
    void Recv(int fd, void* buf, int size, std::coroutine_handle<> handle);
    void Send(int fd, const void* buf, int size, std::coroutine_handle<> handle);
    void Accept(int fd, struct sockaddr* addr, socklen_t* len, std::coroutine_handle<> handle);
    void Connect(int fd, const sockaddr* addr, socklen_t len, std::coroutine_handle<> handle);
    void Cancel(int fd);
    void Cancel(std::coroutine_handle<> h);
    void Register(int fd);
    int Wait(timespec ts = {10,0});
    void Poll() {
        Wait(GetTimeout());
    }
    int Result();
 
private:
    io_uring_sqe* GetSqe() {
        io_uring_sqe* r = io_uring_get_sqe(&Ring_);
        if (!r) {
            Submit();
            r = io_uring_get_sqe(&Ring_);
        }
        return r;
    }
 
    void Submit();
 
    int RingFd_; 
    int EpollFd_; 
    struct io_uring Ring_; 
    std::queue<int> Results_; 
    std::vector<char> Buffer_; 
};
 
} // namespace NNet
 
#endif