init

internal/utils/file.go

@@ -0,0 +1,34 @@
+package utils
+
+import (
+	"os"
+	"path/filepath"
+)
+
+// WriteFile 写入文件
+func WriteFile(filename string, data []byte) error {
+	// 确保目录存在
+	dir := filepath.Dir(filename)
+	if err := os.MkdirAll(dir, 0755); err != nil {
+		return err
+	}
+
+	// 写入文件
+	return os.WriteFile(filename, data, 0644)
+}
+
+// ReadFile 读取文件
+func ReadFile(filename string) ([]byte, error) {
+	return os.ReadFile(filename)
+}
+
+// FileExists 检查文件是否存在
+func FileExists(filename string) bool {
+	_, err := os.Stat(filename)
+	return !os.IsNotExist(err)
+}
+
+// CreateDir 创建目录
+func CreateDir(dir string) error {
+	return os.MkdirAll(dir, 0755)
+}

internal/utils/logger.go

@@ -0,0 +1,41 @@
+package utils
+
+import (
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	"gopkg.in/natefinch/lumberjack.v2"
+)
+
+type Logger struct {
+	*zap.SugaredLogger
+}
+
+func NewLogger() *Logger {
+	// 配置日志轮转
+	lumberJackLogger := &lumberjack.Logger{
+		Filename:   "logs/equipment-analyzer.log",
+		MaxSize:    100, // MB
+		MaxBackups: 3,
+		MaxAge:     28, // days
+		Compress:   true,
+	}
+
+	// 配置编码器
+	encoderConfig := zap.NewProductionEncoderConfig()
+	encoderConfig.TimeKey = "timestamp"
+	encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
+	encoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
+
+	// 创建核心
+	core := zapcore.NewCore(
+		zapcore.NewJSONEncoder(encoderConfig),
+		zapcore.AddSync(lumberJackLogger),
+		zap.NewAtomicLevelAt(zap.InfoLevel),
+	)
+
+	// 创建logger
+	logger := zap.New(core, zap.AddCaller(), zap.AddStacktrace(zapcore.ErrorLevel))
+	sugar := logger.Sugar()
+
+	return &Logger{sugar}
+}