Compare commits
2 Commits
9e158b88e5
...
80479e8c7d
| Author | SHA1 | Date |
|---|---|---|
cles
|
80479e8c7d | 1 month ago |
|
cles
|
9c8bd39a5b | 1 month ago |
11 changed files with 915 additions and 82 deletions
@ -0,0 +1,221 @@ |
|||
import { EventEmitter } from 'events' |
|||
const log = require('electron-log') |
|||
|
|||
class ReconnectManager extends EventEmitter { |
|||
constructor() { |
|||
super() |
|||
this.reconnectTimers = new Map() // IP -> timer对象
|
|||
this.reconnectAttempts = new Map() // IP -> 当前重连次数
|
|||
this.reconnectConfig = { |
|||
enabled: false, |
|||
interval: 5, // 秒
|
|||
maxAttempts: 10 |
|||
} |
|||
this.isReconnecting = new Map() // IP -> boolean，防止重复重连
|
|||
} |
|||
|
|||
// 更新重连配置
|
|||
updateConfig(config) { |
|||
this.reconnectConfig = { |
|||
...this.reconnectConfig, |
|||
...config, |
|||
interval: Math.max(4, config.interval || this.reconnectConfig.interval) // 最小4秒
|
|||
} |
|||
log.info('Reconnect config updated:', this.reconnectConfig) |
|||
} |
|||
|
|||
// 启动重连
|
|||
startReconnect(ip, reason = 'Connection lost') { |
|||
if (!this.reconnectConfig.enabled) { |
|||
log.info(`Reconnect disabled for ${ip}`) |
|||
return |
|||
} |
|||
|
|||
// 防止重复启动重连
|
|||
if (this.isReconnecting.get(ip)) { |
|||
log.warn(`Reconnect already in progress for ${ip}`) |
|||
return |
|||
} |
|||
|
|||
this.isReconnecting.set(ip, true) |
|||
|
|||
// 如果没有当前重连次数，初始化为0
|
|||
if (!this.reconnectAttempts.has(ip)) { |
|||
this.reconnectAttempts.set(ip, 0) |
|||
} |
|||
|
|||
log.info(`${reason}, starting reconnect for ${ip}...`) |
|||
|
|||
// 发送重连状态更新
|
|||
this.emit('reconnect-status', { |
|||
ip, |
|||
isReconnecting: true |
|||
}) |
|||
|
|||
// 设置重连定时器
|
|||
this.scheduleNextAttempt(ip) |
|||
} |
|||
|
|||
// 安排下次重连尝试
|
|||
scheduleNextAttempt(ip) { |
|||
// 递增重连次数
|
|||
const currentAttempts = this.reconnectAttempts.get(ip) || 0 |
|||
const newAttempts = currentAttempts + 1 |
|||
this.reconnectAttempts.set(ip, newAttempts) |
|||
|
|||
if (newAttempts > this.reconnectConfig.maxAttempts) { |
|||
log.error( |
|||
`Max reconnect attempts reached for ${ip} (${newAttempts}/${this.reconnectConfig.maxAttempts})` |
|||
) |
|||
this.stopReconnect(ip, 'Max attempts reached') |
|||
return |
|||
} |
|||
|
|||
log.info( |
|||
`Scheduling reconnect attempt ${newAttempts}/${this.reconnectConfig.maxAttempts} for ${ip}` |
|||
) |
|||
|
|||
const intervalMs = this.reconnectConfig.interval * 1000 |
|||
log.info(`Reconnect timer set for ${ip} in ${this.reconnectConfig.interval}s (${intervalMs}ms)`) |
|||
|
|||
const timer = setTimeout(() => { |
|||
log.info(`Reconnect timer fired for ${ip}, executing attempt ${newAttempts}`) |
|||
this.executeReconnectAttempt(ip) |
|||
}, intervalMs) |
|||
|
|||
// 清除之前的定时器
|
|||
this.clearTimer(ip) |
|||
this.reconnectTimers.set(ip, timer) |
|||
} |
|||
|
|||
// 执行重连尝试
|
|||
executeReconnectAttempt(ip) { |
|||
const attempts = this.reconnectAttempts.get(ip) || 0 |
|||
log.info( |
|||
`Executing reconnect attempt ${attempts}/${this.reconnectConfig.maxAttempts} for ${ip}` |
|||
) |
|||
|
|||
// 发出重连请求事件
|
|||
this.emit('attempt-reconnect', ip, (success, error) => { |
|||
if (success) { |
|||
log.info(`Reconnect attempt ${attempts} successful for ${ip}`) |
|||
this.onReconnectSuccess(ip) |
|||
} else { |
|||
log.warn(`Reconnect attempt ${attempts} failed for ${ip}: ${error?.message || error}`) |
|||
this.onReconnectFailure(ip, error) |
|||
} |
|||
}) |
|||
} |
|||
|
|||
// 重连成功处理
|
|||
onReconnectSuccess(ip) { |
|||
const attempts = this.reconnectAttempts.get(ip) || 0 |
|||
log.info(`Reconnect attempt ${attempts} successful for ${ip}, clearing reconnect state`) |
|||
|
|||
// 停止重连
|
|||
this.stopReconnect(ip, 'Reconnect successful') |
|||
} |
|||
|
|||
// 重连失败处理
|
|||
onReconnectFailure(ip, error) { |
|||
const attempts = this.reconnectAttempts.get(ip) || 0 |
|||
|
|||
log.warn( |
|||
`Reconnect attempt ${attempts} failed for ${ip}, scheduling next attempt in ${this.reconnectConfig.interval}s`, |
|||
error |
|||
) |
|||
|
|||
// 发送重连状态更新
|
|||
this.emit('reconnect-status', { |
|||
ip, |
|||
isReconnecting: true |
|||
}) |
|||
|
|||
// 安排下次重连
|
|||
this.scheduleNextAttempt(ip) |
|||
} |
|||
|
|||
// 停止重连
|
|||
stopReconnect(ip, reason = 'Manual stop') { |
|||
log.info(`Stopping reconnect for ${ip}: ${reason}`) |
|||
|
|||
this.clearTimer(ip) |
|||
this.reconnectAttempts.delete(ip) |
|||
this.isReconnecting.set(ip, false) |
|||
|
|||
// 发送重连状态更新
|
|||
this.emit('reconnect-status', { |
|||
ip, |
|||
isReconnecting: false |
|||
}) |
|||
} |
|||
|
|||
// 清除重连定时器
|
|||
clearTimer(ip) { |
|||
const timer = this.reconnectTimers.get(ip) |
|||
if (timer) { |
|||
clearTimeout(timer) |
|||
this.reconnectTimers.delete(ip) |
|||
log.debug(`Cleared reconnect timer for ${ip}`) |
|||
} |
|||
} |
|||
|
|||
// 重置重连计数
|
|||
resetAttempts(ip) { |
|||
this.reconnectAttempts.set(ip, 0) |
|||
log.debug(`Reset reconnect attempts for ${ip}`) |
|||
} |
|||
|
|||
// 检查是否正在重连
|
|||
isReconnectInProgress(ip) { |
|||
return this.isReconnecting.get(ip) || false |
|||
} |
|||
|
|||
// 获取重连状态
|
|||
getReconnectStatus(ip) { |
|||
return { |
|||
isReconnecting: this.isReconnecting.get(ip) || false, |
|||
attempts: this.reconnectAttempts.get(ip) || 0, |
|||
maxAttempts: this.reconnectConfig.maxAttempts, |
|||
config: this.reconnectConfig |
|||
} |
|||
} |
|||
|
|||
// 获取所有重连状态
|
|||
getAllReconnectStatus() { |
|||
const status = {} |
|||
for (const [ip] of this.isReconnecting) { |
|||
status[ip] = this.getReconnectStatus(ip) |
|||
} |
|||
return status |
|||
} |
|||
|
|||
// 清理指定IP的所有重连状态
|
|||
cleanup(ip) { |
|||
log.info(`Cleaning up reconnect state for ${ip}`) |
|||
this.clearTimer(ip) |
|||
this.reconnectAttempts.delete(ip) |
|||
this.isReconnecting.delete(ip) |
|||
} |
|||
|
|||
// 清理所有重连状态
|
|||
cleanupAll() { |
|||
log.info('Cleaning up all reconnect states') |
|||
|
|||
// 清除所有定时器
|
|||
for (const [ip] of this.reconnectTimers) { |
|||
this.clearTimer(ip) |
|||
} |
|||
|
|||
// 清除所有状态
|
|||
this.reconnectAttempts.clear() |
|||
this.isReconnecting.clear() |
|||
} |
|||
|
|||
// 获取配置
|
|||
getConfig() { |
|||
return { ...this.reconnectConfig } |
|||
} |
|||
} |
|||
|
|||
export default ReconnectManager |
|||
Loading…
Reference in new issue