[config] 规范代码

internal/config/config.go

@@ -1,14 +1,5 @@
 package config
 
-import (
-	"fmt"
-	"os"
-	"sysmonitord/pkg/logger"
-
-	"go.uber.org/zap"
-	"gopkg.in/yaml.v3"
-)
-
 type Config struct {
 	Audit   AuditConfig   `yaml:"audit"`
 	Scanner ScannerConfig `yaml:"scanner"`
@@ -46,34 +37,3 @@ type FileScannerConfig struct {
 	FastHashSize     int64
 	FastHashChunk    int64
 }
-
-func LoadConfig(path string) (*Config, error) {
-	data, err := os.ReadFile(path)
-	if err != nil {
-		return nil, fmt.Errorf("无法读取配置文件： %w", err)
-	}
-
-	var cfg Config
-	if err := yaml.Unmarshal(data, &cfg); err != nil {
-		return nil, fmt.Errorf("无法解析配置文件： %w", err)
-	}
-
-	// 解析 FastHashSize
-	cfg.Scanner.File.FastHashSize, err = ParseSize(cfg.Scanner.File.FastHashSizeRaw)
-	if err != nil {
-		return nil, fmt.Errorf("解析 fast_hash_size 失败: %w", err)
-	}
-
-	// 解析 FastHashChunk
-	cfg.Scanner.File.FastHashChunk, err = ParseSize(cfg.Scanner.File.FastHashChunkRaw)
-	if err != nil {
-		return nil, fmt.Errorf("解析 fast_hash_chunk 失败: %w", err)
-	}
-
-	logger.Log.Debug("配置加载完成",
-		zap.Int64("fast_hash_size", cfg.Scanner.File.FastHashSize),
-		zap.Int64("fast_hash_chunk", cfg.Scanner.File.FastHashChunk),
-	)
-
-	return &cfg, nil
-}

internal/config/utils.go

@@ -2,13 +2,49 @@ package config
 
 import (
 	"fmt"
+	"os"
 	"regexp"
 	"strconv"
 	"strings"
 
 	"sysmonitord/internal/scanner/hash"
+	"sysmonitord/pkg/logger"
+
+	"go.uber.org/zap"
+	"gopkg.in/yaml.v3"
 )
 
+func LoadConfig(path string) (*Config, error) {
+	data, err := os.ReadFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("无法读取配置文件： %w", err)
+	}
+
+	var cfg Config
+	if err := yaml.Unmarshal(data, &cfg); err != nil {
+		return nil, fmt.Errorf("无法解析配置文件： %w", err)
+	}
+
+	// 解析 FastHashSize
+	cfg.Scanner.File.FastHashSize, err = ParseSize(cfg.Scanner.File.FastHashSizeRaw)
+	if err != nil {
+		return nil, fmt.Errorf("解析 fast_hash_size 失败: %w", err)
+	}
+
+	// 解析 FastHashChunk
+	cfg.Scanner.File.FastHashChunk, err = ParseSize(cfg.Scanner.File.FastHashChunkRaw)
+	if err != nil {
+		return nil, fmt.Errorf("解析 fast_hash_chunk 失败: %w", err)
+	}
+
+	logger.Log.Debug("配置加载完成",
+		zap.Int64("fast_hash_size", cfg.Scanner.File.FastHashSize),
+		zap.Int64("fast_hash_chunk", cfg.Scanner.File.FastHashChunk),
+	)
+
+	return &cfg, nil
+}
+
 func ParseSize(sizeStr string) (int64, error) {
 	sizeStr = strings.TrimSpace(sizeStr)
 	if sizeStr == "" {