96 lines
2.7 KiB
Bash
Executable File
96 lines
2.7 KiB
Bash
Executable File
#!/bin/bash
|
||
# 织忆恢复脚本 — 从备份恢复数据
|
||
# 用法: ./restore.sh [backup_dir]
|
||
# backup_dir: 可选，如不填则列出可用备份供选择
|
||
|
||
set -e
|
||
|
||
BACKUP_ROOT="/home/muc/backups/memoryweave"
|
||
DATA_DIR="/var/lib/memoryweave"
|
||
STAMP=$(date +%Y%m%d-%H%M%S)
|
||
|
||
if [ -n "$1" ]; then
|
||
BACKUP_DIR="$1"
|
||
else
|
||
echo "=== 可用备份 ==="
|
||
backups=($(ls -t "$BACKUP_ROOT"/))
|
||
if [ ${#backups[@]} -eq 0 ]; then
|
||
echo "错误: 没有找到任何备份"
|
||
exit 1
|
||
fi
|
||
for i in "${!backups[@]}"; do
|
||
echo " [$i] ${backups[$i]}"
|
||
done
|
||
echo ""
|
||
read -p "选择备份 [0-${#backups[@]}-1]: " idx
|
||
BACKUP_DIR="$BACKUP_ROOT/${backups[$idx]}"
|
||
fi
|
||
|
||
if [ ! -d "$BACKUP_DIR" ]; then
|
||
echo "错误: 备份目录不存在: $BACKUP_DIR"
|
||
exit 1
|
||
fi
|
||
|
||
echo "[$STAMP] 织忆恢复开始..."
|
||
echo "[$STAMP] 使用备份: $BACKUP_DIR"
|
||
read -p "即将停止 zhiyid 并恢复数据，确认? [y/N]: " confirm
|
||
[ "$confirm" = "y" ] || { echo "取消"; exit 0; }
|
||
|
||
# 停止服务
|
||
echo "[$STAMP] 停止 zhiyid..."
|
||
systemctl --user stop zhiyid 2>/dev/null || sudo systemctl stop zhiyid 2>/dev/null || true
|
||
|
||
# 备份当前数据（出事前的快照）
|
||
CURRENT_BACKUP="$DATA_DIR/pre-restore-$STAMP"
|
||
echo "[$STAMP] 备份当前数据到: $CURRENT_BACKUP"
|
||
mkdir -p "$CURRENT_BACKUP"
|
||
cp -r "$DATA_DIR"/*.db "$DATA_DIR"/*.lance "$CURRENT_BACKUP/" 2>/dev/null || true
|
||
|
||
# 恢复 SQLite
|
||
for db in graph.db memoryweave.db; do
|
||
echo "[$STAMP] 恢复 $db..."
|
||
if [ -f "$BACKUP_DIR/$db" ]; then
|
||
cp "$BACKUP_DIR/$db" "$DATA_DIR/$db"
|
||
else
|
||
echo "[$STAMP] 警告: 备份中无 $db，跳过"
|
||
fi
|
||
done
|
||
|
||
# 恢复 LanceDB
|
||
echo "[$STAMP] 恢复 LanceDB 数据..."
|
||
if [ -f "$BACKUP_DIR/lance-data.tar.gz" ]; then
|
||
tar xzf "$BACKUP_DIR/lance-data.tar.gz" -C "$DATA_DIR"
|
||
else
|
||
echo "[$STAMP] 警告: 备份中无 lance-data.tar.gz，跳过"
|
||
fi
|
||
|
||
# 验证
|
||
echo "[$STAMP] 验证恢复..."
|
||
ok=true
|
||
for db in graph.db memoryweave.db; do
|
||
if [ -s "$DATA_DIR/$db" ]; then
|
||
echo "[$STAMP] $db: OK ($(du -sh "$DATA_DIR/$db" | cut -f1))"
|
||
else
|
||
echo "[$STAMP] $db: 失败"
|
||
ok=false
|
||
fi
|
||
done
|
||
for lance in episodes.lance memories.lance tombstones.lance; do
|
||
if [ -s "$DATA_DIR/$lance" ]; then
|
||
echo "[$STAMP] $lance: OK"
|
||
fi
|
||
done
|
||
|
||
# 重启
|
||
echo "[$STAMP] 重启 zhiyid..."
|
||
systemctl --user start zhiyid 2>/dev/null || sudo systemctl start zhiyid 2>/dev/null || true
|
||
sleep 2
|
||
|
||
# 健康检查
|
||
if curl -sf http://127.0.0.1:7821/health > /dev/null 2>&1; then
|
||
echo "[$STAMP] ✅ 恢复成功，zhiyid 健康检查通过"
|
||
else
|
||
echo "[$STAMP] ⚠️ zhiyid 未响应，请手动检查"
|
||
fi
|
||
|
||
echo "[$STAMP] 完成。当前数据快照已保存到: $CURRENT_BACKUP" |