feat: AIO batchmode

Signed-off-by: usamoi <usamoi@outlook.com>

Author: usamoi

.github/workflows/check.yml

@@ -11,6 +11,11 @@ concurrency:
 
 jobs:
   style:
+    if: |
+      (github.event_name == 'push' && !contains(github.event.head_commit.message, 'job: -style')) ||
+      (github.event_name == 'pull_request' && !contains(github.event.pull_request.body, 'job: -style')) ||
+      github.event_name == 'workflow_dispatch'
+
     runs-on: "ubuntu-24.04"
 
     steps:
@@ -100,6 +105,11 @@ jobs:
           ! grep -P '\t' -r ./sql
 
   lint:
+    if: |
+      (github.event_name == 'push' && !contains(github.event.head_commit.message, 'job: -lint')) ||
+      (github.event_name == 'pull_request' && !contains(github.event.pull_request.body, 'job: -lint')) ||
+      github.event_name == 'workflow_dispatch'
+
     strategy:
       matrix:
         arch: ["x86_64", "aarch64"]
@@ -817,7 +827,7 @@ jobs:
 
     strategy:
       matrix:
-        version: ["17"]
+        version: ["17", "18"]
 
     runs-on: "ubuntu-24.04"
 
@@ -851,7 +861,7 @@ jobs:
           sudo chmod 700 /var/lib/postgresql
           sudo chown postgres:postgres /var/lib/postgresql
 
-          sudo pacman -S python clang valgrind --noconfirm
+          sudo pacman -S python clang valgrind liburing numactl --noconfirm
           curl https://sh.rustup.rs -sSf | sh -s -- -y --default-toolchain beta
 
           curl -fsSL https://github.com/risinglightdb/sqllogictest-rs/releases/download/v0.28.4/sqllogictest-bin-v0.28.4-$(uname -m)-unknown-linux-musl.tar.gz | tar -xOzf - ./sqllogictest | sudo install -m 755 /dev/stdin /usr/local/bin/sqllogictest
@@ -861,9 +871,16 @@ jobs:
 
       - name: Install PostgreSQL
         run: |
+          if [ "${{ matrix.version }}" = "17" ]; then
+            VERSION=17.6
+          fi
+          if [ "${{ matrix.version }}" = "18" ]; then
+            VERSION=18.0
+          fi
+
           mkdir ~/postgresql-install
-          curl -fsSL https://ftp.postgresql.org/pub/source/v17.6/postgresql-17.6.tar.bz2 | tar -xj -C ~/postgresql-install
-          pushd ~/postgresql-install/postgresql-17.6
+          curl -fsSL https://ftp.postgresql.org/pub/source/v$VERSION/postgresql-$VERSION.tar.bz2 | tar -xj -C ~/postgresql-install
+          pushd ~/postgresql-install/postgresql-$VERSION
           ./configure \
               --prefix=/usr \
               --sysconfdir=/etc \
@@ -881,6 +898,9 @@ jobs:
               --with-system-tzdata=/usr/share/zoneinfo \
               --with-uuid=e2fs \
               --with-zstd \
+              ${{ matrix.version >= 18 && '--with-libcurl'  || '' }} \
+              ${{ matrix.version >= 18 && '--with-libnuma'  || '' }} \
+              ${{ matrix.version >= 18 && '--with-liburing' || '' }} \
               --enable-debug \
               --enable-cassert \
               CC='sccache clang' \

src/index/vchordg/scanners/default.rs

@@ -108,8 +108,8 @@ impl SearchBuilder for DefaultBuilder {
         let Some(vector) = vector else {
             return Box::new(std::iter::empty()) as Box<dyn Iterator<Item = (f32, [u16; 3], bool)>>;
         };
-        let vertex_hints = Hints::default().full(true);
-        let vector_hints = Hints::default().full(true);
+        let vertex_hints = Hints::default().full(true).batch(true);
+        let vector_hints = Hints::default().full(true).batch(true);
         let make_vertex_plain_prefetcher = MakePlainPrefetcher { index };
         let make_vertex_simple_prefetcher = MakeSimplePrefetcher { index };
         let make_vertex_stream_prefetcher = MakeStreamPrefetcher {

src/index/vchordrq/scanners/default.rs

@@ -111,8 +111,8 @@ impl SearchBuilder for DefaultBuilder {
         let Some(vector) = vector else {
             return Box::new(std::iter::empty()) as Box<dyn Iterator<Item = (f32, [u16; 3], bool)>>;
         };
-        let search_hints = Hints::default().full(true);
-        let rerank_hints = Hints::default().full(false);
+        let search_hints = Hints::default().full(true).batch(false);
+        let rerank_hints = Hints::default().full(false).batch(true);
         let make_h1_plain_prefetcher = MakeH1PlainPrefetcher { index };
         let make_h0_plain_prefetcher = MakeH0PlainPrefetcher { index };
         let make_h0_simple_prefetcher = MakeH0SimplePrefetcher { index };

src/index/vchordrq/scanners/maxsim.rs

@@ -103,8 +103,8 @@ impl SearchBuilder for MaxsimBuilder {
             pgrx::error!("maxsim search with rerank_in_table is not supported");
         }
         assert!(matches!(opfamily.distance_kind(), DistanceKind::Dot));
-        let search_hints = Hints::default().full(true);
-        let rerank_hints = Hints::default().full(false);
+        let search_hints = Hints::default().full(true).batch(false);
+        let rerank_hints = Hints::default().full(false).batch(true);
         let make_h1_plain_prefetcher = MakeH1PlainPrefetcher { index };
         let make_h0_plain_prefetcher = MakeH0PlainPrefetcher { index };
         let make_h0_simple_prefetcher = MakeH0SimplePrefetcher { index };