kira_log.go

package kira

import (
	"io"
	"os"

	"github.com/go-kira/kira/modules/config"
	"github.com/go-kira/kira/modules/log"
	lumberjack "gopkg.in/natefinch/lumberjack.v2"
)

func setupLogger(config *config.Config, w io.Writer, fields log.Fields) *log.Logger {
	logger := log.New(w, setupFormatter(config), fields)
	logger.SetLevel(log.LevelStrings[config.GetString("log.level", "info")])

	return logger
}

func setupWriter(config *config.Config) io.Writer {
	switch config.GetString("log.log") {
	case "stderr":
		return os.Stderr
	case "stdin":
		return os.Stdin
	case "stdout":
		return os.Stdout
	case "file":
		return &lumberjack.Logger{
			Filename:   config.GetString("log.file", "./storage/logs/kira.log"),
			MaxSize:    config.GetInt("log.file_max_size", 100),
			MaxBackups: config.GetInt("log.file_max_backups", 3),
			MaxAge:     config.GetInt("log.file_max_age", 28),
			Compress:   config.GetBool("log.file_max_compress", false),
		}
	}

	return os.Stderr
}

// setupFormatter to setup the logger formatter.
func setupFormatter(config *config.Config) log.Formatter {
	switch config.GetString("log.formatter") {
	case "default":
	case "cli":
		return log.NewDefaultFormatter()
	case "json":
		return log.NewJSONFormatter()
	}

	return log.NewDefaultFormatter()
}