mirror of
https://github.com/ruvnet/RuVector.git
synced 2026-05-23 04:27:11 +00:00
893c93ab3e
## Summary Complete RuVector-Postgres v2 implementation with all major features: - 148 pg_extern SQL functions across 27 source files - Docker Hub publication ready with multi-arch builds (PG14-17) - Full pgvector drop-in compatibility verified ## New Features - **Hybrid Search** (7 functions): BM25 + vector fusion with RRF/linear/learned - **Multi-Tenancy** (17 functions): Tenant isolation, RLS, quotas - **Self-Healing** (23 functions): Problem detection, remediation strategies - **Integrity Control** (4 functions): Mincut gating, contracted graphs - **Self-Learning** (10 functions): Query trajectory tracking, optimization ## Infrastructure - GitHub Actions workflow for Docker Hub publication - CI workflow for testing PG14-17 - Integration test Docker setup with baseline testing - Benchmark suite for e2e, hybrid, integrity testing ## Files Changed - New: src/healing/, src/hybrid/, src/integrity/, src/tenancy/, src/workers/ - New: sql/ruvector--2.0.0.sql (SQL migration) - New: docker/publish-dockerhub.sh, docker-compose.integration.yml - Updated: Dockerfile for PG17 default, multi-arch builds - Updated: HNSW/IVFFlat index access methods with full pgrx AM support 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
130 lines
5.1 KiB
Docker
130 lines
5.1 KiB
Docker
# RuVector-Postgres Development & Testing Dockerfile
|
|
# Multi-stage build with PostgreSQL version support (14-17)
|
|
# Default: PostgreSQL 17 (latest with pgrx 0.12 support)
|
|
# Note: PostgreSQL 18 requires pgrx 0.15.0+ (planned for future release)
|
|
|
|
ARG PG_VERSION=17
|
|
ARG RUST_VERSION=1.83
|
|
|
|
# ============================================================================
|
|
# Stage 1: Base Builder with Rust and PostgreSQL dev dependencies
|
|
# ============================================================================
|
|
FROM rust:${RUST_VERSION}-bookworm AS base-builder
|
|
|
|
ARG PG_VERSION
|
|
|
|
# Add PostgreSQL APT repository
|
|
RUN sh -c 'echo "deb http://apt.postgresql.org/pub/repos/apt bookworm-pgdg main" > /etc/apt/sources.list.d/pgdg.list' && \
|
|
wget --quiet -O - https://www.postgresql.org/media/keys/ACCC4CF8.asc | apt-key add -
|
|
|
|
# Install PostgreSQL development dependencies for specified version
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
postgresql-${PG_VERSION} \
|
|
postgresql-server-dev-${PG_VERSION} \
|
|
libclang-dev \
|
|
clang \
|
|
pkg-config \
|
|
libssl-dev \
|
|
cmake \
|
|
wget \
|
|
git \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install pgrx (compatible with pgrx = "0.12" in Cargo.toml)
|
|
RUN cargo install cargo-pgrx --version 0.12.6 --locked
|
|
|
|
# Initialize pgrx for the specified PostgreSQL version
|
|
RUN cargo pgrx init --pg${PG_VERSION} /usr/lib/postgresql/${PG_VERSION}/bin/pg_config
|
|
|
|
# Set PGRX environment for consistent builds
|
|
ENV PGRX_PG_CONFIG_PATH=/usr/lib/postgresql/${PG_VERSION}/bin/pg_config
|
|
ENV PGRX_HOME=/root/.pgrx
|
|
ENV PG_VERSION=${PG_VERSION}
|
|
|
|
# ============================================================================
|
|
# Stage 2: Dependency Cache Builder
|
|
# ============================================================================
|
|
FROM base-builder AS deps-builder
|
|
|
|
ARG PG_VERSION
|
|
|
|
WORKDIR /build/ruvector-postgres
|
|
|
|
# Copy only dependency files first for better caching
|
|
COPY crates/ruvector-postgres/Cargo.toml ./
|
|
COPY crates/ruvector-postgres/build.rs ./
|
|
|
|
# Create dummy src to build dependencies
|
|
RUN mkdir -p src && \
|
|
echo "fn main() {}" > src/main.rs && \
|
|
echo "#[no_mangle] pub extern \"C\" fn pg_finfo_dummy() {}" > src/lib.rs
|
|
|
|
# Build dependencies only (this layer is cached)
|
|
RUN cargo build --release --features pg${PG_VERSION} || true
|
|
RUN rm -rf src
|
|
|
|
# ============================================================================
|
|
# Stage 3: Extension Builder
|
|
# ============================================================================
|
|
FROM deps-builder AS extension-builder
|
|
|
|
ARG PG_VERSION
|
|
|
|
# Copy actual source code
|
|
COPY crates/ruvector-postgres/Cargo.toml ./
|
|
COPY crates/ruvector-postgres/build.rs ./
|
|
COPY crates/ruvector-postgres/ruvector.control ./
|
|
COPY crates/ruvector-postgres/src ./src/
|
|
COPY crates/ruvector-postgres/sql ./sql/
|
|
COPY crates/ruvector-postgres/benches ./benches/
|
|
|
|
# Build the extension with the specified PG version feature and all features enabled
|
|
RUN cargo pgrx package \
|
|
--pg-config /usr/lib/postgresql/${PG_VERSION}/bin/pg_config \
|
|
--features pg${PG_VERSION},graph-complete
|
|
|
|
# pgrx generates .control and .so but not SQL - copy our hand-written SQL file
|
|
RUN cp sql/ruvector--2.0.0.sql target/release/ruvector-pg${PG_VERSION}/usr/share/postgresql/${PG_VERSION}/extension/ 2>/dev/null || true
|
|
|
|
# ============================================================================
|
|
# Stage 4: Runtime (Production)
|
|
# ============================================================================
|
|
FROM postgres:${PG_VERSION}-bookworm AS runtime
|
|
|
|
ARG PG_VERSION
|
|
|
|
# Install runtime dependencies
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
libssl3 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Copy built extension from builder
|
|
COPY --from=extension-builder /build/ruvector-postgres/target/release/ruvector-pg${PG_VERSION}/usr/share/postgresql/${PG_VERSION}/extension/* /usr/share/postgresql/${PG_VERSION}/extension/
|
|
COPY --from=extension-builder /build/ruvector-postgres/target/release/ruvector-pg${PG_VERSION}/usr/lib/postgresql/${PG_VERSION}/lib/* /usr/lib/postgresql/${PG_VERSION}/lib/
|
|
|
|
# Copy initialization script with proper permissions
|
|
COPY --chmod=644 crates/ruvector-postgres/docker/init.sql /docker-entrypoint-initdb.d/
|
|
|
|
# Set environment variables
|
|
ENV POSTGRES_USER=ruvector
|
|
ENV POSTGRES_PASSWORD=ruvector
|
|
ENV POSTGRES_DB=ruvector_test
|
|
ENV PG_VERSION=${PG_VERSION}
|
|
|
|
# PostgreSQL performance tuning
|
|
ENV POSTGRES_INITDB_ARGS="--data-checksums"
|
|
|
|
# Labels for version tracking
|
|
LABEL org.opencontainers.image.title="RuVector PostgreSQL Extension v2"
|
|
LABEL org.opencontainers.image.description="High-performance vector database extension for PostgreSQL with 230+ SQL functions, Flash Attention, GNN, hybrid search, multi-tenancy, and self-healing"
|
|
LABEL org.opencontainers.image.version="2.0.0"
|
|
LABEL org.opencontainers.image.vendor="ruv.io"
|
|
LABEL org.opencontainers.image.source="https://github.com/ruvnet/ruvector"
|
|
LABEL ruvector.pg.version="${PG_VERSION}"
|
|
LABEL ruvector.features="attention,gnn,hybrid,tenancy,healing,learning,hyperbolic,graph"
|
|
|
|
# Health check
|
|
HEALTHCHECK --interval=5s --timeout=5s --start-period=10s --retries=5 \
|
|
CMD pg_isready -U $POSTGRES_USER -d $POSTGRES_DB || exit 1
|
|
|
|
EXPOSE 5432
|