Skip to content

feat: support Arrow FlightSQL #275

feat: support Arrow FlightSQL

feat: support Arrow FlightSQL #275

Workflow file for this run

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