services/ws/log.go

   1 package main
   2
   3 import (
   4         "context"
   5
   6         "github.com/Sirupsen/logrus"
   7 )
   8
   9 var (
  10         loggerCtxKey = new(int)
  11         rootLogger   = logrus.New()
  12 )
  13
  14 func contextWithLogger(ctx context.Context, logger *logrus.Entry) context.Context {
  15         return context.WithValue(ctx, loggerCtxKey, logger)
  16 }
  17
  18 func logger(ctx context.Context) *logrus.Entry {
  19         if ctx != nil {
  20                 if logger, ok := ctx.Value(loggerCtxKey).(*logrus.Entry); ok {
  21                         return logger
  22                 }
  23         }
  24         return rootLogger.WithFields(nil)
  25 }