forked from paradedb/paradedb
-
Notifications
You must be signed in to change notification settings - Fork 0
335 lines (293 loc) · 15.7 KB
/
test-pg_search.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
# workflows/test-pg_search.yml
#
# Test pg_search
# Run unit and integration tests for the pg_search extension.
name: Test pg_search
on:
pull_request:
types: [opened, synchronize, reopened, ready_for_review]
branches:
- dev
- main
paths:
- ".github/workflows/test-pg_search.yml"
- "pg_search/**"
- "!pg_search/README.md"
- "tokenizers/**"
- "shared/**"
push:
branches:
- dev # Run CI on dev. This is important to fill the GitHub Actions cache in a way that pull requests can see it
workflow_dispatch:
inputs:
test_upgrade_version:
description: "Upcoming pg_search version to test upgrading against"
required: false
default: ""
concurrency:
group: test-pg_search-${{ github.head_ref || github.ref }}
cancel-in-progress: true
jobs:
test-pg_search-postgres:
name: Test pg_search on PostgreSQL ${{ matrix.pg_version }} for ${{ matrix.arch }}
runs-on: ${{ matrix.runner }}
if: github.event.pull_request.draft == false
strategy:
matrix:
include:
- runner: depot-ubuntu-latest-8
pg_version: 12
arch: amd64
- runner: depot-ubuntu-latest-8
pg_version: 13
arch: amd64
- runner: depot-ubuntu-latest-8
pg_version: 14
arch: amd64
- runner: depot-ubuntu-latest-8
pg_version: 15
arch: amd64
- runner: depot-ubuntu-latest-8
pg_version: 16
arch: amd64
env:
default_pg_version: 16
steps:
# For the Rust cache to get filled, we need to run the CI on the dev branch after every merge. This only
# needs to happen once, so we skip the workflow for all but one of the matrix jobs in that case.
- name: Check if Skipping
id: check_skip
run: |
if [[ "${{ github.event_name }}" == "push" && "${{ matrix.pg_version }}" != "${{ env.default_pg_version }}" ]]; then
echo "This is a push event to fill Rust cache. Skipping this job."
echo "skip_remaining_steps=true" >> $GITHUB_OUTPUT
fi
- name: Checkout Git Repository
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
uses: actions/checkout@v4
- name: Install Rust
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
uses: dtolnay/rust-toolchain@stable
# This checks that the version in Cargo.toml is incremented to the next release. We only run it
# on PRs to main, which are our release promotion PRs.
- name: Check version in Cargo.toml
if: steps.check_skip.outputs.skip_remaining_steps != 'true' && github.base_ref == 'main'
working-directory: pg_search/
run: |
CARGO_VERSION=$(grep "^version" Cargo.toml | head -1 | awk -F '"' '{print $2}')
RELEASE_VERSION="${{ vars.VERSION_MAJOR }}.${{ vars.VERSION_MINOR }}.${{ vars.VERSION_PATCH }}"
if [ "$CARGO_VERSION" != "$RELEASE_VERSION" ]; then
echo "Version in Cargo.toml ($CARGO_VERSION) does not match upcoming release version ($RELEASE_VERSION), did you forget to increment it?"
exit 1
fi
echo "Version check passed!"
# Caches from base branches are available to PRs, but not across unrelated branches, so we only
# save the cache on the 'dev' branch, but load it on all branches.
- name: Install Rust Cache
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
uses: Swatinem/rust-cache@v2
with:
prefix-key: "v1"
shared-key: ${{ runner.os }}-rust-cache-pg_search-${{ HashFiles('Cargo.lock') }}
cache-targets: true
cache-on-failure: true
cache-all-crates: true
save-if: ${{ github.ref == 'refs/heads/dev' }}
- name: Install & Configure Supported PostgreSQL Version
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
run: |
wget --quiet -O - https://www.postgresql.org/media/keys/ACCC4CF8.asc | sudo apt-key add -
sudo sh -c 'echo "deb http://apt.postgresql.org/pub/repos/apt/ $(lsb_release -cs)-pgdg main" > /etc/apt/sources.list.d/pgdg.list'
sudo apt-get update && sudo apt-get install -y postgresql-${{ matrix.pg_version }} postgresql-server-dev-${{ matrix.pg_version }}
sudo chown -R $(whoami) /usr/share/postgresql/${{ matrix.pg_version }}/ /usr/lib/postgresql/${{ matrix.pg_version }}/ /var/lib/postgresql/${{ matrix.pg_version }}/
echo "/usr/lib/postgresql/${{ matrix.pg_version }}/bin" >> $GITHUB_PATH
# Needed for hybrid search unit tests
- name: Install pgvector
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
run: |
git clone --branch v0.7.4 https://github.com/pgvector/pgvector.git
cd pgvector/
sudo PG_CONFIG=/usr/lib/postgresql/${{ matrix.pg_version }}/bin/pg_config make -j
sudo PG_CONFIG=/usr/lib/postgresql/${{ matrix.pg_version }}/bin/pg_config make install -j
- name: Extract pgrx version
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
working-directory: pg_search/
run: echo "PGRX_VERSION=$(cargo tree --depth 1 -i pgrx -p pg_search | head -n 1 | cut -f2 -dv)" >> $GITHUB_ENV
- name: Install pgrx & llvm-tools-preview
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
run: |
cargo install -j $(nproc) --locked cargo-pgrx --version ${{ env.PGRX_VERSION }}
rustup component add llvm-tools-preview
cargo pgrx init "--pg${{ matrix.pg_version }}=/usr/lib/postgresql/${{ matrix.pg_version }}/bin/pg_config"
- name: Add pg_search to shared_preload_libraries
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
working-directory: /home/runner/.pgrx/data-${{ matrix.pg_version }}/
run: sed -i "s/^#shared_preload_libraries = .*/shared_preload_libraries = 'pg_search'/" postgresql.conf
# TODO: Uncomment the upgrade test after we release 0.10.0
# The integration tests also test upgrading the extension when passed the '-u' flag (only on promotion PRs)
- name: Run pg_search Cargo Test Suite
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
working-directory: pg_search/
run: |
# Variables (we disable telemetry to avoid skewing the user metrics with CI runs)
BASE_RELEASE="0.10.0"
PARADEDB_TELEMETRY=false
# echo ""
# echo "Enabling code coverage..."
# echo -e "\n# Enable code coverage on Linux only, for CI builds\n[target.'cfg(target_os=\"linux\")']\nrustflags = [\"-Cinstrument-coverage\"]" >> ../.cargo/config.toml
# mkdir -p ../target/coverage ../target/coverage-report
# # If this is a promotion PR, we test upgrading the extension by pulling the earliest version of the extension compatible with our current default Postgres version
# if [[ "${{ github.base_ref }}" == "main" && "${{ matrix.pg_version }}" == "${{ env.default_pg_version }}" ]]; then
# echo ""
# echo "Promotion PR detected! Running extension upgrade test..."
# if [[ "${{ github.event.inputs.test_upgrade_version }}" == "" ]]; then
# echo "Using the GitHub Variables version..."
# NEW_VERSION=${{ vars.VERSION_MAJOR }}.${{ vars.VERSION_MINOR }}.${{ vars.VERSION_PATCH }}
# else
# echo "Using the workflow_dispatch version..."
# NEW_VERSION="${{ github.event.inputs.test_upgrade_version }}"
# fi
# echo "Running extension upgrade test from v$BASE_RELEASE to v$NEW_VERSION..."
# echo ""
# echo "Downloading pg_search v$BASE_RELEASE..."
# curl -LOJ "https://github.com/paradedb/paradedb/releases/download/v$BASE_RELEASE/pg_search-v$BASE_RELEASE-pg${{ matrix.pg_version }}-${{ matrix.arch }}-ubuntu2204.deb"
# sudo dpkg -i "pg_search-v$BASE_RELEASE-pg${{ matrix.pg_version }}-${{ matrix.arch }}-ubuntu2204.deb"
# echo ""
# echo "Starting Postgres..."
# cargo pgrx start pg${{ matrix.pg_version }}
# echo ""
# echo "Loading pg_search v$BASE_RELEASE..."
# psql -h localhost -d postgres -p 288${{ matrix.pg_version }} -c "CREATE EXTENSION pg_search VERSION '$BASE_RELEASE';"
# echo ""
# echo "Verifying current pg_search installed version..."
# psql -h localhost -d postgres -p 288${{ matrix.pg_version }} -c "SELECT extname, extversion FROM pg_extension WHERE extname = 'pg_search';"
# echo ""
# echo "Building pg_search v$NEW_VERSION..."
# sudo chown -R "$(whoami)" "/usr/share/postgresql/${{ matrix.pg_version }}/extension/" "/usr/lib/postgresql/${{ matrix.pg_version }}/lib/"
# cargo pgrx install --features icu --pg-config="/usr/lib/postgresql/${{ matrix.pg_version }}/bin/pg_config" --release
# echo ""
# echo "Upgrading pg_search to v$NEW_VERSION..."
# psql -h localhost -d postgres -p 288${{ matrix.pg_version }} -c "ALTER EXTENSION pg_search UPDATE TO '$NEW_VERSION';"
# echo ""
# echo "Verifying current pg_search installed version..."
# psql -h localhost -d postgres -p 288${{ matrix.pg_version }} -c "SELECT extname, extversion FROM pg_extension WHERE extname = 'pg_search';"
# echo ""
# echo "Extension successfully upgraded from v$BASE_RELEASE to v$NEW_VERSION!"
# echo ""
# echo "Restarting Postgres..."
# cargo pgrx stop pg${{ matrix.pg_version }}
# cargo pgrx start pg${{ matrix.pg_version }}
# else
# echo ""
# echo "Building pg_search..."
# cargo pgrx install --features icu --pg-config="/usr/lib/postgresql/${{ matrix.pg_version }}/bin/pg_config" --release
# echo ""
# echo "Starting Postgres..."
# cargo pgrx start pg${{ matrix.pg_version }}
# fi
echo ""
echo "Building pg_search..."
cargo pgrx install --features icu --pg-config="/usr/lib/postgresql/${{ matrix.pg_version }}/bin/pg_config" --release
echo ""
echo "Starting Postgres..."
cargo pgrx start pg${{ matrix.pg_version }}
# Necessary for the ephemeral Postgres test to have proper permissions
sudo chown -R $(whoami) /var/run/postgresql/
echo ""
echo "Running Rust tests..."
export DATABASE_URL=postgresql://localhost:288${{ matrix.pg_version }}/postgres
export PG_CONFIG=/usr/lib/postgresql/${{ matrix.pg_version }}/bin/pg_config
cargo pgrx test "pg${{ matrix.pg_version }}" --features icu
cat ~/.pgrx/${{ matrix.pg_version}}.log
- name: Print the Postgres Logs
if: steps.check_skip.outputs.skip_remaining_steps != 'true' && always()
run: cat ~/.pgrx/${{ matrix.pg_version}}.log
test-pg_search-pgrx-postgres:
name: Test pg_search on pgrx PostgreSQL ${{ matrix.pg_version }} for ${{ matrix.arch }}
runs-on: ${{ matrix.runner }}
if: github.event.pull_request.draft == false
strategy:
matrix:
include:
- runner: depot-ubuntu-latest-8
pg_version: 16
arch: amd64
env:
default_pg_version: 16
steps:
# For the Rust cache to get filled, we need to run the CI on the dev branch after every merge. This only
# needs to happen once, so we skip the workflow for all but one of the matrix jobs in that case.
- name: Check if Skipping
id: check_skip
run: |
if [[ "${{ github.event_name }}" == "push" && "${{ matrix.pg_version }}" != "${{ env.default_pg_version }}" ]]; then
echo "This is a push event to fill Rust cache. Skipping this job."
echo "skip_remaining_steps=true" >> $GITHUB_OUTPUT
fi
- name: Checkout Git Repository
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
uses: actions/checkout@v4
- name: Install Rust
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
uses: dtolnay/rust-toolchain@stable
# This checks that the version in Cargo.toml is incremented to the next release. We only run it
# on PRs to main, which are our release promotion PRs.
- name: Check version in Cargo.toml
if: steps.check_skip.outputs.skip_remaining_steps != 'true' && github.base_ref == 'main'
working-directory: pg_search/
run: |
CARGO_VERSION=$(grep "^version" Cargo.toml | head -1 | awk -F '"' '{print $2}')
RELEASE_VERSION="${{ vars.VERSION_MAJOR }}.${{ vars.VERSION_MINOR }}.${{ vars.VERSION_PATCH }}"
if [ "$CARGO_VERSION" != "$RELEASE_VERSION" ]; then
echo "Version in Cargo.toml ($CARGO_VERSION) does not match upcoming release version ($RELEASE_VERSION), did you forget to increment it?"
exit 1
fi
echo "Version check passed!"
# Caches from base branches are available to PRs, but not across unrelated branches, so we only
# save the cache on the 'dev' branch, but load it on all branches.
- name: Install Rust Cache
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
uses: Swatinem/rust-cache@v2
with:
prefix-key: "v1"
shared-key: ${{ runner.os }}-rust-cache-pg_search--enable-cassert${{ HashFiles('Cargo.lock') }}
cache-targets: true
cache-on-failure: true
cache-all-crates: true
save-if: ${{ github.ref == 'refs/heads/dev' }}
- name: Extract pgrx version
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
working-directory: pg_search/
run: echo "PGRX_VERSION=$(grep '^pgrx = ' Cargo.toml | sed -E 's/pgrx = "(.*)"/\1/')" >> $GITHUB_ENV
- name: Install pgrx & llvm-tools-preview
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
run: |
cargo install -j $(nproc) --locked cargo-pgrx --version ${{ env.PGRX_VERSION }}
rustup component add llvm-tools-preview
cargo pgrx init "--pg${{ matrix.pg_version }}=download"
# Needed for hybrid search unit tests
- name: Install pgvector
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
run: |
git clone --branch v0.7.4 https://github.com/pgvector/pgvector.git
cd pgvector/
PG_CONFIG=~/.pgrx/16.*/pgrx-install/bin/pg_config make -j
PG_CONFIG=~/.pgrx/16.*/pgrx-install/bin/pg_config make install -j
- name: Add pg_search to shared_preload_libraries
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
working-directory: /home/runner/.pgrx/data-${{ matrix.pg_version }}/
run: sed -i "s/^#shared_preload_libraries = .*/shared_preload_libraries = 'pg_search'/" postgresql.conf
- name: Run pg_search Cargo Test Suite Against pgrx Postgres
if: steps.check_skip.outputs.skip_remaining_steps != 'true'
working-directory: pg_search/
run: |
PARADEDB_TELEMETRY=false
cargo pgrx stop
cargo pgrx install --pg-config ~/.pgrx/${{ matrix.pg_version }}.*/pgrx-install/bin/pg_config --features=icu
cargo pgrx start
~/.pgrx/${{ matrix.pg_version }}.*/pgrx-install/bin/createdb -p 28816 -h localhost pg_search
DATABASE_URL=postgresql://localhost:28816/pg_search cargo test --features=pg${{ matrix.pg_version }},icu -- --skip replication --skip ephemeral
cat ~/.pgrx/${{ matrix.pg_version}}.log
- name: Print the Postgres Logs
if: steps.check_skip.outputs.skip_remaining_steps != 'true' && always()
run: cat ~/.pgrx/${{ matrix.pg_version}}.log