chore: use zerolog instead of logrus

Author: efectn

handler/document.go

@@ -10,7 +10,7 @@ import (
 	"github.com/armbian/ansi-hastebin/internal/keygenerator"
 	"github.com/armbian/ansi-hastebin/internal/storage"
 	"github.com/go-chi/chi/v5"
-	"github.com/sirupsen/logrus"
+	"github.com/rs/zerolog/log"
 )
 
 // DocumentHandler manages document operations
@@ -50,34 +50,34 @@ func (h *DocumentHandler) HandleGet(w http.ResponseWriter, r *http.Request) {
 	data, err := h.Store.Get(key, false)
 
 	if data != "" && err == nil {
-		logrus.WithField("key", key).Info("Retrieved document")
+		log.Info().Str("key", key).Msg("Retrieved document")
 		w.Header().Set("Content-Type", "application/json")
 		if r.Method == http.MethodHead {
 			w.WriteHeader(http.StatusOK)
 			return
 		}
 		json.NewEncoder(w).Encode(map[string]string{"data": data, "key": key})
 	} else {
-		logrus.WithField("key", key).Warn("Document not found")
+		log.Info().Str("key", key).Msg("Document not found")
 		http.Error(w, `{"message": "Document not found."}`, http.StatusNotFound)
 	}
 }
 
 // Handle retrieving raw document
 func (h *DocumentHandler) HandleRawGet(w http.ResponseWriter, r *http.Request) {
-
 	key := strings.Split(chi.URLParam(r, "id"), ".")[0]
 	data, err := h.Store.Get(key, false)
+
 	if data != "" && err == nil {
-		logrus.WithField("key", key).Info("Retrieved raw document")
+		log.Info().Str("key", key).Msg("Retrieved raw document")
 		w.Header().Set("Content-Type", "text/plain; charset=UTF-8")
 		if r.Method == http.MethodHead {
 			w.WriteHeader(http.StatusOK)
 			return
 		}
 		w.Write([]byte(data))
 	} else {
-		logrus.WithField("key", key).Warn("Raw document not found")
+		log.Info().Str("key", key).Msg("Raw document not found")
 		http.Error(w, `{"message": "Document not found."}`, http.StatusNotFound)
 	}
 }
@@ -91,15 +91,15 @@ func (h *DocumentHandler) HandlePost(w http.ResponseWriter, r *http.Request) {
 	}
 
 	if h.MaxLength > 0 && buffer.Len() > h.MaxLength {
-		logrus.Warn("Document exceeds max length")
+		log.Info().Str("key", "").Msg("Document exceeds max length")
 		http.Error(w, `{"message": "Document exceeds maximum length."}`, http.StatusBadRequest)
 		return
 	}
 
 	key := h.KeyGenerator.Generate(h.KeyLength)
 	h.Store.Set(key, buffer.String(), false)
 
-	logrus.WithField("key", key).Info("Added document")
+	log.Info().Str("key", key).Msg("Added document")
 	w.Header().Set("Content-Type", "application/json")
 	json.NewEncoder(w).Encode(map[string]string{"key": key})
 }
@@ -113,15 +113,15 @@ func (h *DocumentHandler) HandlePutLog(w http.ResponseWriter, r *http.Request) {
 	}
 
 	if h.MaxLength > 0 && buffer.Len() > h.MaxLength {
-		logrus.Warn("Document exceeds max length")
+		log.Info().Str("key", "").Msg("Document exceeds max length")
 		http.Error(w, `{"message": "Document exceeds maximum length."}`, http.StatusBadRequest)
 		return
 	}
 
 	key := h.KeyGenerator.Generate(h.KeyLength)
 	h.Store.Set(key, buffer.String(), false)
 
-	logrus.WithField("key", key).Info("Added document with log link")
+	log.Info().Str("key", key).Msg("Added document with log link")
 	w.Header().Set("Content-Type", "text/plain")
 	fmt.Fprintf(w, "\nhttps://%s/%s\n\n", r.Host, key)
 }
@@ -136,7 +136,7 @@ func (h *DocumentHandler) readBody(r *http.Request, buffer *strings.Builder) err
 	} else {
 		data, err := io.ReadAll(r.Body)
 		if err != nil {
-			logrus.Error("Connection error: ", err)
+			log.Error().Err(err).Msg("Error reading request body")
 			return err
 		}
 		buffer.WriteString(string(data))

handler/document_test.go

@@ -10,7 +10,8 @@ import (
 	"testing"
 
 	"github.com/go-chi/chi/v5"
-	"github.com/sirupsen/logrus"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
 	"github.com/stretchr/testify/require"
 )
 
@@ -176,7 +177,7 @@ func BenchmarkHandlePost(b *testing.B) {
 	handler.RegisterRoutes(router)
 
 	// Disable logging for benchmark
-	logrus.StandardLogger().Level = logrus.FatalLevel
+	log.Logger = log.Level(zerolog.Disabled)
 
 	body := bytes.NewBufferString("benchmark content")
 	for i := 0; i < b.N; i++ {
@@ -190,7 +191,7 @@ func BenchmarkHandleGet(b *testing.B) {
 	handler.RegisterRoutes(router)
 
 	// Disable logging for benchmark
-	logrus.StandardLogger().Level = logrus.FatalLevel
+	log.Logger = log.Level(zerolog.Disabled)
 
 	// Add document
 	handler.Store.Set("test123", "benchmark data", false)
@@ -206,7 +207,7 @@ func BenchmarkHandlePutLog(b *testing.B) {
 	handler.RegisterRoutes(router)
 
 	// Disable logging for benchmark
-	logrus.StandardLogger().Level = logrus.FatalLevel
+	log.Logger = log.Level(zerolog.Disabled)
 
 	body := bytes.NewBufferString("benchmark log entry")
 	for i := 0; i < b.N; i++ {
@@ -220,7 +221,7 @@ func BenchmarkHandleRawGet(b *testing.B) {
 	handler.RegisterRoutes(router)
 
 	// Disable logging for benchmark
-	logrus.StandardLogger().Level = logrus.FatalLevel
+	log.Logger = log.Level(zerolog.Disabled)
 
 	// Add document
 	handler.Store.Set("test123", "benchmark data", false)