61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
|
package logs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
// LogJSONFormatter is a custom log formatter for JSON output
|
||
|
|
type LogJSONFormatter struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
// Format formats the log entry to JSON
|
||
|
|
func (f *LogJSONFormatter) Format(entry *logrus.Entry) ([]byte, error) {
|
||
|
|
data := make(logrus.Fields)
|
||
|
|
|
||
|
|
// Include the log level, message, and time in the JSON output
|
||
|
|
data["level"] = entry.Level.String()
|
||
|
|
data["msg"] = entry.Message
|
||
|
|
data["time"] = entry.Time.UTC().Format(time.RFC3339)
|
||
|
|
|
||
|
|
// Include additional fields from the entry
|
||
|
|
for k, v := range entry.Data {
|
||
|
|
data[k] = v
|
||
|
|
}
|
||
|
|
|
||
|
|
// Marshal the JSON data
|
||
|
|
jsonData, err := json.Marshal(data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to marshal log entry to JSON: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Append a newline to the JSON data
|
||
|
|
return append(jsonData, '\n'), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ParseLogLine parses the log line and extracts the log level, message, and time
|
||
|
|
func ParseLogLine(line string) (logrus.Level, string, time.Time, error) {
|
||
|
|
var entry struct {
|
||
|
|
Level string `json:"level"`
|
||
|
|
Msg string `json:"msg"`
|
||
|
|
Time time.Time `json:"time"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Unmarshal the JSON data
|
||
|
|
err := json.Unmarshal([]byte(line), &entry)
|
||
|
|
if err != nil {
|
||
|
|
return logrus.InfoLevel, "", time.Time{}, fmt.Errorf("failed to unmarshal log line: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Convert the log level string to logrus.Level
|
||
|
|
level, err := logrus.ParseLevel(entry.Level)
|
||
|
|
if err != nil {
|
||
|
|
return logrus.InfoLevel, "", time.Time{}, fmt.Errorf("failed to parse log level: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return level, entry.Msg, entry.Time, nil
|
||
|
|
}
|