Dockerfile

# This file is inspired by github.com/neondatabase/neon's docker file.

ARG PG_MAJOR=15
ARG EXTENSION_DIR=/usr/share/postgresql/${PG_MAJOR}/extension
ARG INCLUDE_DIR=/usr/include/postgresql/${PG_MAJOR}
ARG LIB_DIR=/usr/lib/postgresql/${PG_MAJOR}
ARG BIN_DIR=${LIB_DIR}/bin
ARG PGCONFIG=${BIN_DIR}/pg_config

#########################################################################################
#
# Layer "pg-build"
# Used to copy postgres header files from, without needing postgres installed
#
#########################################################################################

FROM postgres:${PG_MAJOR}-bullseye AS pg-build

RUN apt update && \
    apt install -y postgresql-server-dev-$PG_MAJOR 


#########################################################################################
#
# Layer "build-deps"
# Used to copy postgres header files from, without needing postgres installed
#
#########################################################################################
FROM debian:bullseye-slim AS build-deps

ARG EXTENSION_DIR
ARG LIB_DIR

COPY --from=pg-build ${EXTENSION_DIR}/ ${EXTENSION_DIR}/
COPY --from=pg-build ${LIB_DIR}/ ${LIB_DIR}/

ENV CC=gcc
ENV CXX=g++

RUN apt update && \
    apt install -y git autoconf automake libtool build-essential bison flex libreadline-dev \
    zlib1g-dev libxml2-dev libcurl4-openssl-dev libossp-uuid-dev wget pkg-config libssl-dev \
    libicu-dev libxslt1-dev liblz4-dev libzstd-dev zstd clang-13


#########################################################################################
#
# Layer "vector-ext"
# compile pgvector extension
#
#########################################################################################
FROM build-deps AS vector-ext

ARG EXTENSION_DIR
ARG LIB_DIR
ARG INCLUDE_DIR
ARG PGCONFIG

ENV PGVECTOR_VERSION 0.7.0
ENV PGVECTOR_SHA 1b5503a35c265408b6eb282621c5e1e75f7801afc04eecb950796cfee2e3d1d8

COPY --from=pg-build ${EXTENSION_DIR}/ ${EXTENSION_DIR}/
COPY --from=pg-build ${LIB_DIR}/ ${LIB_DIR}/
COPY --from=pg-build ${INCLUDE_DIR}/ ${INCLUDE_DIR}/

RUN wget https://github.com/pgvector/pgvector/archive/refs/tags/v${PGVECTOR_VERSION}.tar.gz -O pgvector.tar.gz && \
    echo "${PGVECTOR_SHA} pgvector.tar.gz" | sha256sum --check && \
    mkdir pgvector-src && cd pgvector-src && tar xvzf ../pgvector.tar.gz --strip-components=1 -C . && \
    make -j $(getconf _NPROCESSORS_ONLN) OPTFLAGS="" PG_CONFIG=${PGCONFIG} && \
    make -j $(getconf _NPROCESSORS_ONLN) install OPTFLAGS="" PG_CONFIG=${PGCONFIG}

RUN mkdir /out /out/lib /out/share /out/share/extension && \
    cp ${LIB_DIR}/lib/vector* /out/lib/ && \
    cp ${EXTENSION_DIR}/vector* /out/share/extension/ && \
    echo 'trusted = true' >> /out/share/extension/vector.control

    
#########################################################################################
#
# Final image
#
#########################################################################################

FROM postgres:${PG_MAJOR}-bullseye

LABEL maintainer="Encore - https://encore.dev"
ARG EXTENSION_DIR
ARG LIB_DIR

ENV POSTGIS_MAJOR 3

RUN apt update \
    && apt install -y --no-install-recommends \
        postgresql-$PG_MAJOR-postgis-$POSTGIS_MAJOR \
        postgresql-$PG_MAJOR-postgis-$POSTGIS_MAJOR-scripts

COPY --from=vector-ext /out/lib/ ${LIB_DIR}/lib/
COPY --from=vector-ext /out/share/extension/ ${EXTENSION_DIR}/

# Ensure extensions are trusted. Note: not all extensions
RUN for ext in address_standardizer address_standardizer-3 address_standardizer_data_us \
    address_standardizer_data_us-3 autoinc amcheck bloom btree_gin btree_gist citext \
    cube dblink dict_int dict_xsyn earthdistance fuzzystrmatch hstore insert_username \
    intagg intarray isn lo ltree moddatetime pageinspect pg_buffercache pgcrypto \
    pg_freespacemap pg_prewarm pgrowlocks pg_stat_statements pgstattuple pg_trgm \
    pg_visibility plpgsql postgis postgis-3 postgis_raster postgis_raster-3 postgis_sfcgal \
    postgis_sfcgal-3 postgis_tiger_geocoder postgis_tiger_geocoder-3 postgis_topology \
    postgis_topology-3 postgres_fdw refint seg sslinfo tablefunc tsm_system_rows \
    tsm_system_time unaccent uuid-ossp vector; do \
        if grep -q "trusted" "$EXTENSION_DIR/$ext.control"; then \
            echo "INFO: $ext is already trusted"; \
        else \
            echo "INFO: $ext is not trusted, adding trusted = true to $ext.control"; \
            echo "trusted = true" >> "$EXTENSION_DIR/$ext.control"; \
        fi \
    done