mirror of
https://github.com/ruvnet/RuVector.git
synced 2026-05-24 05:43:58 +00:00
ee40332ce1
- Fix simsimd Option/Result type mismatch in scaled_dot.rs - Fix f32/f64 type conversions in poincare.rs and lorentz.rs - Fix AVX512 missing wrapper functions by using AVX2 fallback - Fix Vec<Vec<f32>> to JsonB for pgrx pg_extern compatibility - Fix DashMap get() to get_mut() for mutable access - Fix router.rs dereference for best_score comparison - Update Dockerfile to copy pre-written SQL file for pgrx - Simplify init.sql to use correct function names - Add postgres-cli npm package for CLI tooling All changes tested successfully in Docker with: - Extension loads with AVX2 SIMD support (8 floats/op) - Distance functions verified working - PostgreSQL 16 container runs successfully 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
78 lines
2.9 KiB
Docker
78 lines
2.9 KiB
Docker
# RuVector-Postgres Development & Testing Dockerfile
|
|
# Multi-stage build for PostgreSQL 16 with pgrx and all dependencies
|
|
|
|
FROM rust:1.83-bookworm AS builder
|
|
|
|
# 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
|
|
RUN apt-get update && apt-get install -y \
|
|
postgresql-16 \
|
|
postgresql-server-dev-16 \
|
|
libclang-dev \
|
|
clang \
|
|
pkg-config \
|
|
libssl-dev \
|
|
cmake \
|
|
wget \
|
|
&& 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 PostgreSQL 16
|
|
RUN cargo pgrx init --pg16 /usr/lib/postgresql/16/bin/pg_config
|
|
|
|
# Set PGRX environment for consistent builds
|
|
ENV PGRX_PG_CONFIG_PATH=/usr/lib/postgresql/16/bin/pg_config
|
|
ENV PGRX_HOME=/root/.pgrx
|
|
|
|
# Set working directory - use /build to avoid any parent Cargo.toml issues
|
|
WORKDIR /build/ruvector-postgres
|
|
|
|
# Copy only the postgres crate - this is a standalone crate with no workspace dependencies
|
|
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 all features (standalone, no workspace)
|
|
RUN cargo pgrx package \
|
|
--pg-config /usr/lib/postgresql/16/bin/pg_config \
|
|
--features pg16
|
|
|
|
# pgrx only generates .control and .so - copy pre-written SQL file
|
|
RUN if [ ! -f target/release/ruvector-pg16/usr/share/postgresql/16/extension/ruvector--0.1.0.sql ]; then \
|
|
echo "Copying pre-written SQL file..." && \
|
|
cp sql/ruvector--0.1.0.sql target/release/ruvector-pg16/usr/share/postgresql/16/extension/; \
|
|
fi
|
|
|
|
# Runtime image
|
|
FROM postgres:16-bookworm
|
|
|
|
# Install runtime dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
libssl3 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Copy built extension from builder (path uses crate name 'ruvector' not 'ruvector_postgres')
|
|
COPY --from=builder /build/ruvector-postgres/target/release/ruvector-pg16/usr/share/postgresql/16/extension/* /usr/share/postgresql/16/extension/
|
|
COPY --from=builder /build/ruvector-postgres/target/release/ruvector-pg16/usr/lib/postgresql/16/lib/* /usr/lib/postgresql/16/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
|
|
|
|
# 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
|