github.com/pawelgaczynski/gain@v0.4.0-alpha.0.20230821120126-41f1e60a18da/logger/logger.go (about)

     1  // Copyright (c) 2023 Paweł Gaczyński
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  
    15  package logger
    16  
    17  import (
    18  	"os"
    19  
    20  	"github.com/rs/zerolog"
    21  )
    22  
    23  const (
    24  	DebugLevel = zerolog.DebugLevel
    25  	InfoLevel  = zerolog.InfoLevel
    26  	WarnLevel  = zerolog.WarnLevel
    27  	ErrorLevel = zerolog.ErrorLevel
    28  	FatalLevel = zerolog.FatalLevel
    29  	PanicLevel = zerolog.PanicLevel
    30  	Disabled   = zerolog.Disabled
    31  	TraceLevel = zerolog.TraceLevel
    32  	NoLevel    = zerolog.NoLevel
    33  )
    34  
    35  func NewLogger(component string, level zerolog.Level, pretty bool) zerolog.Logger {
    36  	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
    37  	logger := zerolog.New(os.Stdout).With().Timestamp().Str("component", component).Logger().Level(level)
    38  
    39  	if pretty {
    40  		logger = logger.Output(zerolog.ConsoleWriter{Out: os.Stderr})
    41  	}
    42  
    43  	return logger
    44  }