mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-23 11:41:53 +08:00
feat: v2.1.0 核心重构与功能增强
## 架构重构
- 全局变量消除,迁移至 Config/State 对象
- SMB 插件融合(smb/smb2/smbghost/smbinfo)
- 服务探测重构,实现 Nmap 风格 fallback 机制
- 输出系统重构,TXT 实时刷盘 + 双写机制
- i18n 框架升级至 go-i18n
## 性能优化
- 正则表达式预编译
- 内存优化 map[string]struct{}
- 并发指纹匹配
- SOCKS5 连接复用
- 滑动窗口调度 + 自适应线程池
## 新功能
- Web 管理界面
- 多格式 POC 适配(xray/afrog)
- 增强指纹库(3139条)
- Favicon hash 指纹识别
- 插件选择性编译(Build Tags)
- fscan-lab 靶场环境
- 默认端口扩展(62→133)
## 构建系统
- 添加 no_local tag 支持排除本地插件
- 多版本构建:fscan/fscan-nolocal/fscan-web
- CI 添加 snapshot 模式支持仅测试构建
## Bug 修复
- 修复 120+ 个问题,包括 RDP panic、批量扫描漏报、
JSON 输出格式、Redis 检测、Context 超时等
## 测试增强
- 单元测试覆盖率 74-100%
- 并发安全测试
- 集成测试(Web/端口/服务/SSH/ICMP)
This commit is contained in:
@@ -0,0 +1,647 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"runtime"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"github.com/shadow1ng/fscan/common/i18n"
|
||||
)
|
||||
|
||||
/*
|
||||
ProgressManager.go - 固定底部进度条管理器
|
||||
|
||||
提供固定在终端底部的进度条显示,与正常输出内容分离。
|
||||
使用终端控制码实现位置固定和内容保护。
|
||||
*/
|
||||
|
||||
// ProgressManager 进度条管理器
|
||||
type ProgressManager struct {
|
||||
mu sync.RWMutex
|
||||
enabled bool
|
||||
total int64
|
||||
current int64
|
||||
description string
|
||||
startTime time.Time
|
||||
isActive bool
|
||||
terminalHeight int
|
||||
reservedLines int // 为进度条保留的行数
|
||||
lastContentLine int // 最后一行内容的位置
|
||||
|
||||
// 输出缓冲相关
|
||||
outputMutex sync.Mutex
|
||||
|
||||
// 活跃指示器相关
|
||||
spinnerIndex int
|
||||
lastActivity time.Time
|
||||
activityTicker *time.Ticker
|
||||
stopActivityChan chan struct{}
|
||||
|
||||
// 内存监控相关
|
||||
lastMemUpdate time.Time
|
||||
memStats runtime.MemStats
|
||||
|
||||
// 进度条更新控制(减少 Windows 终端的重复输出)
|
||||
lastRenderedPercent int
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// ANSI终端控制码常量
|
||||
// =============================================================================
|
||||
|
||||
const (
|
||||
// AnsiClearLine 光标和行控制 - 清除当前行并回到行首
|
||||
AnsiClearLine = "\033[2K\r"
|
||||
// AnsiMoveCursor 向上移动N行(格式化字符串)
|
||||
AnsiMoveCursor = "\033[%dA"
|
||||
|
||||
// AnsiRed 颜色代码 - 红色文本
|
||||
AnsiRed = "\033[31m"
|
||||
// AnsiGreen 绿色文本
|
||||
AnsiGreen = "\033[32m"
|
||||
// AnsiYellow 黄色文本
|
||||
AnsiYellow = "\033[33m"
|
||||
// AnsiCyan 青色文本
|
||||
AnsiCyan = "\033[36m"
|
||||
// AnsiGray 灰色文本
|
||||
AnsiGray = "\033[90m"
|
||||
// AnsiReset 重置所有属性
|
||||
AnsiReset = "\033[0m"
|
||||
)
|
||||
|
||||
var (
|
||||
globalProgressManager *ProgressManager
|
||||
progressMutex sync.Mutex
|
||||
|
||||
// 活跃指示器字符序列(旋转动画)
|
||||
spinnerChars = []string{"|", "/", "-", "\\"}
|
||||
|
||||
// 活跃指示器更新间隔
|
||||
activityUpdateInterval = 500 * time.Millisecond
|
||||
)
|
||||
|
||||
// GetProgressManager 获取全局进度条管理器
|
||||
func GetProgressManager() *ProgressManager {
|
||||
progressMutex.Lock()
|
||||
defer progressMutex.Unlock()
|
||||
|
||||
if globalProgressManager == nil {
|
||||
globalProgressManager = &ProgressManager{
|
||||
enabled: true,
|
||||
reservedLines: 2, // 保留2行:进度条 + 空行
|
||||
terminalHeight: getTerminalHeight(),
|
||||
}
|
||||
}
|
||||
return globalProgressManager
|
||||
}
|
||||
|
||||
// InitProgress 初始化进度条
|
||||
func (pm *ProgressManager) InitProgress(total int64, description string) {
|
||||
fv := GetFlagVars()
|
||||
if fv.DisableProgress || fv.Silent {
|
||||
pm.enabled = false
|
||||
return
|
||||
}
|
||||
|
||||
pm.mu.Lock()
|
||||
defer pm.mu.Unlock()
|
||||
|
||||
pm.total = total
|
||||
pm.current = 0
|
||||
pm.description = description
|
||||
pm.startTime = time.Now()
|
||||
pm.isActive = true
|
||||
pm.enabled = true
|
||||
pm.lastActivity = time.Now()
|
||||
pm.spinnerIndex = 0
|
||||
pm.lastMemUpdate = time.Now().Add(-2 * time.Second) // 强制首次更新内存
|
||||
pm.lastRenderedPercent = -1 // 强制首次渲染
|
||||
|
||||
// 为进度条保留空间
|
||||
pm.setupProgressSpace()
|
||||
|
||||
// 启动活跃指示器
|
||||
pm.startActivityIndicator()
|
||||
|
||||
// 初始显示进度条
|
||||
pm.renderProgress()
|
||||
}
|
||||
|
||||
// UpdateProgress 更新进度
|
||||
func (pm *ProgressManager) UpdateProgress(increment int64) {
|
||||
if !pm.enabled || !pm.isActive {
|
||||
return
|
||||
}
|
||||
|
||||
pm.mu.Lock()
|
||||
defer pm.mu.Unlock()
|
||||
|
||||
pm.current += increment
|
||||
if pm.current > pm.total {
|
||||
pm.current = pm.total
|
||||
}
|
||||
|
||||
// 更新活跃时间
|
||||
pm.lastActivity = time.Now()
|
||||
|
||||
pm.renderProgress()
|
||||
}
|
||||
|
||||
// =============================================================================================
|
||||
// 已删除的死代码(未使用):SetProgress 设置当前进度
|
||||
// =============================================================================================
|
||||
|
||||
// FinishProgress 完成进度条
|
||||
func (pm *ProgressManager) FinishProgress() {
|
||||
if !pm.enabled || !pm.isActive {
|
||||
return
|
||||
}
|
||||
|
||||
pm.mu.Lock()
|
||||
defer pm.mu.Unlock()
|
||||
|
||||
pm.current = pm.total
|
||||
pm.renderProgress()
|
||||
|
||||
// 停止活跃指示器
|
||||
pm.stopActivityIndicator()
|
||||
|
||||
// 显示完成信息
|
||||
pm.showCompletionInfo()
|
||||
|
||||
// 清理进度条区域,恢复正常输出
|
||||
pm.clearProgressArea()
|
||||
pm.isActive = false
|
||||
}
|
||||
|
||||
// setupProgressSpace 设置进度条空间
|
||||
func (pm *ProgressManager) setupProgressSpace() {
|
||||
// 简化设计:进度条在原地更新,不需要预留额外空间
|
||||
// 只是标记进度条开始的位置
|
||||
pm.lastContentLine = 0
|
||||
}
|
||||
|
||||
// =============================================================================================
|
||||
// 已删除的死代码(未使用):moveToContentArea 和 moveToProgressLine 方法
|
||||
// =============================================================================================
|
||||
|
||||
// renderProgress 渲染进度条(使用锁避免输出冲突)
|
||||
func (pm *ProgressManager) renderProgress() {
|
||||
pm.outputMutex.Lock()
|
||||
defer pm.outputMutex.Unlock()
|
||||
|
||||
pm.renderProgressUnsafe()
|
||||
}
|
||||
|
||||
// generateProgressBar 生成进度条字符串
|
||||
func (pm *ProgressManager) generateProgressBar() string {
|
||||
if pm.total == 0 {
|
||||
spinner := pm.getActivityIndicator()
|
||||
memInfo := pm.getMemoryInfo()
|
||||
|
||||
// 获取TCP包统计(包含原HTTP请求)
|
||||
packetCount := GetGlobalState().GetPacketCount()
|
||||
tcpSuccess := GetGlobalState().GetTCPSuccessPacketCount()
|
||||
tcpFailed := GetGlobalState().GetTCPFailedPacketCount()
|
||||
udpCount := GetGlobalState().GetUDPPacketCount()
|
||||
|
||||
packetInfo := ""
|
||||
if packetCount > 0 {
|
||||
// 构建简化的包统计信息:只显示TCP和UDP
|
||||
details := make([]string, 0, 2)
|
||||
if tcpSuccess > 0 || tcpFailed > 0 {
|
||||
details = append(details, fmt.Sprintf("TCP:%d✓%d✗", tcpSuccess, tcpFailed))
|
||||
}
|
||||
if udpCount > 0 {
|
||||
details = append(details, fmt.Sprintf("UDP:%d", udpCount))
|
||||
}
|
||||
|
||||
if len(details) > 0 {
|
||||
packetInfo = fmt.Sprintf(" 发包:%d[%s]", packetCount, strings.Join(details, ","))
|
||||
} else {
|
||||
packetInfo = fmt.Sprintf(" 发包:%d", packetCount)
|
||||
}
|
||||
}
|
||||
|
||||
return fmt.Sprintf("%s %s 等待中...%s %s", pm.description, spinner, packetInfo, memInfo)
|
||||
}
|
||||
|
||||
percentage := float64(pm.current) / float64(pm.total) * 100
|
||||
elapsed := time.Since(pm.startTime)
|
||||
|
||||
// 获取并发状态
|
||||
concurrencyStatus := GetConcurrencyMonitor().GetConcurrencyStatus()
|
||||
|
||||
// 计算预估剩余时间
|
||||
var eta string
|
||||
if pm.current > 0 {
|
||||
totalTime := elapsed * time.Duration(pm.total) / time.Duration(pm.current)
|
||||
remaining := totalTime - elapsed
|
||||
if remaining > 0 {
|
||||
eta = fmt.Sprintf(" ETA:%s", formatDuration(remaining))
|
||||
}
|
||||
}
|
||||
|
||||
// 计算速度
|
||||
speed := float64(pm.current) / elapsed.Seconds()
|
||||
speedStr := ""
|
||||
if speed > 0 {
|
||||
speedStr = fmt.Sprintf(" (%.1f/s)", speed)
|
||||
}
|
||||
|
||||
// 生成进度条
|
||||
barWidth := 30
|
||||
filled := int(percentage * float64(barWidth) / 100)
|
||||
bar := ""
|
||||
|
||||
if GetFlagVars().NoColor {
|
||||
// 无颜色版本
|
||||
bar = "[" +
|
||||
fmt.Sprintf("%s%s",
|
||||
string(make([]rune, filled)),
|
||||
string(make([]rune, barWidth-filled))) +
|
||||
"]"
|
||||
for i := 0; i < filled; i++ {
|
||||
bar = bar[:i+1] + "=" + bar[i+2:]
|
||||
}
|
||||
for i := filled; i < barWidth; i++ {
|
||||
bar = bar[:i+1] + "-" + bar[i+2:]
|
||||
}
|
||||
} else {
|
||||
// 彩色版本
|
||||
bar = "|"
|
||||
for i := 0; i < barWidth; i++ {
|
||||
if i < filled {
|
||||
bar += "#"
|
||||
} else {
|
||||
bar += "."
|
||||
}
|
||||
}
|
||||
bar += "|"
|
||||
}
|
||||
|
||||
// 生成活跃指示器
|
||||
spinner := pm.getActivityIndicator()
|
||||
|
||||
// 获取TCP包统计(包含原HTTP请求)
|
||||
packetCount := GetGlobalState().GetPacketCount()
|
||||
tcpSuccess := GetGlobalState().GetTCPSuccessPacketCount()
|
||||
tcpFailed := GetGlobalState().GetTCPFailedPacketCount()
|
||||
udpCount := GetGlobalState().GetUDPPacketCount()
|
||||
|
||||
packetInfo := ""
|
||||
if packetCount > 0 {
|
||||
// 构建简化的包统计信息:只显示TCP和UDP
|
||||
details := make([]string, 0, 2)
|
||||
if tcpSuccess > 0 || tcpFailed > 0 {
|
||||
details = append(details, fmt.Sprintf("TCP:%d✓%d✗", tcpSuccess, tcpFailed))
|
||||
}
|
||||
if udpCount > 0 {
|
||||
details = append(details, fmt.Sprintf("UDP:%d", udpCount))
|
||||
}
|
||||
|
||||
if len(details) > 0 {
|
||||
packetInfo = fmt.Sprintf(" 发包:%d[%s]", packetCount, strings.Join(details, ","))
|
||||
} else {
|
||||
packetInfo = fmt.Sprintf(" 发包:%d", packetCount)
|
||||
}
|
||||
}
|
||||
|
||||
// 构建基础进度条
|
||||
baseProgress := fmt.Sprintf("%s %s %6.1f%% %s (%d/%d)%s%s%s",
|
||||
pm.description, spinner, percentage, bar, pm.current, pm.total, speedStr, eta, packetInfo)
|
||||
|
||||
// 添加内存信息
|
||||
memInfo := pm.getMemoryInfo()
|
||||
|
||||
// 添加并发状态
|
||||
if concurrencyStatus != "" {
|
||||
return fmt.Sprintf("%s [%s] %s", baseProgress, concurrencyStatus, memInfo)
|
||||
}
|
||||
|
||||
return fmt.Sprintf("%s %s", baseProgress, memInfo)
|
||||
}
|
||||
|
||||
// showCompletionInfo 显示完成信息
|
||||
func (pm *ProgressManager) showCompletionInfo() {
|
||||
elapsed := time.Since(pm.startTime)
|
||||
|
||||
// 换行并显示完成信息
|
||||
fmt.Print("\n")
|
||||
|
||||
completionMsg := i18n.GetText("progress_scan_completed")
|
||||
if GetFlagVars().NoColor {
|
||||
fmt.Printf("[完成] %s %d/%d (耗时: %s)\n",
|
||||
completionMsg, pm.total, pm.total, formatDuration(elapsed))
|
||||
} else {
|
||||
fmt.Printf("%s[完成] %s %d/%d%s %s(耗时: %s)%s\n",
|
||||
AnsiGreen, completionMsg, pm.total, pm.total, AnsiReset,
|
||||
AnsiGray, formatDuration(elapsed), AnsiReset)
|
||||
}
|
||||
}
|
||||
|
||||
// clearProgressArea 清理进度条区域
|
||||
func (pm *ProgressManager) clearProgressArea() {
|
||||
// 简单清除当前行
|
||||
fmt.Print(AnsiClearLine)
|
||||
}
|
||||
|
||||
// IsActive 检查进度条是否活跃
|
||||
func (pm *ProgressManager) IsActive() bool {
|
||||
pm.mu.RLock()
|
||||
defer pm.mu.RUnlock()
|
||||
return pm.isActive && pm.enabled
|
||||
}
|
||||
|
||||
// getTerminalHeight 获取终端高度
|
||||
func getTerminalHeight() int {
|
||||
// 对于固定底部进度条,我们暂时禁用终端高度检测
|
||||
// 因为在不同终端环境中可能会有问题
|
||||
// 改为使用相对定位方式
|
||||
return 0 // 返回0表示使用简化模式
|
||||
}
|
||||
|
||||
// formatDuration 格式化时间间隔
|
||||
func formatDuration(d time.Duration) string {
|
||||
if d < time.Minute {
|
||||
return fmt.Sprintf("%.1fs", d.Seconds())
|
||||
}
|
||||
if d < time.Hour {
|
||||
return fmt.Sprintf("%.1fm", d.Minutes())
|
||||
}
|
||||
return fmt.Sprintf("%.1fh", d.Hours())
|
||||
}
|
||||
|
||||
// InitProgressBar 初始化进度条(全局函数,方便其他模块调用)
|
||||
func InitProgressBar(total int64, description string) {
|
||||
GetProgressManager().InitProgress(total, description)
|
||||
}
|
||||
|
||||
// UpdateProgressBar 更新进度条
|
||||
func UpdateProgressBar(increment int64) {
|
||||
GetProgressManager().UpdateProgress(increment)
|
||||
}
|
||||
|
||||
// =============================================================================================
|
||||
// 已删除的死代码(未使用):SetProgressBar 全局函数
|
||||
// =============================================================================================
|
||||
|
||||
// FinishProgressBar 完成进度条
|
||||
func FinishProgressBar() {
|
||||
GetProgressManager().FinishProgress()
|
||||
}
|
||||
|
||||
// IsProgressActive 检查进度条是否活跃
|
||||
func IsProgressActive() bool {
|
||||
return GetProgressManager().IsActive()
|
||||
}
|
||||
|
||||
// GetProgressPercent 获取当前进度百分比 (0-100)
|
||||
func GetProgressPercent() float64 {
|
||||
return GetProgressManager().GetPercent()
|
||||
}
|
||||
|
||||
// GetPercent 获取当前进度百分比
|
||||
func (pm *ProgressManager) GetPercent() float64 {
|
||||
pm.mu.RLock()
|
||||
defer pm.mu.RUnlock()
|
||||
|
||||
if !pm.isActive || pm.total == 0 {
|
||||
return 0
|
||||
}
|
||||
return float64(pm.current) / float64(pm.total) * 100
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// 日志输出协调功能
|
||||
// =============================================================================
|
||||
|
||||
// LogWithProgress 在进度条活跃时协调日志输出
|
||||
func LogWithProgress(message string) {
|
||||
pm := GetProgressManager()
|
||||
if !pm.IsActive() {
|
||||
// 如果进度条不活跃,直接输出
|
||||
fmt.Println(message)
|
||||
return
|
||||
}
|
||||
|
||||
pm.outputMutex.Lock()
|
||||
defer pm.outputMutex.Unlock()
|
||||
|
||||
// 清除当前行(清除进度条)
|
||||
// Windows 通过 progress_manager_win.go 已启用 ANSI 支持
|
||||
fmt.Print(AnsiClearLine)
|
||||
|
||||
// 输出日志消息
|
||||
fmt.Println(message)
|
||||
|
||||
// 不重绘进度条,等待下次 UpdateProgress 自动绘制
|
||||
}
|
||||
|
||||
// renderProgressUnsafe 不加锁的进度条渲染(内部使用)
|
||||
func (pm *ProgressManager) renderProgressUnsafe() {
|
||||
if !pm.enabled || !pm.isActive {
|
||||
return
|
||||
}
|
||||
|
||||
// 计算当前百分比(避免除零)
|
||||
currentPercent := 0
|
||||
if pm.total > 0 {
|
||||
currentPercent = int((pm.current * 100) / pm.total)
|
||||
}
|
||||
|
||||
// 只在百分比变化时更新,减少不必要的渲染
|
||||
if currentPercent == pm.lastRenderedPercent && currentPercent < 100 {
|
||||
return
|
||||
}
|
||||
pm.lastRenderedPercent = currentPercent
|
||||
|
||||
// 生成进度条内容
|
||||
progressBar := pm.generateProgressBar()
|
||||
|
||||
// 移动到行首(Windows 已通过 progress_manager_win.go 启用 ANSI 支持)
|
||||
fmt.Print("\r")
|
||||
|
||||
// 输出进度条(带颜色,如果启用)
|
||||
if GetFlagVars().NoColor {
|
||||
fmt.Print(progressBar)
|
||||
} else {
|
||||
fmt.Printf("%s%s%s", AnsiCyan, progressBar, AnsiReset)
|
||||
}
|
||||
|
||||
// 刷新输出
|
||||
_ = os.Stdout.Sync()
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// 活跃指示器相关方法
|
||||
// =============================================================================
|
||||
|
||||
// startActivityIndicator 启动活跃指示器
|
||||
func (pm *ProgressManager) startActivityIndicator() {
|
||||
// 防止重复启动
|
||||
if pm.activityTicker != nil {
|
||||
return
|
||||
}
|
||||
|
||||
pm.activityTicker = time.NewTicker(activityUpdateInterval)
|
||||
pm.stopActivityChan = make(chan struct{})
|
||||
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case <-pm.activityTicker.C:
|
||||
// 只有在活跃状态下才更新指示器
|
||||
if pm.isActive && pm.enabled {
|
||||
pm.mu.Lock()
|
||||
pm.spinnerIndex = (pm.spinnerIndex + 1) % len(spinnerChars)
|
||||
pm.mu.Unlock()
|
||||
|
||||
// 只有在长时间没有进度更新时才重新渲染
|
||||
// 这样可以避免频繁更新时的性能问题
|
||||
if time.Since(pm.lastActivity) > 2*time.Second {
|
||||
pm.renderProgress()
|
||||
}
|
||||
}
|
||||
case <-pm.stopActivityChan:
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
// stopActivityIndicator 停止活跃指示器
|
||||
func (pm *ProgressManager) stopActivityIndicator() {
|
||||
if pm.activityTicker != nil {
|
||||
pm.activityTicker.Stop()
|
||||
pm.activityTicker = nil
|
||||
}
|
||||
|
||||
if pm.stopActivityChan != nil {
|
||||
close(pm.stopActivityChan)
|
||||
pm.stopActivityChan = nil
|
||||
}
|
||||
}
|
||||
|
||||
// getActivityIndicator 获取当前活跃指示器字符
|
||||
func (pm *ProgressManager) getActivityIndicator() string {
|
||||
// 如果最近有活动(2秒内),显示静态指示器
|
||||
if time.Since(pm.lastActivity) <= 2*time.Second {
|
||||
return "●" // 实心圆表示活跃
|
||||
}
|
||||
|
||||
// 如果长时间没有活动,显示旋转指示器表明程序仍在运行
|
||||
return spinnerChars[pm.spinnerIndex]
|
||||
}
|
||||
|
||||
// getMemoryInfo 获取内存使用信息
|
||||
func (pm *ProgressManager) getMemoryInfo() string {
|
||||
// 限制内存统计更新频率以提高性能(每秒最多一次)
|
||||
now := time.Now()
|
||||
if now.Sub(pm.lastMemUpdate) >= time.Second {
|
||||
runtime.ReadMemStats(&pm.memStats)
|
||||
pm.lastMemUpdate = now
|
||||
}
|
||||
|
||||
// 获取当前使用的内存(以MB为单位)
|
||||
memUsedMB := float64(pm.memStats.Alloc) / 1024 / 1024
|
||||
|
||||
// 根据内存使用量选择颜色
|
||||
var colorCode string
|
||||
if GetFlagVars().NoColor {
|
||||
return fmt.Sprintf("内存:%.1fMB", memUsedMB)
|
||||
}
|
||||
|
||||
// 根据内存使用量设置颜色
|
||||
if memUsedMB < 50 {
|
||||
colorCode = AnsiGreen // 绿色 - 内存使用较低
|
||||
} else if memUsedMB < 100 {
|
||||
colorCode = AnsiYellow // 黄色 - 内存使用中等
|
||||
} else {
|
||||
colorCode = AnsiRed // 红色 - 内存使用较高
|
||||
}
|
||||
|
||||
return fmt.Sprintf("%s内存:%.1fMB%s", colorCode, memUsedMB, AnsiReset)
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// 并发监控器 (从 concurrency_monitor.go 合并)
|
||||
// =============================================================================
|
||||
|
||||
/*
|
||||
ConcurrencyMonitor - 并发监控器
|
||||
|
||||
监控两个层级的并发:
|
||||
1. 主扫描器线程数 (-t 参数控制)
|
||||
2. 插件内连接线程数 (-mt 参数控制)
|
||||
*/
|
||||
|
||||
// ConcurrencyMonitor 并发监控器
|
||||
type ConcurrencyMonitor struct {
|
||||
// 主扫描器层级
|
||||
activePluginTasks int64 // 当前活跃的插件任务数
|
||||
totalPluginTasks int64 // 总插件任务数
|
||||
|
||||
// 插件内连接层级已移除 - 原代码为死代码,无任何调用者
|
||||
}
|
||||
|
||||
// 已移除 PluginConnectionInfo 结构体 - 原为死代码,无任何使用
|
||||
|
||||
var (
|
||||
globalConcurrencyMonitor *ConcurrencyMonitor
|
||||
concurrencyMutex sync.Once
|
||||
)
|
||||
|
||||
// GetConcurrencyMonitor 获取全局并发监控器
|
||||
func GetConcurrencyMonitor() *ConcurrencyMonitor {
|
||||
concurrencyMutex.Do(func() {
|
||||
globalConcurrencyMonitor = &ConcurrencyMonitor{
|
||||
activePluginTasks: 0,
|
||||
totalPluginTasks: 0,
|
||||
}
|
||||
})
|
||||
return globalConcurrencyMonitor
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// 主扫描器层级监控
|
||||
// =============================================================================
|
||||
|
||||
// StartPluginTask 开始插件任务
|
||||
func (m *ConcurrencyMonitor) StartPluginTask() {
|
||||
atomic.AddInt64(&m.activePluginTasks, 1)
|
||||
atomic.AddInt64(&m.totalPluginTasks, 1)
|
||||
}
|
||||
|
||||
// FinishPluginTask 完成插件任务
|
||||
func (m *ConcurrencyMonitor) FinishPluginTask() {
|
||||
atomic.AddInt64(&m.activePluginTasks, -1)
|
||||
}
|
||||
|
||||
// GetPluginTaskStats 获取插件任务统计
|
||||
func (m *ConcurrencyMonitor) GetPluginTaskStats() (active int64, total int64) {
|
||||
return atomic.LoadInt64(&m.activePluginTasks), atomic.LoadInt64(&m.totalPluginTasks)
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// 已移除插件内连接层级监控 - 原为死代码,无任何调用者
|
||||
// =============================================================================
|
||||
|
||||
// 已移除未使用的 Reset 方法
|
||||
|
||||
// GetConcurrencyStatus 获取并发状态字符串
|
||||
func (m *ConcurrencyMonitor) GetConcurrencyStatus() string {
|
||||
activePlugins, _ := m.GetPluginTaskStats()
|
||||
|
||||
if activePlugins == 0 {
|
||||
return ""
|
||||
}
|
||||
|
||||
return fmt.Sprintf("%s:%d", i18n.GetText("concurrency_plugin"), activePlugins)
|
||||
}
|
||||
|
||||
// 已移除未使用的 GetDetailedStatus 方法
|
||||
Reference in New Issue
Block a user