[logger]引入zap日志记录

cmd/start/start.go

@@ -0,0 +1,31 @@
+package start
+
+import (
+	"fmt"
+	"os"
+	"sysmonitord/internal/config"
+	"sysmonitord/pkg/logger"
+
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var StartCmd = &cobra.Command{
+	Use:   "start",
+	Short: "启动系统监控守护服务",
+	Long:  "sysmonitord start 命令用于启动系统监控守护服务，首次启动会进行全量扫描建立白名单。",
+	Run: func(cmd *cobra.Command, args []string) {
+		logger.Log.Info("正在启动系统监控守护服务...")
+
+		cfg, err := config.LoadConfig("./config.yaml")
+		if err != nil {
+			logger.Log.Error("加载配置文件失败", zap.Error(err))
+			os.Exit(1)
+		}
+
+		logger.Log.Info("配置文件加载成功",
+			zap.String("审计服务器地址", fmt.Sprintf("%s:%d", cfg.Audit.Server, cfg.Audit.Port)),
+		)
+		// Todo: 初始化扫描
+	},
+}

config.yaml

@@ -0,0 +1,15 @@
+server:
+  host: "127.0.0.1"
+  port: 8080
+
+audit:
+  enabled: true
+  server: "192.168.1.100"
+  port: 9000
+  buffer_size: 1000
+
+scanner:
+  file:
+  exclude_paths:
+    - /proc
+    - /sys

go.mod

@@ -0,0 +1,12 @@
+module sysmonitord
+
+go 1.26.1
+
+require (
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/spf13/cobra v1.10.2 // indirect
+	github.com/spf13/pflag v1.0.10 // indirect
+	go.uber.org/multierr v1.11.0 // indirect
+	go.uber.org/zap v1.27.1 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

go.sum

@@ -0,0 +1,17 @@
+github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.10.2 h1:DMTTonx5m65Ic0GOoRY2c16WCbHxOOw6xxezuLaBpcU=
+github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiTUUS4=
+github.com/spf13/pflag v1.0.9/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/spf13/pflag v1.0.10 h1:4EBh2KAYBwaONj6b2Ye1GiHfwjqyROoF4RwYO+vPwFk=
+github.com/spf13/pflag v1.0.10/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
+go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
+go.uber.org/zap v1.27.1 h1:08RqriUEv8+ArZRYSTXy1LeBScaMpVSTBhCeaZYfMYc=
+go.uber.org/zap v1.27.1/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E=
+go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

internal/config/config.go

@@ -0,0 +1,41 @@
+package config
+
+import (
+	"fmt"
+	"os"
+
+	"gopkg.in/yaml.v3"
+)
+
+type Config struct {
+	Audit   AuditConfig   `yaml:"audit"`
+	Scanner ScannerConfig `yaml:"scanner"`
+}
+
+type AuditConfig struct {
+	Enabled    bool   `yaml:"enabled"`
+	Server     string `yaml:"server"`
+	Port       int    `yaml:"port"`
+	BufferSize int    `yaml:"buffer_size"`
+}
+
+type ScannerConfig struct {
+	File FileScannerConfig `yaml:"file"`
+}
+
+type FileScannerConfig struct {
+	ExcludePaths []string `yaml:"exclude_paths"`
+}
+
+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)
+	}
+	return &cfg, nil
+}

main.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"os"
+	"sysmonitord/cmd/start"
+	"sysmonitord/pkg/logger"
+
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+func main() {
+	logger.InitLogger()
+	defer logger.Sync()
+
+	var rootCmd = &cobra.Command{
+		Use:   "sysmonitord",
+		Short: "Sysmonitord 是一个 Linux 系统安全监控工具",
+	}
+
+	rootCmd.AddCommand(start.StartCmd)
+
+	if err := rootCmd.Execute(); err != nil {
+		logger.Log.Error("命令执行失败", zap.Error(err))
+		os.Exit(1)
+	}
+}

pkg/logger/logger.go

@@ -0,0 +1,41 @@
+package logger
+
+import (
+	"os"
+
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+)
+
+var Log *zap.Logger
+
+func InitLogger() {
+	encoderConfig := zapcore.EncoderConfig{
+		TimeKey:        "time",
+		LevelKey:       "level",
+		NameKey:        "logger",
+		CallerKey:      "caller",
+		MessageKey:     "msg",
+		StacktraceKey:  "stacktrace",
+		LineEnding:     zapcore.DefaultLineEnding,
+		EncodeLevel:    zapcore.LowercaseLevelEncoder,
+		EncodeTime:     zapcore.ISO8601TimeEncoder,
+		EncodeDuration: zapcore.StringDurationEncoder,
+		EncodeCaller:   zapcore.ShortCallerEncoder,
+	}
+
+	// Debug
+	consoleEncoder := zapcore.NewConsoleEncoder(encoderConfig)
+
+	core := zapcore.NewTee(
+		zapcore.NewCore(consoleEncoder, zapcore.AddSync(os.Stdout), zap.DebugLevel),
+	)
+
+	Log = zap.New(core, zap.AddCaller())
+}
+
+func Sync() {
+	if Log != nil {
+		_ = Log.Sync()
+	}
+}