feat: support Arrow FlightSQL #275
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: PostgreSQL Replication E2E Test | |
on: | |
push: | |
branches: [ "main" ] | |
pull_request: | |
branches: [ "main" ] | |
jobs: | |
replication-test: | |
runs-on: ubuntu-latest | |
steps: | |
- uses: actions/checkout@v4 | |
- name: Set up Go | |
uses: actions/setup-go@v5 | |
with: | |
go-version: '1.23' | |
- name: Set up Python | |
uses: actions/setup-python@v5 | |
with: | |
python-version: '3.10' | |
- name: Install dependencies | |
run: | | |
go get . | |
pip3 install "sqlglot[rs]" pyarrow pandas | |
sudo apt-get update | |
sudo apt-get install --yes --no-install-recommends postgresql-client | |
# Install DuckDB CLI and extensions | |
curl -LJO https://github.com/duckdb/duckdb/releases/download/v1.1.3/duckdb_cli-linux-amd64.zip | |
unzip duckdb_cli-linux-amd64.zip | |
chmod +x duckdb | |
sudo mv duckdb /usr/local/bin | |
duckdb -c 'INSTALL json from core' | |
- name: Build MyDuck | |
run: go build -v | |
- name: Start Primary PostgreSQL | |
run: | | |
docker run -d --name pg-primary \ | |
-e POSTGRES_USER=postgres \ | |
-e POSTGRES_PASSWORD=password \ | |
-e POSTGRES_DB=testdb \ | |
-p 15432:5432 \ | |
postgres:latest \ | |
-c wal_level=logical \ | |
-c max_wal_senders=30 | |
# Wait for PostgreSQL to be ready | |
for i in {1..30}; do | |
if psql "postgres://postgres:password@localhost:15432/testdb" -c "SELECT 1" >/dev/null 2>&1; then | |
break | |
fi | |
sleep 1 | |
done | |
- name: Configure Primary for Replication | |
run: | | |
# Create test data | |
psql "postgres://postgres:password@localhost:15432/testdb" <<-EOSQL | |
CREATE TABLE test_table (id int primary key, name text); | |
INSERT INTO test_table VALUES (1, 'initial data 1'), (2, 'initial data 2'); | |
CREATE PUBLICATION testpub FOR TABLE test_table; | |
EOSQL | |
- name: Start MyDuck and Test Initial Replication | |
run: | | |
# Start MyDuck | |
./myduckserver & | |
sleep 5 | |
# Create subscription | |
psql -h 127.0.0.1 -p 5432 -U postgres <<-EOSQL | |
CREATE SUBSCRIPTION testsub | |
CONNECTION 'dbname=testdb host=localhost port=15432 user=postgres password=password' | |
PUBLICATION testpub; | |
EOSQL | |
sleep 5 | |
# Verify initial data | |
psql -h 127.0.0.1 -p 5432 -U postgres -c "SELECT 1 FROM test_table WHERE id = 1 AND name = 'initial data 1';" | grep -q 1 | |
psql -h 127.0.0.1 -p 5432 -U postgres -c "SELECT 1 FROM test_table WHERE id = 2 AND name = 'initial data 2';" | grep -q 1 | |
- name: Test Ongoing Replication | |
run: | | |
# Insert new data in primary | |
psql "postgres://postgres:password@localhost:15432/testdb" \ | |
-c "INSERT INTO test_table VALUES (3, 'new data 3');" | |
sleep 2 | |
# Verify replication of new data | |
psql -h 127.0.0.1 -p 5432 -U postgres -c "SELECT 1 FROM test_table WHERE id = 3 AND name = 'new data 3';" | grep -q 1 | |
# Kill MyDuck | |
pkill myduckserver | |
sleep 2 | |
- name: Test Replication Recovery | |
run: | | |
# Insert more data in primary while MyDuck is down | |
psql "postgres://postgres:password@localhost:15432/testdb" \ | |
-c "INSERT INTO test_table VALUES (4, 'offline data 4');" | |
# Restart MyDuck | |
./myduckserver & | |
sleep 5 | |
# Verify replication catches up | |
psql -h 127.0.0.1 -p 5432 -U postgres -c "SELECT 1 FROM test_table WHERE id = 4 AND name = 'offline data 4';" | grep -q 1 | |
# Kill MyDuck | |
pkill myduckserver | |
- name: Cleanup | |
if: always() | |
run: | | |
pkill myduckserver || true | |
docker rm -f pg-primary || true |