X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=io_uring_engine.h;h=470bacf9f4c450cb49d9a25185e8100eb4b073b3;hb=498466c6434e479cd973fe7655b4feb51dd2c378;hp=186c72e27b92437586b2607ca2f436512666acc8;hpb=896e1b19baed764eadb320c95c13726efba546d0;p=plocate

diff --git a/io_uring_engine.h b/io_uring_engine.h
index 186c72e..470bacf 100644
--- a/io_uring_engine.h
+++ b/io_uring_engine.h
@@ -3,28 +3,40 @@
 
 #include <functional>
 #include <queue>
-#include <stdint.h>
-#include <string>
+#include <stddef.h>
+#include <string_view>
+#include <sys/socket.h>
+#include <sys/types.h>
+
+struct io_uring_sqe;
 #ifndef WITHOUT_URING
 #include <liburing.h>
 #endif
-#include <sys/socket.h>
 
 class IOUringEngine {
 public:
 	IOUringEngine(size_t slop_bytes);
 	void submit_read(int fd, size_t len, off_t offset, std::function<void(std::string_view)> cb);
+
+	// NOTE: We just do the stat() to get the data into the dentry cache for fast access;
+	// we don't care about the return value. Thus, the callback has no parameter lists.
+	// If we have no io_uring, the callback will be made immediately, with no stat() call
+	// being done.
+	void submit_stat(const char *path, std::function<void()> cb);
+	bool get_supports_stat() { return supports_stat; }
+
 	void finish();
 	size_t get_waiting_reads() const { return pending_reads + queued_reads.size(); }
 
 private:
 #ifndef WITHOUT_URING
 	void submit_read_internal(io_uring_sqe *sqe, int fd, size_t len, off_t offset, std::function<void(std::string_view)> cb);
+	void submit_stat_internal(io_uring_sqe *sqe, char *path, std::function<void()> cb);
 
 	io_uring ring;
 #endif
 	size_t pending_reads = 0;  // Number of requests we have going in the ring.
-	bool using_uring;
+	bool using_uring, supports_stat = false;
 	const size_t slop_bytes;
 
 	struct QueuedRead {
@@ -35,23 +47,40 @@ private:
 	};
 	std::queue<QueuedRead> queued_reads;
 
+	struct QueuedStat {
+		char *pathname;  // Owned by us.
+		std::function<void()> cb;
+	};
+	std::queue<QueuedStat> queued_stats;
+
+	enum Op { OP_READ,
+	          OP_STAT };
+
 	struct PendingRead {
-		void *buf;
-		size_t len;
-		std::function<void(std::string_view)> cb;
+		Op op;
 
-		// For re-submission.
-		int fd;
-		off_t offset;
-		iovec iov;
+		std::function<void(std::string_view)> read_cb;
+		std::function<void()> stat_cb;
+
+		union {
+			struct {
+				void *buf;
+				size_t len;
+
+				// For re-submission.
+				int fd;
+				off_t offset;
+				iovec iov;
+			} read;
+			struct {
+				char *pathname;
+				struct statx *buf;
+			} stat;
+		};
 	};
 
 	// 256 simultaneous requests should be ample, for slow and fast media alike.
 	static constexpr size_t queue_depth = 256;
 };
 
-// A wrapper around pread() that returns an incomplete read.
-// Always synchronous (no io_uring).
-void complete_pread(int fd, void *ptr, size_t len, off_t offset);
-
 #endif  // !defined(IO_URING_ENGINE_H)