mirror of
https://github.com/langhuihui/monibuca.git
synced 2026-05-08 15:01:05 +08:00
langhuihui
44 lines
1.5 KiB
Go
44 lines
1.5 KiB
Go
package plugin_logrotate
|
|
|
|
import (
|
|
"io"
|
|
"log/slog"
|
|
|
|
"github.com/alchemy/rotoslog"
|
|
|
|
"github.com/phsym/console-slog"
|
|
"m7s.live/m7s/v5"
|
|
"m7s.live/m7s/v5/pkg"
|
|
"m7s.live/m7s/v5/plugin/logrotate/pb"
|
|
)
|
|
|
|
type LogRotatePlugin struct {
|
|
pb.UnimplementedLogrotateServer
|
|
m7s.Plugin
|
|
Path string `default:"./logs" desc:"日志文件存放目录"`
|
|
Size uint64 `default:"1048576" desc:"日志文件大小，单位：字节"`
|
|
Days int `default:"1" desc:"日志文件保留天数"`
|
|
Formatter string `default:"2006-01-02T15" desc:"日志文件名格式"`
|
|
MaxFiles uint64 `default:"7" desc:"最大日志文件数量"`
|
|
Level string `default:"info" desc:"日志级别"`
|
|
handler slog.Handler
|
|
}
|
|
|
|
var _ = m7s.InstallPlugin[LogRotatePlugin](&pb.Logrotate_ServiceDesc, pb.RegisterLogrotateHandler)
|
|
|
|
func (config *LogRotatePlugin) OnInit() (err error) {
|
|
var lv slog.LevelVar
|
|
lv.UnmarshalText([]byte(config.Level))
|
|
if config.Level == "trace" {
|
|
lv.Set(pkg.TraceLevel)
|
|
}
|
|
builder := func(w io.Writer, opts *slog.HandlerOptions) slog.Handler {
|
|
return console.NewHandler(w, &console.HandlerOptions{NoColor: true, Level: lv.Level(),TimeFormat: "2006-01-02 15:04:05.000"})
|
|
}
|
|
config.handler, err = rotoslog.NewHandler(rotoslog.LogHandlerBuilder(builder), rotoslog.LogDir(config.Path), rotoslog.MaxFileSize(config.Size), rotoslog.DateTimeLayout(config.Formatter), rotoslog.MaxRotatedFiles(config.MaxFiles))
|
|
if err == nil {
|
|
config.PostToServer(config.handler)
|
|
}
|
|
return
|
|
}
|