OLD | NEW |
(Empty) | |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 package prod |
| 6 |
| 7 import ( |
| 8 "fmt" |
| 9 |
| 10 "github.com/luci/luci-go/common/logging" |
| 11 "golang.org/x/net/context" |
| 12 "google.golang.org/appengine/log" |
| 13 ) |
| 14 |
| 15 // useLogging adds a logging.Logger implementation to the context which logs to |
| 16 // appengine's log handler. |
| 17 func useLogging(c context.Context) context.Context { |
| 18 return logging.SetFactory(c, func(ic context.Context) logging.Logger { |
| 19 return &loggerImpl{AEContext(ic)} |
| 20 }) |
| 21 } |
| 22 |
| 23 type loggerImpl struct { |
| 24 aeCtx context.Context |
| 25 } |
| 26 |
| 27 func (gl *loggerImpl) Debugf(format string, args ...interface{}) { |
| 28 gl.LogCall(logging.Debug, 1, format, args) |
| 29 } |
| 30 func (gl *loggerImpl) Infof(format string, args ...interface{}) { |
| 31 gl.LogCall(logging.Info, 1, format, args) |
| 32 } |
| 33 func (gl *loggerImpl) Warningf(format string, args ...interface{}) { |
| 34 gl.LogCall(logging.Warning, 1, format, args) |
| 35 } |
| 36 func (gl *loggerImpl) Errorf(format string, args ...interface{}) { |
| 37 gl.LogCall(logging.Error, 1, format, args) |
| 38 } |
| 39 |
| 40 // TODO(riannucci): prefix with caller's code location. |
| 41 func (gl *loggerImpl) LogCall(l logging.Level, calldepth int, format string, arg
s []interface{}) { |
| 42 if gl.aeCtx == nil || !logging.IsLogging(gl.aeCtx, l) { |
| 43 return |
| 44 } |
| 45 |
| 46 var logf func(context.Context, string, ...interface{}) |
| 47 switch l { |
| 48 case logging.Debug: |
| 49 logf = log.Debugf |
| 50 case logging.Info: |
| 51 logf = log.Infof |
| 52 case logging.Warning: |
| 53 logf = log.Warningf |
| 54 |
| 55 case logging.Error: |
| 56 fallthrough |
| 57 default: |
| 58 logf = log.Errorf |
| 59 } |
| 60 |
| 61 fields := logging.GetFields(gl.aeCtx) |
| 62 if len(fields) > 0 { |
| 63 logf(gl.aeCtx, "%s :: %s", fmt.Sprintf(format, args...), fields.
FieldString(true)) |
| 64 } else { |
| 65 logf(gl.aeCtx, format, args...) |
| 66 } |
| 67 } |
OLD | NEW |