cmd/bpa-operator/vendor/go.uber.org/zap/zapcore/tee.go

   1 // Copyright (c) 2016 Uber Technologies, Inc.
   2 //
   3 // Permission is hereby granted, free of charge, to any person obtaining a copy
   4 // of this software and associated documentation files (the "Software"), to deal
   5 // in the Software without restriction, including without limitation the rights
   6 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   7 // copies of the Software, and to permit persons to whom the Software is
   8 // furnished to do so, subject to the following conditions:
   9 //
  10 // The above copyright notice and this permission notice shall be included in
  11 // all copies or substantial portions of the Software.
  12 //
  13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  18 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  19 // THE SOFTWARE.
  20
  21 package zapcore
  22
  23 import "go.uber.org/multierr"
  24
  25 type multiCore []Core
  26
  27 // NewTee creates a Core that duplicates log entries into two or more
  28 // underlying Cores.
  29 //
  30 // Calling it with a single Core returns the input unchanged, and calling
  31 // it with no input returns a no-op Core.
  32 func NewTee(cores ...Core) Core {
  33         switch len(cores) {
  34         case 0:
  35                 return NewNopCore()
  36         case 1:
  37                 return cores[0]
  38         default:
  39                 return multiCore(cores)
  40         }
  41 }
  42
  43 func (mc multiCore) With(fields []Field) Core {
  44         clone := make(multiCore, len(mc))
  45         for i := range mc {
  46                 clone[i] = mc[i].With(fields)
  47         }
  48         return clone
  49 }
  50
  51 func (mc multiCore) Enabled(lvl Level) bool {
  52         for i := range mc {
  53                 if mc[i].Enabled(lvl) {
  54                         return true
  55                 }
  56         }
  57         return false
  58 }
  59
  60 func (mc multiCore) Check(ent Entry, ce *CheckedEntry) *CheckedEntry {
  61         for i := range mc {
  62                 ce = mc[i].Check(ent, ce)
  63         }
  64         return ce
  65 }
  66
  67 func (mc multiCore) Write(ent Entry, fields []Field) error {
  68         var err error
  69         for i := range mc {
  70                 err = multierr.Append(err, mc[i].Write(ent, fields))
  71         }
  72         return err
  73 }
  74
  75 func (mc multiCore) Sync() error {
  76         var err error
  77         for i := range mc {
  78                 err = multierr.Append(err, mc[i].Sync())
  79         }
  80         return err
  81 }