Доавление векторного поиска и репликации

2026-02-04 04:31:09 +00:00 · 2026-01-29 00:42:30 +03:00
parent 09d5739256
commit 657f1b96f2
19 changed files with 448 additions and 71 deletions
@@ -0,0 +1,77 @@
+#!/bin/sh
+set -e
+
+echo "=== Настройка репликации ==="
+echo "Этот узел: NODE_ID=${NODE_ID}"
+echo "Удаленный хост: ${REMOTE_HOST}"
+
+echo "Ждем локальную базу..."
+sleep 10
+
+until PGPASSWORD="${POSTGRES_PASSWORD}" psql -h db -U "${POSTGRES_USER}" -d "${POSTGRES_DB}" -c '\q' 2>/dev/null; do
+    echo "Локальная база не готова, ждем..."
+    sleep 2
+done
+echo "Локальная база готова"
+
+echo "Настройка генераторов ID (NODE_ID=${NODE_ID})..."
+
+PGPASSWORD="${POSTGRES_PASSWORD}" psql -h db -U "${POSTGRES_USER}" -d "${POSTGRES_DB}" <<EOF
+DO \$\$
+DECLARE
+    r RECORD;
+BEGIN
+    FOR r IN
+        SELECT table_schema, table_name, column_name
+        FROM information_schema.columns
+        WHERE is_identity = 'YES'
+          AND table_schema = 'public'
+    LOOP
+        EXECUTE format(
+            'ALTER TABLE %I.%I ALTER COLUMN %I SET GENERATED BY DEFAULT AS IDENTITY (START WITH %s INCREMENT BY 2)',
+            r.table_schema, r.table_name, r.column_name, ${NODE_ID}
+        );
+        RAISE NOTICE 'Настроен ID для %.%', r.table_name, r.column_name;
+    END LOOP;
+END \$\$;
+EOF
+
+echo "Ждем удаленный хост ${REMOTE_HOST}:${REMOTE_PORT}..."
+TIMEOUT=300
+ELAPSED=0
+
+while ! PGPASSWORD="${POSTGRES_PASSWORD}" psql -h "${REMOTE_HOST}" -p ${REMOTE_PORT} -U "${POSTGRES_USER}" -d "${POSTGRES_DB}" -c '\q' 2>/dev/null; do
+    sleep 5
+    ELAPSED=$((ELAPSED + 5))
+    if [ $ELAPSED -ge $TIMEOUT ]; then
+        echo "Таймаут ожидания удаленного хоста. Репликация НЕ настроена."
+        echo "Вы можете запустить этот скрипт вручную позже:"
+        echo "docker compose restart replication-setup"
+        exit 0
+    fi
+    echo "Удаленный хост недоступен, ждем... (${ELAPSED}s/${TIMEOUT}s)"
+done
+echo "Удаленный хост доступен"
+
+EXISTING=$(PGPASSWORD="${POSTGRES_PASSWORD}" psql -h db -U "${POSTGRES_USER}" -d "${POSTGRES_DB}" -tAc "SELECT COUNT(*) FROM pg_subscription WHERE subname = 'sub_from_remote';")
+
+if [ "$EXISTING" -gt 0 ]; then
+    echo "Подписка уже существует, пропускаем создание"
+else
+    echo "Создаем подписку на удаленный хост..."
+
+    PGPASSWORD="${POSTGRES_PASSWORD}" psql -h db -U "${POSTGRES_USER}" -d "${POSTGRES_DB}" <<EOF
+CREATE SUBSCRIPTION sub_from_remote
+CONNECTION 'host=${REMOTE_HOST} port=${REMOTE_PORT} user=${POSTGRES_USER} password=${POSTGRES_PASSWORD} dbname=${POSTGRES_DB}'
+PUBLICATION all_tables_pub
+WITH (
+    origin = none
+);
+EOF
+
+    echo "Подписка создана!"
+fi
+
+echo ""
+echo "Репликация настроена!"
+echo "Этот узел (${NODE_ID}) теперь синхронизирован с ${REMOTE_HOST}"