90 lines
3.5 KiB
Bash
90 lines
3.5 KiB
Bash
#!/bin/bash
|
||
# =============================================================================
|
||
# 步骤 2b：迁移后重建 Neo4j 索引（向量索引 + 全文索引 + 混合索引）
|
||
# 在【迁移目标机】上执行本脚本
|
||
# 必须在 02_load_neo4j.sh 或 02_load_neo4j_by_ship.sh 之后执行
|
||
# 配置项请修改 migration_config.sh
|
||
# =============================================================================
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
source "${SCRIPT_DIR}/migration_config.sh"
|
||
|
||
set -e
|
||
|
||
INDEXING_URL="http://localhost:${DST_KGRAG_PORT}/neo4j_indexing"
|
||
|
||
echo "=========================================="
|
||
echo " kgrag 索引重建脚本 - 步骤 2b"
|
||
echo " 接口地址: ${INDEXING_URL}"
|
||
echo " Neo4j: ${NEO4J_URI}"
|
||
echo "=========================================="
|
||
echo ""
|
||
echo "正在调用索引重建接口（可能需要几分钟）..."
|
||
echo ""
|
||
|
||
RESPONSE=$(curl -s -w "\n%{http_code}" -X POST "${INDEXING_URL}" \
|
||
-H "Content-Type: application/json" \
|
||
-d "{
|
||
\"neo4j_uri\": \"${NEO4J_URI}\",
|
||
\"neo4j_username\": \"${NEO4J_USER}\",
|
||
\"neo4j_password\": \"${NEO4J_PASSWORD}\",
|
||
\"force_refresh\": true
|
||
}" \
|
||
--max-time 300)
|
||
|
||
HTTP_CODE=$(echo "$RESPONSE" | tail -n1)
|
||
BODY=$(echo "$RESPONSE" | sed '$d')
|
||
|
||
echo "HTTP 状态码: ${HTTP_CODE}"
|
||
echo "返回内容: ${BODY}"
|
||
|
||
if [ "$HTTP_CODE" = "409" ]; then
|
||
echo " [WARN] 已有索引任务在执行（409），跳过接口调用，直接走兜底 cypher"
|
||
elif [ "$HTTP_CODE" != "200" ]; then
|
||
echo ""
|
||
echo " [ERROR] 接口调用失败，HTTP ${HTTP_CODE}"
|
||
echo " 请确认 kgrag 容器已启动，端口 ${DST_KGRAG_PORT} 可访问"
|
||
exit 1
|
||
fi
|
||
|
||
# 检查 hybrid_index_status（HTTP 200 但混合索引可能静默失败返回 207）
|
||
HYBRID_STATUS=$(echo "$BODY" | python3 -c "import sys,json; d=json.load(sys.stdin); print(d.get('data',{}).get('hybrid_index_status','unknown'))" 2>/dev/null || echo "unknown")
|
||
if [ "$HYBRID_STATUS" != "success" ]; then
|
||
echo " [WARN] hybrid_index_status=${HYBRID_STATUS}，混合索引未由接口建立，将由下方兜底命令补建"
|
||
else
|
||
echo " [OK] /neo4j_indexing 全部成功（含混合索引）"
|
||
fi
|
||
|
||
# ----------------------------------------------------------
|
||
# 补建混合索引（build_hybrid_indexes 可能静默失败，在此兜底）
|
||
# 对应 neo4j_indexing.py 中 VECTOR_INDEX_NAME / FULLTEXT_INDEX_NAME /
|
||
# FULLTEXT_FIELD_INDEX_NAME 三个常量
|
||
# ----------------------------------------------------------
|
||
echo ""
|
||
echo "补建混合索引（global_searchable_* 系列）..."
|
||
|
||
NEO4J_CONTAINER="neo4j"
|
||
NEO4J_PASS="${NEO4J_PASSWORD}"
|
||
|
||
docker exec "${NEO4J_CONTAINER}" cypher-shell \
|
||
-u "${NEO4J_USER}" -p "${NEO4J_PASS}" \
|
||
"CREATE VECTOR INDEX global_searchable_embedding IF NOT EXISTS
|
||
FOR (n:Searchable) ON (n.embedding)
|
||
OPTIONS {indexConfig: {\`vector.dimensions\`: 1024, \`vector.similarity_function\`: 'cosine'}}"
|
||
|
||
docker exec "${NEO4J_CONTAINER}" cypher-shell \
|
||
-u "${NEO4J_USER}" -p "${NEO4J_PASS}" \
|
||
"CREATE FULLTEXT INDEX global_searchable_content_search IF NOT EXISTS
|
||
FOR (n:Searchable) ON EACH [n.\`名称\`]"
|
||
|
||
docker exec "${NEO4J_CONTAINER}" cypher-shell \
|
||
-u "${NEO4J_USER}" -p "${NEO4J_PASS}" \
|
||
"CREATE FULLTEXT INDEX global_searchable_fulltext_search IF NOT EXISTS
|
||
FOR (n:Searchable) ON EACH [n.fulltext]"
|
||
|
||
echo ""
|
||
echo "=========================================="
|
||
echo " 索引重建完成（含混合索引兜底）！"
|
||
echo " 下一步：执行 python 03_update_neo4j_urls.py"
|
||
echo "=========================================="
|