seba-631-632: exporter enhancements for logger and topics
Change-Id: If10e56a7ccfce758712ea02df9656d4f413dbf84
diff --git a/common/logger/logger.go b/common/logger/logger.go
new file mode 100644
index 0000000..6f9d4c4
--- /dev/null
+++ b/common/logger/logger.go
@@ -0,0 +1,110 @@
+/*
+ * Copyright 2018-present Open Networking Foundation
+
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+
+ * http://www.apache.org/licenses/LICENSE-2.0
+
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package logger
+
+import (
+ lkh "github.com/gfremex/logrus-kafka-hook"
+ log "github.com/sirupsen/logrus"
+ "time"
+)
+
+var (
+ myLogger *log.Entry
+)
+
+func Setup(kafkaBroker string, level string) {
+
+ logger := log.New()
+ //logger.SetReportCaller(true)
+ myLogger = logger.WithField("topics", []string{"kafka-exporter.log"})
+
+ var logLevel log.Level = log.DebugLevel
+ switch level {
+ case "TRACE":
+ logLevel = log.TraceLevel
+ case "INFO":
+ logLevel = log.InfoLevel
+ case "WARN":
+ logLevel = log.WarnLevel
+ case "ERROR":
+ logLevel = log.ErrorLevel
+ default:
+ logLevel = log.DebugLevel
+ }
+ logger.Println("Setting Log Level", logLevel)
+ logger.SetLevel(logLevel)
+
+ if len(kafkaBroker) > 0 {
+ myLogger.Debug("Setting up kafka integration")
+ hook, err := lkh.NewKafkaHook(
+ "kh",
+ []log.Level{log.DebugLevel, log.InfoLevel, log.WarnLevel, log.ErrorLevel},
+ &log.JSONFormatter{
+ TimestampFormat: time.RFC3339Nano,
+ FieldMap: log.FieldMap{
+ log.FieldKeyTime: "@timestamp",
+ log.FieldKeyLevel: "levelname",
+ log.FieldKeyMsg: "message",
+ },
+ },
+ []string{kafkaBroker},
+ )
+
+ if err != nil {
+ myLogger.Error(err)
+ }
+
+ logger.Hooks.Add(hook)
+ myLogger.WithField("kafkaBroker", kafkaBroker).Debug("Logger setup done")
+ }
+}
+
+func GetLogger() *log.Entry {
+ return myLogger
+}
+
+func WithField(key string, value interface{}) *log.Entry {
+ return myLogger.WithField(key, value)
+}
+
+func WithFields(fields log.Fields) *log.Entry {
+ return myLogger.WithFields(fields)
+}
+
+func Panic(msg string, args ...interface{}) {
+ myLogger.Panicf(msg, args...)
+}
+
+func Fatal(msg string, args ...interface{}) {
+ myLogger.Fatalf(msg, args...)
+}
+
+func Error(msg string, args ...interface{}) {
+ myLogger.Errorf(msg, args...)
+}
+
+func Warn(msg string, args ...interface{}) {
+ myLogger.Warnf(msg, args...)
+}
+
+func Info(msg string, args ...interface{}) {
+ myLogger.Infof(msg, args...)
+}
+
+func Debug(msg string, args ...interface{}) {
+ myLogger.Debugf(msg, args...)
+}