lance-io 6.0.0

I/O utilities for Lance
Documentation 
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

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright The Lance Authors

//! Protocol types for communication between UringReader and the io_uring thread.

use bytes::BytesMut;
use std::io;
use std::os::unix::io::RawFd;
use std::sync::Mutex;
use std::task::Waker;
use std::thread::ThreadId;

pub(super) struct RequestState {
    pub completed: bool,
    pub waker: Option<Waker>,
    pub err: Option<io::Error>,
    pub buffer: BytesMut,
    /// Accumulated bytes read across retries (for handling short reads).
    pub bytes_read: usize,
}

/// I/O request object that contains all state for a single read operation.
/// This is shared between the submitter, uring thread, and future via Arc.
pub(super) struct IoRequest {
    /// File descriptor to read from.
    pub fd: RawFd,

    /// Byte offset to start reading from.
    pub offset: u64,

    /// Number of bytes to read.
    pub length: usize,

    pub thread_id: ThreadId,

    /// Completion flag - set to true when operation completes.
    pub state: Mutex<RequestState>,
}

impl IoRequest {
    /// Mark this request as failed with the given error.
    ///
    /// Sets the error, marks completed, and wakes any waiting future.
    /// Used when a request cannot be submitted (e.g. SQ full).
    pub(super) fn fail(&self, err: io::Error) {
        let mut state = self.state.lock().unwrap();
        state.err = Some(err);
        state.completed = true;
        if let Some(waker) = state.waker.take() {
            drop(state);
            waker.wake();
        }
    }
}