| 13 | } |
| 14 | |
| 15 | func (f *NestedJSONFormatter) Format(entry *logrus.Entry) ([]byte, error) { |
| 16 | data := make(logrus.Fields, len(entry.Data)+4) |
| 17 | data["level"] = entry.Level.String() |
| 18 | data["msg"] = entry.Message |
| 19 | data["time"] = entry.Time |
| 20 | |
| 21 | val, ok := entry.Data["bucket"] |
| 22 | if ok { |
| 23 | data["bucket"] = val |
| 24 | } |
| 25 | |
| 26 | b := &bytes.Buffer{} |
| 27 | |
| 28 | encoder := json.NewEncoder(b) |
| 29 | encoder.SetEscapeHTML(!f.DisableHTMLEscape) |
| 30 | if f.PrettyPrint { |
| 31 | encoder.SetIndent("", " ") |
| 32 | } |
| 33 | if err := encoder.Encode(data); err != nil { |
| 34 | return nil, fmt.Errorf("failed to marshal fields to JSON, %w", err) |
| 35 | } |
| 36 | |
| 37 | return b.Bytes(), nil |
| 38 | } |