add code

Author: lelvisl

logger.go

@@ -0,0 +1,79 @@
+package logger
+
+import (
+	"context"
+	"os"
+
+	log "github.com/sirupsen/logrus"
+)
+
+//LogConfig is a configuration for logger
+type LogConfig struct {
+	Title       string `yaml:"title" json:"title"`
+	Type        string `yaml:"type" json:"type"`
+	NetworkType string `yaml:"network type" json:"network type"`
+	Host        string `yaml:"host" json:"host"`
+	Severity    string `yaml:"severity" json:"severity"`
+	Facility    string `yaml:"facility" json:"facility"`
+	Port        string `yaml:"port" json:"port"`
+	FilePath    string `yaml:"file path" json:"file path"`
+	FileName    string `yaml:"file name" json:"file name"`
+	DebugMode   bool   `yaml:"debug mode" json:"debug mode"`
+}
+
+type ctxlog struct{}
+
+//WithLogger put logger to context
+func WithLogger(ctx context.Context, logger *log.Logger) context.Context {
+	return context.WithValue(ctx, ctxlog{}, logger)
+}
+
+//Logger get logger from context
+func Logger(ctx context.Context) *log.Logger {
+	//	l, _ := ctx.Value("logger").(*log.Logger)
+	l, ok := ctx.Value(ctxlog{}).(*log.Logger)
+	if !ok {
+		//return DefaultLogger
+		l = initLogger(LogConfig{Type: "stdout", Severity: "LOG_INFO"})
+	}
+	return l
+}
+
+func cleateLogger(config LogConfig) *log.Logger {
+	logger := initLogger(config)
+	return logger
+}
+
+func initLogger(config LogConfig) *log.Logger {
+	logger := log.New()
+	switch config.Type {
+	case "syslog":
+		logger = initSyslogger(config)
+	case "logstash":
+		logger = initLogstashlogger(config)
+	case "stdout":
+		logger.Out = os.Stdout
+		return logger
+	case "stderr":
+		logger.Out = os.Stderr
+		return logger
+	default:
+	}
+	if config.DebugMode {
+		logger.Out = os.Stdout
+	}
+	logger.Formatter = &log.TextFormatter{}
+	logger.Level = logLevel[config.Severity]
+	return logger
+}
+
+var logLevel = map[string]log.Level{
+	"LOG_EMERG":   log.PanicLevel,
+	"LOG_ALERT":   log.PanicLevel,
+	"LOG_CRIT":    log.FatalLevel,
+	"LOG_ERR":     log.ErrorLevel,
+	"LOG_WARNING": log.WarnLevel,
+	"LOG_NOTICE":  log.InfoLevel,
+	"LOG_INFO":    log.InfoLevel,
+	"LOG_DEBUG":   log.DebugLevel,
+}

logstash.go

@@ -0,0 +1,24 @@
+package logger
+
+import (
+	"net"
+	"strings"
+
+	"github.com/bshuster-repo/logrus-logstash-hook"
+	log "github.com/sirupsen/logrus"
+)
+
+func initLogstashlogger(config LogConfig) *log.Logger {
+	logger := log.New()
+	conn, err := net.Dial(config.NetworkType, config.Host+":"+config.Port)
+	if err != nil {
+		log.Fatal(err)
+	}
+	hook := logrustash.New(conn, logrustash.DefaultFormatter(log.Fields{"type": strings.ToLower(config.Title)}))
+	if err != nil {
+		log.Fatal(err)
+	}
+	logger.Hooks.Add(hook)
+	//logger.Out = ioutil.Discard
+	return logger
+}

syslogger.go

@@ -0,0 +1,63 @@
+package logger
+
+import (
+	"io/ioutil"
+	"log/syslog"
+
+	log "github.com/sirupsen/logrus"
+	logrus_syslog "github.com/sirupsen/logrus/hooks/syslog"
+)
+
+func initSyslogger(config LogConfig) *log.Logger {
+	var logSeverity = map[string]syslog.Priority{
+		"LOG_EMERG":   syslog.LOG_EMERG,
+		"LOG_ALERT":   syslog.LOG_ALERT,
+		"LOG_CRIT":    syslog.LOG_CRIT,
+		"LOG_ERR":     syslog.LOG_ERR,
+		"LOG_WARNING": syslog.LOG_WARNING,
+		"LOG_NOTICE":  syslog.LOG_NOTICE,
+		"LOG_INFO":    syslog.LOG_INFO,
+		"LOG_DEBUG":   syslog.LOG_DEBUG,
+	}
+	var logFacility = map[string]syslog.Priority{
+		"LOG_KERN":     syslog.LOG_KERN,
+		"LOG_USER":     syslog.LOG_USER,
+		"LOG_MAIL":     syslog.LOG_MAIL,
+		"LOG_DAEMON":   syslog.LOG_DAEMON,
+		"LOG_AUTH":     syslog.LOG_AUTH,
+		"LOG_SYSLOG":   syslog.LOG_SYSLOG,
+		"LOG_LPR":      syslog.LOG_LPR,
+		"LOG_NEWS":     syslog.LOG_NEWS,
+		"LOG_UUCP":     syslog.LOG_UUCP,
+		"LOG_CRON":     syslog.LOG_CRON,
+		"LOG_AUTHPRIV": syslog.LOG_AUTHPRIV,
+		"LOG_FTP":      syslog.LOG_FTP,
+
+		"LOG_LOCAL0": syslog.LOG_LOCAL0,
+		"LOG_LOCAL1": syslog.LOG_LOCAL1,
+		"LOG_LOCAL2": syslog.LOG_LOCAL2,
+		"LOG_LOCAL3": syslog.LOG_LOCAL3,
+		"LOG_LOCAL4": syslog.LOG_LOCAL4,
+		"LOG_LOCAL5": syslog.LOG_LOCAL5,
+		"LOG_LOCAL6": syslog.LOG_LOCAL6,
+		"LOG_LOCAL7": syslog.LOG_LOCAL7,
+	}
+	logger := log.New()
+	hook, err := logrus_syslog.NewSyslogHook(
+		config.NetworkType,
+		config.Host+":"+config.Port,
+		logSeverity[config.Severity]|logFacility[config.Facility],
+		config.Title)
+	if err != nil {
+		log.Errorln(err)
+		hook, err = logrus_syslog.NewSyslogHook(
+			"", "", logSeverity[config.Severity]|logFacility[config.Facility],
+			config.Title)
+
+	} else {
+		logger.Hooks.Add(hook)
+	}
+
+	logger.Out = ioutil.Discard
+	return logger
+}