adapt openGauss for annbenchmark

Author: createart

.github/workflows/benchmarks.yml

@@ -72,6 +72,7 @@ jobs:
           - voyager
           - vsag
           - weaviate
+          - openGauss
         include:
           - library: pynndescent
             dataset: random-xs-16-hamming

ann_benchmarks/algorithms/openGauss/Dockerfile

@@ -1,14 +1,12 @@
 FROM ann-benchmarks
 
-# https://github.com/pgvector/pgvector/blob/master/Dockerfile
-
-RUN git clone https://github.com/pgvector/pgvector /tmp/pgvector
+RUN add-apt-repository ppa:opengauss/opengauss
 
 RUN DEBIAN_FRONTEND=noninteractive apt-get -y install tzdata
-RUN apt-get update && apt-get install -y --no-install-recommends build-essential postgresql-common
-RUN /usr/share/postgresql-common/pgdg/apt.postgresql.org.sh -y
-RUN apt-get install -y --no-install-recommends postgresql-16 postgresql-server-dev-16
-RUN sh -c 'echo "local all all trust" > /etc/postgresql/16/main/pg_hba.conf'
+RUN sed -i 's/bookworm/jammy/g' /etc/apt/sources.list.d/opengauss-ubuntu-opengauss-bookworm.list
+RUN apt-get update && apt-get install -y --no-install-recommends build-essential 
+RUN apt install opengauss
+RUN sh -c 'echo "local all all trust" > /var/lib/opengauss/data/pg_hba.conf'
 
 # Dynamically set OPTFLAGS based on the architecture
 RUN ARCH=$(uname -m) && \
@@ -24,16 +22,15 @@ RUN ARCH=$(uname -m) && \
     make OPTFLAGS="$OPTFLAGS" && \
     make install
 
-USER postgres
-RUN service postgresql start && \
-    psql -c "CREATE USER ann WITH ENCRYPTED PASSWORD 'ann'" && \
-    psql -c "CREATE DATABASE ann" && \
-    psql -c "GRANT ALL PRIVILEGES ON DATABASE ann TO ann" && \
-    psql -d ann -c "GRANT ALL ON SCHEMA public TO ann" && \
-    psql -d ann -c "CREATE EXTENSION vector" && \
-    psql -c "ALTER USER ann SET maintenance_work_mem = '4GB'" && \
-    psql -c "ALTER USER ann SET max_parallel_maintenance_workers = 0" && \
-    psql -c "ALTER SYSTEM SET shared_buffers = '4GB'"
+USER opengauss
+RUN service opengauss start && \
+    gsql -c "CREATE USER ann WITH ENCRYPTED PASSWORD 'ann'" && \
+    gsql -c "CREATE DATABASE ann" && \
+    gsql -c "GRANT ALL PRIVILEGES ON DATABASE ann TO ann" && \
+    gsql -d ann -c "GRANT ALL ON SCHEMA public TO ann" && \
+    gsql -c "ALTER USER ann SET maintenance_work_mem = '4GB'" && \
+    gsql -c "ALTER USER ann SET max_parallel_maintenance_workers = 0" && \
+    gsql -c "ALTER SYSTEM SET shared_buffers = '4GB'"
 USER root
 
-RUN pip install psycopg[binary] pgvector
+RUN pip install psycopg[binary] 

requirements.txt

@@ -10,6 +10,3 @@ jinja2==3.1.2
 pytest==7.2.2
 datasets==2.12.0
 requests==2.31.0
-psycopg
-psycopg-binary
-pgvector