logs

Author: rolodato

client.go

@@ -64,7 +64,6 @@ func NewClient(apiKey string, options ...Option) (*Client, error) {
 	c.client = resty.
 		New().
 		OnBeforeRequest(newRestyLogRequestMiddleware(c.log)).
-		OnAfterResponse(newRestyLogResponseMiddleware(c.log)).
 		OnAfterResponse(newRestyLogResponseMiddleware(c.log))
 	c.client.SetLogger(restySlogLogger{c.log})
 
@@ -159,8 +158,6 @@ func (c *Client) GetFlags(ctx context.Context, ec EvaluationContext) (f Flags, e
 func (c *Client) UpdateEnvironment(ctx context.Context) error {
 	var env environments.EnvironmentModel
 	resp, err := c.client.
-		OnBeforeRequest(newRestyLogRequestMiddleware(c.log)).
-		OnAfterResponse(newRestyLogResponseMiddleware(c.log)).
 		NewRequest().
 		SetContext(ctx).
 		SetResult(&env).

logger.go

@@ -32,7 +32,7 @@ func (s restySlogLogger) Debugf(format string, v ...interface{}) {
 func newRestyLogRequestMiddleware(logger *slog.Logger) resty.RequestMiddleware {
 	return func(c *resty.Client, req *resty.Request) error {
 		// Create a child logger with request metadata
-		reqLogger := logger.With(
+		reqLogger := logger.WithGroup("http").With(
 			"method", req.Method,
 			"url", req.URL,
 		)
@@ -57,11 +57,16 @@ func newRestyLogResponseMiddleware(logger *slog.Logger) resty.ResponseMiddleware
 		if reqLogger == nil {
 			reqLogger = logger
 		}
-		reqLogger.Debug("response",
+		reqLogger = reqLogger.With(
 			slog.Int("status", resp.StatusCode()),
 			slog.Duration("duration", time.Since(startTime)),
 			slog.Int64("content_length", resp.Size()),
 		)
+		if resp.IsError() {
+			reqLogger.Error("error response")
+		} else {
+			reqLogger.Debug("response")
+		}
 		return nil
 	}
 }