elektron/elektronLogging/ElektronFormatter.go

package elektronLogging

import (
	"bytes"
	"fmt"
	"github.com/fatih/color"
	log "github.com/sirupsen/logrus"
	"strings"
)

type ElektronFormatter struct {
	TimestampFormat string
}

func (f ElektronFormatter) getColor(entry *log.Entry) *color.Color {
	switch entry.Level {
	case log.InfoLevel:
		return color.New(color.FgGreen, color.Bold)
	case log.WarnLevel:
		return color.New(color.FgYellow, color.Bold)
	case log.ErrorLevel:
		return color.New(color.FgRed, color.Bold)
	case log.FatalLevel:
		return color.New(color.FgRed, color.Bold)
	default:
		return color.New(color.FgWhite, color.Bold)
	}
}
func (f ElektronFormatter) Format(entry *log.Entry) ([]byte, error) {
	var b *bytes.Buffer

	if entry.Buffer != nil {
		b = entry.Buffer
	} else {
		b = &bytes.Buffer{}
	}

	levelColor := f.getColor(entry)
	level := levelColor.Sprintf("[%s]:", strings.ToUpper(entry.Level.String()))
	message := fmt.Sprintf("%s %s ", level, entry.Time.Format(f.TimestampFormat))
	if entry.Message != "" {
		message = fmt.Sprintf("%s %s %s ", level, entry.Time.Format(f.TimestampFormat), entry.Message)
	}
	var formattedFields []string
	for key, value := range entry.Data {
		formattedFields = append(formattedFields,
			strings.Join([]string{key, fmt.Sprintf("%s", value)}, "="))
	}

	b.WriteString(message)
	b.WriteString(strings.Join(formattedFields, ", "))
	b.WriteByte('\n')
	return b.Bytes(), nil
}