123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116 |
- package logs
- import (
- "encoding/json"
- "time"
- )
- type multiFileLogWriter struct {
- writers [LevelDebug + 1 + 1]*fileLogWriter
- fullLogWriter *fileLogWriter
- Separate []string `json:"separate"`
- }
- var levelNames = [...]string{"emergency", "alert", "critical", "error", "warning", "notice", "info", "debug"}
- func (f *multiFileLogWriter) Init(config string) error {
- writer := newFileWriter().(*fileLogWriter)
- err := writer.Init(config)
- if err != nil {
- return err
- }
- f.fullLogWriter = writer
- f.writers[LevelDebug+1] = writer
-
- json.Unmarshal([]byte(config), f)
- jsonMap := map[string]interface{}{}
- json.Unmarshal([]byte(config), &jsonMap)
- for i := LevelEmergency; i < LevelDebug+1; i++ {
- for _, v := range f.Separate {
- if v == levelNames[i] {
- jsonMap["filename"] = f.fullLogWriter.fileNameOnly + "." + levelNames[i] + f.fullLogWriter.suffix
- jsonMap["level"] = i
- bs, _ := json.Marshal(jsonMap)
- writer = newFileWriter().(*fileLogWriter)
- writer.Init(string(bs))
- f.writers[i] = writer
- }
- }
- }
- return nil
- }
- func (f *multiFileLogWriter) Destroy() {
- for i := 0; i < len(f.writers); i++ {
- if f.writers[i] != nil {
- f.writers[i].Destroy()
- }
- }
- }
- func (f *multiFileLogWriter) WriteMsg(when time.Time, msg string, level int) error {
- if f.fullLogWriter != nil {
- f.fullLogWriter.WriteMsg(when, msg, level)
- }
- for i := 0; i < len(f.writers)-1; i++ {
- if f.writers[i] != nil {
- if level == f.writers[i].Level {
- f.writers[i].WriteMsg(when, msg, level)
- }
- }
- }
- return nil
- }
- func (f *multiFileLogWriter) Flush() {
- for i := 0; i < len(f.writers); i++ {
- if f.writers[i] != nil {
- f.writers[i].Flush()
- }
- }
- }
- func newFilesWriter() Logger {
- return &multiFileLogWriter{}
- }
- func init() {
- Register(AdapterMultiFile, newFilesWriter)
- }
|