262 lines
11 KiB
JavaScript
262 lines
11 KiB
JavaScript
const Docker = require('dockerode');
|
||
const fs = require('fs-extra');
|
||
const { execSshCommand, execScpCommand } = require('./sshUtils');
|
||
const log = require('./logger');
|
||
|
||
// Инициализируем Docker клиент через socket
|
||
const docker = new Docker({ socketPath: '/var/run/docker.sock' });
|
||
|
||
/**
|
||
* Экспорт Docker образов и данных с локальной машины
|
||
*/
|
||
const exportDockerImages = async (sendWebSocketLog) => {
|
||
log.info('Экспорт Docker образов и данных с хоста...');
|
||
sendWebSocketLog('info', '📦 Начинаем экспорт Docker образов...', 'export_images', 60);
|
||
|
||
const images = [
|
||
{ name: 'postgres:16', file: 'dapp-postgres.tar' },
|
||
{ name: 'digital_legal_entitydle-ollama:latest', file: 'dapp-ollama.tar' },
|
||
{ name: 'digital_legal_entitydle-vector-search:latest', file: 'dapp-vector-search.tar' },
|
||
{ name: 'digital_legal_entitydle-backend:latest', file: 'dapp-backend.tar' },
|
||
{ name: 'digital_legal_entitydle-frontend:latest', file: 'dapp-frontend.tar' },
|
||
{ name: 'digital_legal_entitydle-frontend-nginx:latest', file: 'dapp-frontend-nginx.tar' },
|
||
{ name: 'digital_legal_entitydle-webssh-agent:latest', file: 'dapp-webssh-agent.tar' }
|
||
];
|
||
|
||
// Экспортируем все образы
|
||
for (let i = 0; i < images.length; i++) {
|
||
const image = images[i];
|
||
const progress = 60 + Math.floor((i / images.length) * 10); // 60-70%
|
||
sendWebSocketLog('info', `📦 Экспорт образа: ${image.name}`, 'export_images', progress);
|
||
|
||
try {
|
||
const dockerImage = docker.getImage(image.name);
|
||
const stream = await dockerImage.get();
|
||
const outputPath = `/tmp/${image.file}`;
|
||
|
||
// Сохраняем stream в файл
|
||
await new Promise((resolve, reject) => {
|
||
const writeStream = fs.createWriteStream(outputPath);
|
||
stream.pipe(writeStream);
|
||
stream.on('end', () => {
|
||
sendWebSocketLog('success', `✅ Экспорт ${image.name} завершен`, 'export_images', progress);
|
||
resolve();
|
||
});
|
||
stream.on('error', reject);
|
||
writeStream.on('error', reject);
|
||
});
|
||
} catch (error) {
|
||
log.error(`Ошибка экспорта ${image.name}: ${error.message}`);
|
||
sendWebSocketLog('error', `❌ Ошибка экспорта ${image.name}`, 'export_images', progress);
|
||
}
|
||
}
|
||
|
||
// Экспортируем данные из volumes
|
||
log.info('Экспорт данных из Docker volumes...');
|
||
sendWebSocketLog('info', '📦 Экспорт данных из Docker volumes...', 'export_data', 70);
|
||
|
||
// PostgreSQL данные
|
||
sendWebSocketLog('info', '📦 Экспорт данных PostgreSQL...', 'export_data', 72);
|
||
await exportVolumeData('digital_legal_entitydle_postgres_data', 'postgres_data.tar.gz', sendWebSocketLog, 72);
|
||
|
||
// Ollama данные
|
||
sendWebSocketLog('info', '📦 Экспорт данных Ollama...', 'export_data', 75);
|
||
await exportVolumeData('digital_legal_entitydle_ollama_data', 'ollama_data.tar.gz', sendWebSocketLog, 75);
|
||
|
||
// Vector Search данные
|
||
sendWebSocketLog('info', '📦 Экспорт данных Vector Search...', 'export_data', 78);
|
||
await exportVolumeData('digital_legal_entitydle_vector_search_data', 'vector_search_data.tar.gz', sendWebSocketLog, 78);
|
||
|
||
// Создаем архив с ВСЕМИ образами и данными приложения
|
||
log.info('Создание архива Docker образов и данных на хосте...');
|
||
sendWebSocketLog('info', '📦 Создание архива всех данных...', 'export_data', 80);
|
||
|
||
try {
|
||
const tarFiles = images.map(img => img.file).join(' ');
|
||
const dataFiles = 'postgres_data.tar.gz ollama_data.tar.gz vector_search_data.tar.gz';
|
||
|
||
// Создаем архив через временный контейнер
|
||
const container = await docker.createContainer({
|
||
Image: 'alpine',
|
||
Cmd: ['sh', '-c', `cd /tmp && tar -czf docker-images-and-data.tar.gz ${tarFiles} ${dataFiles}`],
|
||
HostConfig: {
|
||
Binds: ['/tmp:/tmp'],
|
||
AutoRemove: true
|
||
}
|
||
});
|
||
|
||
await container.start();
|
||
await container.wait();
|
||
|
||
sendWebSocketLog('success', '✅ Архив создан успешно', 'export_data', 80);
|
||
} catch (error) {
|
||
log.error('Ошибка создания архива: ' + error.message);
|
||
sendWebSocketLog('error', '❌ Ошибка создания архива', 'export_data', 80);
|
||
}
|
||
|
||
log.success('Docker образы и данные успешно экспортированы');
|
||
sendWebSocketLog('success', '✅ Экспорт данных завершен', 'export_data', 80);
|
||
};
|
||
|
||
/**
|
||
* Вспомогательная функция для экспорта данных volume
|
||
*/
|
||
const exportVolumeData = async (volumeName, outputFile, sendWebSocketLog, progress) => {
|
||
try {
|
||
const container = await docker.createContainer({
|
||
Image: 'alpine',
|
||
Cmd: ['tar', 'czf', `/backup/${outputFile}`, '-C', '/data', '.'],
|
||
HostConfig: {
|
||
Binds: [
|
||
`${volumeName}:/data:ro`,
|
||
'/tmp:/backup'
|
||
],
|
||
AutoRemove: true
|
||
}
|
||
});
|
||
|
||
await container.start();
|
||
await container.wait();
|
||
|
||
sendWebSocketLog('success', `✅ Экспорт ${outputFile} завершен`, 'export_data', progress);
|
||
} catch (error) {
|
||
log.error(`Ошибка экспорта ${volumeName}: ${error.message}`);
|
||
sendWebSocketLog('error', `❌ Ошибка экспорта ${volumeName}`, 'export_data', progress);
|
||
}
|
||
};
|
||
|
||
/**
|
||
* Передача Docker образов и данных на VDS
|
||
*/
|
||
const transferDockerImages = async (options, sendWebSocketLog) => {
|
||
const { dockerUser } = options;
|
||
|
||
log.info('Передача Docker образов и данных на VDS...');
|
||
sendWebSocketLog('info', '📤 Передача архива на VDS сервер...', 'transfer', 82);
|
||
|
||
// Передаем архив образов и данных на VDS через SCP
|
||
await execScpCommand(
|
||
'/tmp/docker-images-and-data.tar.gz',
|
||
`/home/${dockerUser}/dapp/docker-images-and-data.tar.gz`,
|
||
options
|
||
);
|
||
|
||
sendWebSocketLog('success', '✅ Архив успешно передан на VDS', 'transfer', 85);
|
||
|
||
log.success('Docker образы и данные успешно переданы на VDS');
|
||
};
|
||
|
||
/**
|
||
* Импорт Docker образов и данных на VDS
|
||
*/
|
||
const importDockerImages = async (options, sendWebSocketLog) => {
|
||
const { dockerUser } = options;
|
||
|
||
// Создаем скрипт импорта на VDS
|
||
sendWebSocketLog('info', '📥 Начинаем импорт данных на VDS...', 'import', 85);
|
||
|
||
const importScript = `#!/bin/bash
|
||
set -e
|
||
echo "🚀 Импорт Docker образов и данных на VDS..."
|
||
|
||
# Проверяем наличие архива
|
||
if [ ! -f "./docker-images-and-data.tar.gz" ]; then
|
||
echo "❌ Файл docker-images-and-data.tar.gz не найден!"
|
||
exit 1
|
||
fi
|
||
|
||
# Создаем директорию для распаковки
|
||
mkdir -p ./temp-import
|
||
|
||
# Распаковываем архив
|
||
echo "📦 Распаковка архива..."
|
||
tar -xzf ./docker-images-and-data.tar.gz -C ./temp-import
|
||
|
||
# Импортируем ВСЕ образы приложения
|
||
echo "📦 Импорт образа postgres..."
|
||
docker load -i ./temp-import/dapp-postgres.tar
|
||
|
||
echo "📦 Импорт образа ollama..."
|
||
docker load -i ./temp-import/dapp-ollama.tar
|
||
|
||
echo "📦 Импорт образа vector-search..."
|
||
docker load -i ./temp-import/dapp-vector-search.tar
|
||
|
||
echo "📦 Импорт образа backend..."
|
||
docker load -i ./temp-import/dapp-backend.tar
|
||
|
||
echo "📦 Импорт образа frontend..."
|
||
docker load -i ./temp-import/dapp-frontend.tar
|
||
|
||
echo "📦 Импорт образа frontend-nginx..."
|
||
docker load -i ./temp-import/dapp-frontend-nginx.tar
|
||
|
||
echo "📦 Импорт образа webssh-agent..."
|
||
docker load -i ./temp-import/dapp-webssh-agent.tar
|
||
|
||
# 🆕 Импортируем данные в volumes с правильными именами для соответствия docker-compose
|
||
echo "📦 Импорт данных PostgreSQL..."
|
||
# Удаляем старый volume если существует
|
||
docker volume rm dapp_postgres_data 2>/dev/null || true
|
||
docker volume create dapp_postgres_data
|
||
docker run --rm -v dapp_postgres_data:/data -v ./temp-import:/backup alpine tar xzf /backup/postgres_data.tar.gz -C /data
|
||
|
||
echo "📦 Импорт данных Ollama..."
|
||
# Удаляем старый volume если существует
|
||
docker volume rm dapp_ollama_data 2>/dev/null || true
|
||
docker volume create dapp_ollama_data
|
||
docker run --rm -v dapp_ollama_data:/data -v ./temp-import:/backup alpine tar xzf /backup/ollama_data.tar.gz -C /data
|
||
|
||
echo "📦 Импорт данных Vector Search..."
|
||
# Удаляем старый volume если существует
|
||
docker volume rm dapp_vector_search_data 2>/dev/null || true
|
||
docker volume create dapp_vector_search_data
|
||
docker run --rm -v dapp_vector_search_data:/data -v ./temp-import:/backup alpine tar xzf /backup/vector_search_data.tar.gz -C /data
|
||
|
||
# Очищаем временные файлы
|
||
rm -rf ./temp-import
|
||
|
||
echo "✅ Образы и данные успешно импортированы!"
|
||
echo "📋 Доступные образы:"
|
||
docker images | grep -E "digital_legal_entitydle|postgres"
|
||
echo "📋 Доступные volumes:"
|
||
docker volume ls | grep dapp_`;
|
||
|
||
await execSshCommand(`echo '${importScript}' | sudo tee /home/${dockerUser}/dapp/import-images-and-data.sh`, options);
|
||
await execSshCommand(`sudo chmod +x /home/${dockerUser}/dapp/import-images-and-data.sh`, options);
|
||
|
||
// Импортируем образы и данные
|
||
log.info('Импорт Docker образов и данных...');
|
||
sendWebSocketLog('info', '📥 Импорт Docker образов на VDS...', 'import', 87);
|
||
await execSshCommand(`cd /home/${dockerUser}/dapp && ./import-images-and-data.sh`, options);
|
||
sendWebSocketLog('success', '✅ Импорт Docker образов завершен', 'import', 90);
|
||
|
||
sendWebSocketLog('info', '📥 Импорт данных в volumes...', 'import', 92);
|
||
// Логи от самого скрипта импорта будут видны
|
||
sendWebSocketLog('success', '✅ Импорт данных завершен', 'import', 95);
|
||
|
||
log.success('Docker образы и данные успешно импортированы на VDS');
|
||
};
|
||
|
||
/**
|
||
* Очистка временных файлов на локальной машине
|
||
*/
|
||
const cleanupLocalFiles = async () => {
|
||
log.info('Очистка временных файлов на хосте...');
|
||
try {
|
||
await fs.remove('/tmp/dapp-*.tar');
|
||
await fs.remove('/tmp/*_data.tar.gz');
|
||
await fs.remove('/tmp/docker-images-and-data.tar.gz');
|
||
log.success('Временные файлы очищены');
|
||
} catch (error) {
|
||
log.error('Ошибка очистки файлов: ' + error.message);
|
||
}
|
||
};
|
||
|
||
module.exports = {
|
||
exportDockerImages,
|
||
transferDockerImages,
|
||
importDockerImages,
|
||
cleanupLocalFiles
|
||
};
|