github.com/phuslu/log@v1.0.100/multi.go (about)

     1  package log
     2  
     3  import (
     4  	"io"
     5  )
     6  
     7  // MultiWriter is an alias for MultiLevelWriter
     8  type MultiWriter = MultiLevelWriter
     9  
    10  // MultiLevelWriter is an Writer that log to different writers by different levels
    11  type MultiLevelWriter struct {
    12  	// InfoWriter specifies all the level logs writes to
    13  	InfoWriter Writer
    14  
    15  	// WarnWriter specifies the level greater than or equal to WarnLevel writes to
    16  	WarnWriter Writer
    17  
    18  	// WarnWriter specifies the level greater than or equal to ErrorLevel writes to
    19  	ErrorWriter Writer
    20  
    21  	// ConsoleWriter specifies the console writer
    22  	ConsoleWriter Writer
    23  
    24  	// ConsoleLevel specifies the level greater than or equal to it also writes to
    25  	ConsoleLevel Level
    26  }
    27  
    28  // Close implements io.Closer, and closes the underlying LeveledWriter.
    29  func (w *MultiLevelWriter) Close() (err error) {
    30  	for _, writer := range []Writer{
    31  		w.InfoWriter,
    32  		w.WarnWriter,
    33  		w.ErrorWriter,
    34  		w.ConsoleWriter,
    35  	} {
    36  		if writer == nil {
    37  			continue
    38  		}
    39  		if closer, ok := writer.(io.Closer); ok {
    40  			if err1 := closer.Close(); err1 != nil {
    41  				err = err1
    42  			}
    43  		}
    44  	}
    45  	return
    46  }
    47  
    48  // WriteEntry implements entryWriter.
    49  func (w *MultiLevelWriter) WriteEntry(e *Entry) (n int, err error) {
    50  	var err1 error
    51  	switch e.Level {
    52  	case noLevel, PanicLevel, FatalLevel, ErrorLevel:
    53  		if w.ErrorWriter != nil {
    54  			n, err1 = w.ErrorWriter.WriteEntry(e)
    55  			if err1 != nil && err == nil {
    56  				err = err1
    57  			}
    58  		}
    59  		fallthrough
    60  	case WarnLevel:
    61  		if w.WarnWriter != nil {
    62  			n, err1 = w.WarnWriter.WriteEntry(e)
    63  			if err1 != nil && err == nil {
    64  				err = err1
    65  			}
    66  		}
    67  		fallthrough
    68  	default:
    69  		if w.InfoWriter != nil {
    70  			n, err1 = w.InfoWriter.WriteEntry(e)
    71  			if err1 != nil && err == nil {
    72  				err = err1
    73  			}
    74  		}
    75  	}
    76  
    77  	if w.ConsoleWriter != nil && e.Level >= w.ConsoleLevel {
    78  		_, _ = w.ConsoleWriter.WriteEntry(e)
    79  	}
    80  
    81  	return
    82  }
    83  
    84  var _ Writer = (*MultiLevelWriter)(nil)
    85  
    86  // MultiEntryWriter is an array Writer that log to different writers
    87  type MultiEntryWriter []Writer
    88  
    89  // Close implements io.Closer, and closes the underlying MultiEntryWriter.
    90  func (w *MultiEntryWriter) Close() (err error) {
    91  	for _, writer := range *w {
    92  		if closer, ok := writer.(io.Closer); ok {
    93  			if err1 := closer.Close(); err1 != nil {
    94  				err = err1
    95  			}
    96  		}
    97  	}
    98  	return
    99  }
   100  
   101  // WriteEntry implements entryWriter.
   102  func (w *MultiEntryWriter) WriteEntry(e *Entry) (n int, err error) {
   103  	var err1 error
   104  	for _, writer := range *w {
   105  		n, err1 = writer.WriteEntry(e)
   106  		if err1 != nil && err == nil {
   107  			err = err1
   108  		}
   109  	}
   110  	return
   111  }
   112  
   113  var _ Writer = (*MultiEntryWriter)(nil)
   114  
   115  // MultiIOWriter is an array io.Writer that log to different writers
   116  type MultiIOWriter []io.Writer
   117  
   118  // Close implements io.Closer, and closes the underlying MultiIOWriter.
   119  func (w *MultiIOWriter) Close() (err error) {
   120  	for _, writer := range *w {
   121  		if closer, ok := writer.(io.Closer); ok {
   122  			if err1 := closer.Close(); err1 != nil {
   123  				err = err1
   124  			}
   125  		}
   126  	}
   127  	return
   128  }
   129  
   130  // WriteEntry implements entryWriter.
   131  func (w *MultiIOWriter) WriteEntry(e *Entry) (n int, err error) {
   132  	for _, writer := range *w {
   133  		n, err = writer.Write(e.buf)
   134  		if err != nil {
   135  			return
   136  		}
   137  	}
   138  
   139  	return
   140  }