mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-25 20:51:52 +08:00
feat: 分离结果输出和日志
This commit is contained in:
ZacharyZcR
+78
-198
@@ -1,12 +1,9 @@
|
||||
package Common
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
@@ -16,118 +13,123 @@ import (
|
||||
"github.com/fatih/color"
|
||||
)
|
||||
|
||||
// 全局变量定义
|
||||
var (
|
||||
// 全局变量
|
||||
// 扫描状态管理器，记录最近一次成功和错误的时间
|
||||
status = &ScanStatus{lastSuccess: time.Now(), lastError: time.Now()}
|
||||
|
||||
// 扫描计数
|
||||
Num int64 // 总任务数
|
||||
End int64 // 已完成任务数
|
||||
|
||||
// 文件写入器
|
||||
fileWriter *bufferedFileWriter
|
||||
// Num 表示待处理的总任务数量
|
||||
Num int64
|
||||
// End 表示已经完成的任务数量
|
||||
End int64
|
||||
)
|
||||
|
||||
// ScanStatus 记录扫描状态
|
||||
// ScanStatus 用于记录和管理扫描状态的结构体
|
||||
type ScanStatus struct {
|
||||
mu sync.RWMutex
|
||||
total int64
|
||||
completed int64
|
||||
lastSuccess time.Time
|
||||
lastError time.Time
|
||||
mu sync.RWMutex // 读写互斥锁，用于保护并发访问
|
||||
total int64 // 总任务数
|
||||
completed int64 // 已完成任务数
|
||||
lastSuccess time.Time // 最近一次成功的时间
|
||||
lastError time.Time // 最近一次错误的时间
|
||||
}
|
||||
|
||||
// LogEntry 日志条目
|
||||
// LogEntry 定义单条日志的结构
|
||||
type LogEntry struct {
|
||||
Level string // "ERROR", "INFO", "SUCCESS", "DEBUG"
|
||||
Time time.Time
|
||||
Content string
|
||||
Level string // 日志级别: ERROR/INFO/SUCCESS/DEBUG
|
||||
Time time.Time // 日志时间
|
||||
Content string // 日志内容
|
||||
}
|
||||
|
||||
// LogLevel 定义日志等级常量
|
||||
// 定义系统支持的日志级别常量
|
||||
const (
|
||||
LogLevelAll = "ALL" // 输出所有日志
|
||||
LogLevelError = "ERROR" // 错误日志
|
||||
LogLevelInfo = "INFO" // 信息日志
|
||||
LogLevelSuccess = "SUCCESS" // 成功日志
|
||||
LogLevelDebug = "DEBUG" // 调试日志
|
||||
LogLevelAll = "ALL" // 显示所有级别日志
|
||||
LogLevelError = "ERROR" // 仅显示错误日志
|
||||
LogLevelInfo = "INFO" // 仅显示信息日志
|
||||
LogLevelSuccess = "SUCCESS" // 仅显示成功日志
|
||||
LogLevelDebug = "DEBUG" // 仅显示调试日志
|
||||
)
|
||||
|
||||
// 定义日志颜色映射
|
||||
// 日志级别对应的显示颜色映射
|
||||
var logColors = map[string]color.Attribute{
|
||||
LogLevelError: color.FgRed,
|
||||
LogLevelInfo: color.FgYellow,
|
||||
LogLevelSuccess: color.FgGreen,
|
||||
LogLevelDebug: color.FgBlue,
|
||||
}
|
||||
|
||||
// JsonOutput JSON输出的结构体
|
||||
type JsonOutput struct {
|
||||
Level string `json:"level"`
|
||||
Timestamp time.Time `json:"timestamp"`
|
||||
Message string `json:"message"`
|
||||
LogLevelError: color.FgRed, // 错误日志显示红色
|
||||
LogLevelInfo: color.FgYellow, // 信息日志显示黄色
|
||||
LogLevelSuccess: color.FgGreen, // 成功日志显示绿色
|
||||
LogLevelDebug: color.FgBlue, // 调试日志显示蓝色
|
||||
}
|
||||
|
||||
// InitLogger 初始化日志系统
|
||||
func InitLogger() {
|
||||
// 禁用标准日志输出
|
||||
log.SetOutput(io.Discard)
|
||||
if !DisableSave {
|
||||
fileWriter = newBufferedFileWriter()
|
||||
}
|
||||
}
|
||||
|
||||
// formatLogMessage 格式化日志消息
|
||||
// formatLogMessage 格式化日志消息为标准格式
|
||||
// 返回格式：[时间] [级别] 内容
|
||||
func formatLogMessage(entry *LogEntry) string {
|
||||
timeStr := entry.Time.Format("2006-01-02 15:04:05")
|
||||
return fmt.Sprintf("[%s] [%s] %s", timeStr, entry.Level, entry.Content)
|
||||
}
|
||||
|
||||
// 修改 printLog 函数
|
||||
// printLog 根据日志级别打印日志
|
||||
func printLog(entry *LogEntry) {
|
||||
// 默认情况(LogLevelInfo)下打印 INFO、SUCCESS、ERROR
|
||||
if LogLevel == LogLevelInfo {
|
||||
// 根据当前设置的日志级别过滤日志
|
||||
switch LogLevel {
|
||||
case LogLevelInfo:
|
||||
// INFO模式下只打印 INFO、SUCCESS、ERROR 级别的日志
|
||||
if entry.Level != LogLevelInfo &&
|
||||
entry.Level != LogLevelSuccess &&
|
||||
entry.Level != LogLevelError {
|
||||
return
|
||||
}
|
||||
} else if LogLevel == LogLevelDebug || LogLevel == LogLevelAll {
|
||||
// Debug或ALL模式打印所有日志
|
||||
} else if entry.Level != LogLevel {
|
||||
// 其他情况只打印指定等级的日志
|
||||
return
|
||||
case LogLevelDebug, LogLevelAll:
|
||||
// Debug或ALL模式下打印所有日志
|
||||
default:
|
||||
// 其他模式下只打印指定级别的日志
|
||||
if entry.Level != LogLevel {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
OutputMutex.Lock()
|
||||
defer OutputMutex.Unlock()
|
||||
|
||||
// 确保清除当前进度条
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.Clear()
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
}
|
||||
// 处理进度条
|
||||
clearAndWaitProgress()
|
||||
|
||||
// 打印日志
|
||||
// 打印日志消息
|
||||
logMsg := formatLogMessage(entry)
|
||||
if !NoColor {
|
||||
// 使用彩色输出
|
||||
if colorAttr, ok := logColors[entry.Level]; ok {
|
||||
color.New(colorAttr).Println(logMsg)
|
||||
} else {
|
||||
fmt.Println(logMsg)
|
||||
}
|
||||
} else {
|
||||
// 普通输出
|
||||
fmt.Println(logMsg)
|
||||
}
|
||||
|
||||
// 确保日志完全输出
|
||||
// 等待日志输出完成
|
||||
time.Sleep(50 * time.Millisecond)
|
||||
|
||||
// 重新渲染进度条
|
||||
// 重新显示进度条
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.RenderBlank()
|
||||
}
|
||||
}
|
||||
|
||||
// clearAndWaitProgress 清除进度条并等待
|
||||
func clearAndWaitProgress() {
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.Clear()
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
}
|
||||
}
|
||||
|
||||
// LogError 记录错误日志，自动包含文件名和行号信息
|
||||
func LogError(errMsg string) {
|
||||
// 获取调用者的文件名和行号
|
||||
_, file, line, ok := runtime.Caller(1)
|
||||
if !ok {
|
||||
file = "unknown"
|
||||
@@ -137,182 +139,60 @@ func LogError(errMsg string) {
|
||||
|
||||
errorMsg := fmt.Sprintf("%s:%d - %s", file, line, errMsg)
|
||||
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.Clear()
|
||||
}
|
||||
|
||||
entry := &LogEntry{
|
||||
Level: LogLevelError,
|
||||
Time: time.Now(),
|
||||
Content: errorMsg,
|
||||
}
|
||||
|
||||
printLog(entry)
|
||||
if fileWriter != nil {
|
||||
fileWriter.write(entry)
|
||||
handleLog(entry)
|
||||
}
|
||||
|
||||
// handleLog 统一处理日志的输出
|
||||
func handleLog(entry *LogEntry) {
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.Clear()
|
||||
}
|
||||
|
||||
printLog(entry)
|
||||
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.RenderBlank()
|
||||
}
|
||||
}
|
||||
|
||||
// LogInfo 记录信息日志
|
||||
func LogInfo(msg string) {
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.Clear()
|
||||
}
|
||||
|
||||
entry := &LogEntry{
|
||||
handleLog(&LogEntry{
|
||||
Level: LogLevelInfo,
|
||||
Time: time.Now(),
|
||||
Content: msg,
|
||||
}
|
||||
|
||||
printLog(entry)
|
||||
if fileWriter != nil {
|
||||
fileWriter.write(entry)
|
||||
}
|
||||
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.RenderBlank()
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// LogSuccess 记录成功日志，并更新最后成功时间
|
||||
func LogSuccess(result string) {
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.Clear()
|
||||
}
|
||||
|
||||
entry := &LogEntry{
|
||||
Level: LogLevelSuccess,
|
||||
Time: time.Now(),
|
||||
Content: result,
|
||||
}
|
||||
|
||||
printLog(entry)
|
||||
if fileWriter != nil {
|
||||
fileWriter.write(entry)
|
||||
}
|
||||
handleLog(entry)
|
||||
|
||||
// 更新最后成功时间
|
||||
status.mu.Lock()
|
||||
status.lastSuccess = time.Now()
|
||||
status.mu.Unlock()
|
||||
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.RenderBlank()
|
||||
}
|
||||
}
|
||||
|
||||
// LogDebug 记录调试日志
|
||||
func LogDebug(msg string) {
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.Clear()
|
||||
}
|
||||
|
||||
entry := &LogEntry{
|
||||
handleLog(&LogEntry{
|
||||
Level: LogLevelDebug,
|
||||
Time: time.Now(),
|
||||
Content: msg,
|
||||
}
|
||||
|
||||
printLog(entry)
|
||||
if fileWriter != nil {
|
||||
fileWriter.write(entry)
|
||||
}
|
||||
|
||||
if ProgressBar != nil {
|
||||
ProgressBar.RenderBlank()
|
||||
}
|
||||
}
|
||||
|
||||
func newBufferedFileWriter() *bufferedFileWriter {
|
||||
file, err := os.OpenFile(Outputfile, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
|
||||
if err != nil {
|
||||
fmt.Printf("[ERROR] 打开输出文件失败 %s: %v\n", Outputfile, err)
|
||||
return nil
|
||||
}
|
||||
|
||||
writer := bufio.NewWriter(file)
|
||||
return &bufferedFileWriter{
|
||||
file: file,
|
||||
writer: writer,
|
||||
jsonEnc: json.NewEncoder(writer),
|
||||
}
|
||||
}
|
||||
|
||||
type bufferedFileWriter struct {
|
||||
file *os.File
|
||||
writer *bufio.Writer
|
||||
jsonEnc *json.Encoder
|
||||
mu sync.Mutex // 添加互斥锁保护写入
|
||||
}
|
||||
|
||||
func (w *bufferedFileWriter) write(entry *LogEntry) {
|
||||
if w == nil {
|
||||
return
|
||||
}
|
||||
|
||||
w.mu.Lock()
|
||||
defer w.mu.Unlock()
|
||||
|
||||
var err error
|
||||
if JsonFormat {
|
||||
output := JsonOutput{
|
||||
Level: entry.Level,
|
||||
Timestamp: entry.Time,
|
||||
Message: entry.Content,
|
||||
}
|
||||
err = w.jsonEnc.Encode(output)
|
||||
} else {
|
||||
logMsg := formatLogMessage(entry) + "\n"
|
||||
_, err = w.writer.WriteString(logMsg)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("[ERROR] 写入日志失败: %v\n", err)
|
||||
// 尝试重新打开文件
|
||||
if err := w.reopen(); err != nil {
|
||||
fmt.Printf("[ERROR] 重新打开文件失败: %v\n", err)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// 每隔一定数量的写入才进行一次Flush
|
||||
if err := w.writer.Flush(); err != nil {
|
||||
fmt.Printf("[ERROR] 刷新缓冲区失败: %v\n", err)
|
||||
if err := w.reopen(); err != nil {
|
||||
fmt.Printf("[ERROR] 重新打开文件失败: %v\n", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (w *bufferedFileWriter) reopen() error {
|
||||
if w.file != nil {
|
||||
w.file.Close()
|
||||
}
|
||||
|
||||
file, err := os.OpenFile(Outputfile, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
w.file = file
|
||||
w.writer = bufio.NewWriter(file)
|
||||
w.jsonEnc = json.NewEncoder(w.writer)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (w *bufferedFileWriter) close() {
|
||||
if w != nil {
|
||||
w.writer.Flush()
|
||||
w.file.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func CloseLogger() {
|
||||
if fileWriter != nil {
|
||||
fileWriter.close()
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// CheckErrs 检查是否为需要重试的错误
|
||||
|
||||
Reference in New Issue
Block a user