Add io-uring feature flag for restate-server

.github/workflows/ci.yml

@@ -45,10 +45,6 @@ jobs:
     env:
       RUST_BACKTRACE: full
     steps:
-      # Disabled as uring is not used in production yet.
-      # - name: Install liburing
-      #   run: sudo apt-get update && sudo apt-get install -y liburing-dev
-
       - uses: actions/checkout@v4
 
       - name: Install Rust toolchain
@@ -91,6 +87,10 @@ jobs:
       - name: Check Cargo.lock file is updated
         run: cargo update -w --locked
 
+      # Installing liburing even if not used because of cargo-hakari feature unification
+      - name: Install liburing
+        run: sudo apt-get update && sudo apt-get install -y liburing-dev
+
       - name: Run lint
         run: just lint
 

crates/rocksdb/Cargo.toml

@@ -10,6 +10,7 @@ publish = false
 [features]
 default = []
 test-util = ["restate-types/test-util"]
+io-uring = ["rocksdb/io-uring"]
 
 [dependencies]
 restate-workspace-hack = { workspace = true }

docker/Dockerfile

@@ -113,7 +113,7 @@ COPY --from=upload-true /restate/target/restate-server.debug /
 # NOTE: When modifying this runtime stage, also update docker/Dockerfile.refresh to keep them in sync.
 FROM debian:trixie-slim AS runtime
 # useful for health checks
-RUN apt-get update && apt-get install --no-install-recommends -y jq curl && rm -rf /var/lib/apt/lists/*
+RUN apt update && apt upgrade -y && apt install --no-install-recommends -y jq curl liburing2 && rm -rf /var/lib/apt/lists/*
 # Create symlink for debug symbols fallback path (for read-only /usr/local/bin scenarios)
 RUN mkdir -p /usr/local/bin/.debug && \
     ln -s /tmp/.debug/restate-server.debug /usr/local/bin/.debug/restate-server.debug

docker/Dockerfile.refresh

@@ -23,7 +23,7 @@ ARG BASE_IMAGE
 FROM ${SOURCE_IMAGE} AS source
 
 FROM ${BASE_IMAGE} AS runtime
-RUN apt-get update && apt-get install --no-install-recommends -y jq curl && rm -rf /var/lib/apt/lists/*
+RUN apt update && apt upgrade -y && apt install --no-install-recommends -y jq curl liburing2 && rm -rf /var/lib/apt/lists/*
 # Create symlink for debug symbols fallback path (for read-only /usr/local/bin scenarios)
 RUN mkdir -p /usr/local/bin/.debug && \
     ln -s /tmp/.debug/restate-server.debug /usr/local/bin/.debug/restate-server.debug

docker/debug.Dockerfile

@@ -73,14 +73,15 @@ RUN --mount=type=cache,target=/var/cache/sccache \
 
 # We do not need the Rust toolchain to run the server binary!
 FROM debian:trixie-slim AS runtime
+# useful for health checks
+RUN apt update && apt upgrade -y && apt install -y jq curl liburing2 && rm -rf /var/lib/apt/lists/*
+
 COPY --from=builder /restate/target/restate-server /usr/local/bin
 COPY --from=builder /restate/target/restatectl /usr/local/bin
 COPY --from=builder /restate/target/restate /usr/local/bin
 COPY --from=builder /restate/NOTICE /NOTICE
 COPY --from=builder /restate/LICENSE /LICENSE
 # copy OS roots
 COPY --from=builder /etc/ssl /etc/ssl
-# useful for health checks
-RUN apt-get update && apt-get install -y jq curl && rm -rf /var/lib/apt/lists/*
 WORKDIR /
 ENTRYPOINT ["/usr/local/bin/restate-server"]

server/Cargo.toml

@@ -33,6 +33,7 @@ no-trace-logging = ["tracing/max_level_trace", "tracing/release_max_level_debug"
 metadata-api = ["restate-admin/metadata-api"]
 kafka-oidc = ["restate-node/kafka-oidc"]
 taskdump = ["restate-core/taskdump"]
+io-uring = ["restate-rocksdb/io-uring"]
 
 [dependencies]
 restate-workspace-hack = { workspace = true }

workspace-hack/Cargo.toml

@@ -102,6 +102,8 @@ regex-automata = { version = "0.4", default-features = false, features = ["dfa",
 regex-syntax = { version = "0.8" }
 reqwest = { version = "0.12", default-features = false, features = ["blocking", "http2", "json", "rustls-tls", "rustls-tls-native-roots", "stream"] }
 ring = { version = "0.17", features = ["std"] }
+rust-librocksdb-sys = { git = "https://github.com/restatedev/rust-rocksdb", rev = "432ac4288a219dd70d4c71ccfcd0ad51d6bcc0d1", default-features = false, features = ["bindgen-runtime", "bzip2", "io-uring", "jemalloc", "lz4", "snappy", "static", "zlib", "zstd"] }
+rust-rocksdb = { git = "https://github.com/restatedev/rust-rocksdb", rev = "432ac4288a219dd70d4c71ccfcd0ad51d6bcc0d1", features = ["io-uring", "jemalloc", "multi-threaded-cf"] }
 rustls = { version = "0.23", default-features = false, features = ["logging", "prefer-post-quantum", "ring", "std", "tls12"] }
 rustls-pki-types = { version = "1", features = ["std"] }
 rustls-webpki = { version = "0.103", default-features = false, features = ["aws-lc-rs", "ring", "std"] }
@@ -234,6 +236,8 @@ regex-automata = { version = "0.4", default-features = false, features = ["dfa",
 regex-syntax = { version = "0.8" }
 reqwest = { version = "0.12", default-features = false, features = ["blocking", "http2", "json", "rustls-tls", "rustls-tls-native-roots", "stream"] }
 ring = { version = "0.17", features = ["std"] }
+rust-librocksdb-sys = { git = "https://github.com/restatedev/rust-rocksdb", rev = "432ac4288a219dd70d4c71ccfcd0ad51d6bcc0d1", default-features = false, features = ["bindgen-runtime", "bzip2", "io-uring", "jemalloc", "lz4", "snappy", "static", "zlib", "zstd"] }
+rust-rocksdb = { git = "https://github.com/restatedev/rust-rocksdb", rev = "432ac4288a219dd70d4c71ccfcd0ad51d6bcc0d1", features = ["io-uring", "jemalloc", "multi-threaded-cf"] }
 rustls = { version = "0.23", default-features = false, features = ["logging", "prefer-post-quantum", "ring", "std", "tls12"] }
 rustls-pki-types = { version = "1", features = ["std"] }
 rustls-webpki = { version = "0.103", default-features = false, features = ["aws-lc-rs", "ring", "std"] }