mirror of
https://github.com/langhuihui/monibuca.git
synced 2025-12-24 13:48:04 +08:00
39 lines
1.3 KiB
Go
39 lines
1.3 KiB
Go
package plugin_logrotate
|
|
|
|
import (
|
|
"io"
|
|
"log/slog"
|
|
|
|
"github.com/alchemy/rotoslog"
|
|
|
|
"github.com/phsym/console-slog"
|
|
"m7s.live/v5"
|
|
"m7s.live/v5/pkg"
|
|
"m7s.live/v5/plugin/logrotate/pb"
|
|
)
|
|
|
|
type LogRotatePlugin struct {
|
|
pb.UnimplementedApiServer
|
|
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.Api_ServiceDesc, pb.RegisterApiHandler)
|
|
|
|
func (config *LogRotatePlugin) OnInit() (err error) {
|
|
builder := func(w io.Writer, opts *slog.HandlerOptions) slog.Handler {
|
|
return console.NewHandler(w, &console.HandlerOptions{NoColor: true, Level: pkg.ParseLevel(config.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.AddLogHandler(config.handler)
|
|
}
|
|
return
|
|
}
|