54 lines
1.6 KiB
JavaScript
54 lines
1.6 KiB
JavaScript
const { log } = require('./logger');
|
||
const { config } = require('./config');
|
||
|
||
function httpHealthCheck(host, port, path = '/health', timeout = 5000) {
|
||
return new Promise((resolve) => {
|
||
const http = require('http');
|
||
const req = http.get({ hostname: host, port, path, timeout }, (res) => {
|
||
let body = '';
|
||
res.on('data', (chunk) => { body += chunk; });
|
||
res.on('end', () => {
|
||
resolve({ success: res.statusCode >= 200 && res.statusCode < 400, statusCode: res.statusCode });
|
||
});
|
||
});
|
||
req.on('error', (err) => {
|
||
resolve({ success: false, error: err.message });
|
||
});
|
||
req.on('timeout', () => {
|
||
req.destroy();
|
||
resolve({ success: false, error: 'timeout' });
|
||
});
|
||
});
|
||
}
|
||
|
||
async function healthCheck() {
|
||
log.step('健康检查');
|
||
log.subStep('检查后端服务状态...');
|
||
|
||
const maxRetries = 5;
|
||
const retryDelay = 3000;
|
||
|
||
for (let i = 1; i <= maxRetries; i++) {
|
||
const result = await httpHealthCheck('localhost', parseInt(config.backendPort));
|
||
|
||
if (result.success) {
|
||
log.success('后端服务健康检查通过');
|
||
return { success: true };
|
||
}
|
||
|
||
const detail = result.statusCode ? `HTTP ${result.statusCode}` : (result.error || '无响应');
|
||
if (i < maxRetries) {
|
||
log.subStep(`第 ${i} 次检查失败 (${detail})，${retryDelay/1000}秒后重试...`);
|
||
await new Promise(resolve => setTimeout(resolve, retryDelay));
|
||
}
|
||
}
|
||
|
||
log.warning('健康检查未通过，请手动验证服务状态');
|
||
return { success: false };
|
||
}
|
||
|
||
module.exports = {
|
||
httpHealthCheck,
|
||
healthCheck,
|
||
};
|