diff --git a/cmd/start/start.go b/cmd/start/start.go index 240846e..d9b5ae4 100644 --- a/cmd/start/start.go +++ b/cmd/start/start.go @@ -4,8 +4,10 @@ import ( "fmt" "os" "sysmonitord/internal/config" + "sysmonitord/pkg/logger" "github.com/spf13/cobra" + "go.uber.org/zap" ) var StartCmd = &cobra.Command{ @@ -13,17 +15,17 @@ var StartCmd = &cobra.Command{ Short: "启动系统监控守护服务", Long: "sysmonitord start 命令用于启动系统监控守护服务,首次启动会进行全量扫描建立白名单。", Run: func(cmd *cobra.Command, args []string) { - fmt.Println("正在启动系统监控守护服务...") + logger.Log.Info("正在启动系统监控守护服务...") cfg, err := config.LoadConfig("./config.yaml") if err != nil { - fmt.Println("加载配置文件失败:", err) + logger.Log.Error("加载配置文件失败", zap.Error(err)) os.Exit(1) } - fmt.Println("配置文件加载成功") - fmt.Printf("审计服务器地址:%s:%d\n", cfg.Audit.Server, cfg.Audit.Port) - + logger.Log.Info("配置文件加载成功", + zap.String("审计服务器地址", fmt.Sprintf("%s:%d", cfg.Audit.Server, cfg.Audit.Port)), + ) // Todo: 初始化扫描 }, } diff --git a/go.mod b/go.mod index d15a5e6..61dfcd0 100644 --- a/go.mod +++ b/go.mod @@ -6,5 +6,7 @@ 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 ) diff --git a/go.sum b/go.sum index a2f0a39..e1f4e4b 100644 --- a/go.sum +++ b/go.sum @@ -7,6 +7,10 @@ github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiT 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= diff --git a/main.go b/main.go index d139367..8c1efce 100644 --- a/main.go +++ b/main.go @@ -1,14 +1,18 @@ package main import ( - "fmt" "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 系统安全监控工具", @@ -17,7 +21,7 @@ func main() { rootCmd.AddCommand(start.StartCmd) if err := rootCmd.Execute(); err != nil { - fmt.Println(err) + logger.Log.Error("命令执行失败", zap.Error(err)) os.Exit(1) } } diff --git a/pkg/logger/logger.go b/pkg/logger/logger.go new file mode 100644 index 0000000..b0cfcc3 --- /dev/null +++ b/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() + } +}