hook.go

package coralogix

import (
	"github.com/sirupsen/logrus"
)

// Hook is presenting Coralogix Logger for Logrus library
type Hook struct {
	Writer LoggerManager
}

// NewCoralogixHook build Coralogix logger hook
func NewCoralogixHook(PrivateKey string, ApplicationName string, SubsystemName string) *Hook {
	NewHookInstance := &Hook{
		*NewLoggerManager(
			PrivateKey,
			ApplicationName,
			SubsystemName,
			true,
		),
	}

	go NewHookInstance.Writer.Run()

	return NewHookInstance
}

// Fire send message to Coralogix
func (hook *Hook) Fire(entry *logrus.Entry) error {
	var (
		Level      uint
		Text       interface{}
		Category   string
		ClassName  string
		MethodName string
		ThreadID   string
	)

	switch entry.Level {
	case logrus.TraceLevel:
		Level = 1
	case logrus.DebugLevel:
		Level = 1
	case logrus.InfoLevel:
		Level = 3
	case logrus.WarnLevel:
		Level = 4
	case logrus.ErrorLevel:
		Level = 5
	case logrus.FatalLevel:
		Level = 6
	case logrus.PanicLevel:
		Level = 6
	}

	if Value, Exist := entry.Data["Category"]; Exist {
		Category = Value.(string)
		delete(entry.Data, "Category")
	} else {
		Category = ""
	}

	if Value, Exist := entry.Data["ClassName"]; Exist {
		ClassName = Value.(string)
		delete(entry.Data, "ClassName")
	} else {
		if entry.HasCaller() {
			ClassName = entry.Caller.File
		} else {
			ClassName = ""
		}
	}

	if Value, Exist := entry.Data["MethodName"]; Exist {
		MethodName = Value.(string)
		delete(entry.Data, "MethodName")
	} else {
		if entry.HasCaller() {
			MethodName = entry.Caller.Function
		} else {
			MethodName = ""
		}
	}

	if Value, Exist := entry.Data["ThreadId"]; Exist {
		ThreadID = Value.(string)
		delete(entry.Data, "ThreadId")
	} else {
		ThreadID = ""
	}

	if len(entry.Data) > 0 {
		Text = map[string]interface{}{
			"message": entry.Message,
			"fields":  entry.Data,
		}
	} else {
		Text = entry.Message
	}

	hook.Writer.LogsBuffer.Append(Log{
		float64(entry.Time.Unix()) * 1000.0,
		Level,
		MessageToString(Text),
		Category,
		ClassName,
		MethodName,
		ThreadID,
		0,
	})

	if entry.Level == logrus.FatalLevel || entry.Level == logrus.PanicLevel {
		hook.Writer.Flush()
	}

	return nil
}

// Levels return levels which can be sent with this hook
func (hook *Hook) Levels() []logrus.Level {
	return logrus.AllLevels
}

// Close is a defer function for buffer cleanup before exit
func (hook *Hook) Close() {
	hook.Writer.Stop()
}