Skip to content
Open
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
43 changes: 35 additions & 8 deletions cns/logger/cnslogger.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package logger

import (
"encoding/json"
"fmt"
"maps"
"os"
Expand Down Expand Up @@ -131,17 +132,36 @@ func (c *CNSLogger) Errorf(format string, args ...any) {
c.sendTraceInternal(msg, ai.ErrorLevel)
}

// toJSONString converts any object to a JSON string for logging purposes.
// When the object contains json.RawMessage fields, they will be properly formatted
// instead of being shown as byte arrays. Falls back to %+v if JSON marshaling fails.
func toJSONString(obj any) string {
if obj == nil {
return "null"
}

bytes, err := json.Marshal(obj)
if err != nil {
// Fall back to standard formatting if JSON marshaling fails
return fmt.Sprintf("%+v", obj)
}
return string(bytes)
}

func (c *CNSLogger) Request(tag string, request any, err error) {
c.logger.Request(tag, request, err)
if c.th == nil || c.disableTraceLogging {
return
}

requestString := toJSONString(request)

var msg string
lvl := ai.InfoLevel
if err == nil {
msg = fmt.Sprintf("[%s] Received %T %+v.", tag, request, request)
msg = fmt.Sprintf("[%s] Received %T %s.", tag, request, requestString)
} else {
msg = fmt.Sprintf("[%s] Failed to decode %T %+v %s.", tag, request, request, err.Error())
msg = fmt.Sprintf("[%s] Failed to decode %T %s %s.", tag, request, requestString, err.Error())
lvl = ai.ErrorLevel
}
c.sendTraceInternal(msg, lvl)
Expand All @@ -152,16 +172,19 @@ func (c *CNSLogger) Response(tag string, response any, returnCode types.Response
if c.th == nil || c.disableTraceLogging {
return
}

responseString := toJSONString(response)

var msg string
lvl := ai.InfoLevel
switch {
case err == nil && returnCode == 0:
msg = fmt.Sprintf("[%s] Sent %T %+v.", tag, response, response)
msg = fmt.Sprintf("[%s] Sent %T %s.", tag, response, responseString)
case err != nil:
msg = fmt.Sprintf("[%s] Code:%s, %+v %s.", tag, returnCode.String(), response, err.Error())
msg = fmt.Sprintf("[%s] Code:%s, %s %s.", tag, returnCode.String(), responseString, err.Error())
lvl = ai.ErrorLevel
default:
msg = fmt.Sprintf("[%s] Code:%s, %+v.", tag, returnCode.String(), response)
msg = fmt.Sprintf("[%s] Code:%s, %s.", tag, returnCode.String(), responseString)
}
c.sendTraceInternal(msg, lvl)
}
Expand All @@ -171,16 +194,20 @@ func (c *CNSLogger) ResponseEx(tag string, request, response any, returnCode typ
if c.th == nil || c.disableTraceLogging {
return
}

requestString := toJSONString(request)
responseString := toJSONString(response)

var msg string
lvl := ai.InfoLevel
switch {
case err == nil && returnCode == 0:
msg = fmt.Sprintf("[%s] Sent %T %+v %T %+v.", tag, request, request, response, response)
msg = fmt.Sprintf("[%s] Sent %T %s %T %s.", tag, request, requestString, response, responseString)
case err != nil:
msg = fmt.Sprintf("[%s] Code:%s, %+v, %+v, %s.", tag, returnCode.String(), request, response, err.Error())
msg = fmt.Sprintf("[%s] Code:%s, %s, %s, %s.", tag, returnCode.String(), requestString, responseString, err.Error())
lvl = ai.ErrorLevel
default:
msg = fmt.Sprintf("[%s] Code:%s, %+v, %+v.", tag, returnCode.String(), request, response)
msg = fmt.Sprintf("[%s] Code:%s, %s, %s.", tag, returnCode.String(), requestString, responseString)
}
c.sendTraceInternal(msg, lvl)
}
Expand Down
33 changes: 25 additions & 8 deletions log/logger.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
package log

import (
"encoding/json"
"fmt"
"io"
"log"
Expand Down Expand Up @@ -199,34 +200,50 @@ func (logger *Logger) rotate() error {
return nil
}

// toJSONString converts any object to a JSON string for logging purposes.
// When the object contains json.RawMessage fields, they will be properly formatted
// instead of being shown as byte arrays. Falls back to %+v if JSON marshaling fails.
func toJSONString(obj interface{}) string {
if obj == nil {
return "null"
}

bytes, err := json.Marshal(obj)
if err != nil {
// Fall back to standard formatting if JSON marshaling fails
return fmt.Sprintf("%+v", obj)
}
return string(bytes)
}

// Request logs a structured request.
func (logger *Logger) Request(tag string, request interface{}, err error) {
if err == nil {
logger.Printf("[%s] Received %T %+v.", tag, request, request)
logger.Printf("[%s] Received %T %s.", tag, request, toJSONString(request))
} else {
logger.Errorf("[%s] Failed to decode %T %+v %s.", tag, request, request, err.Error())
logger.Errorf("[%s] Failed to decode %T %s %s.", tag, request, toJSONString(request), err.Error())
}
}

// Response logs a structured response.
func (logger *Logger) Response(tag string, response interface{}, returnCode int, returnStr string, err error) {
if err == nil && returnCode == 0 {
logger.Printf("[%s] Sent %T %+v.", tag, response, response)
logger.Printf("[%s] Sent %T %s.", tag, response, toJSONString(response))
} else if err != nil {
logger.Errorf("[%s] Code:%s, %+v %s.", tag, returnStr, response, err.Error())
logger.Errorf("[%s] Code:%s, %s %s.", tag, returnStr, toJSONString(response), err.Error())
} else {
logger.Errorf("[%s] Code:%s, %+v.", tag, returnStr, response)
logger.Errorf("[%s] Code:%s, %s.", tag, returnStr, toJSONString(response))
}
}

// ResponseEx logs a structured response and the request associate with it.
func (logger *Logger) ResponseEx(tag string, request interface{}, response interface{}, returnCode int, returnStr string, err error) {
if err == nil && returnCode == 0 {
logger.Printf("[%s] Sent %T %+v %T %+v.", tag, request, request, response, response)
logger.Printf("[%s] Sent %T %s %T %s.", tag, request, toJSONString(request), response, toJSONString(response))
} else if err != nil {
logger.Errorf("[%s] Code:%s, %+v, %+v %s.", tag, returnStr, request, response, err.Error())
logger.Errorf("[%s] Code:%s, %s, %s %s.", tag, returnStr, toJSONString(request), toJSONString(response), err.Error())
} else {
logger.Errorf("[%s] Code:%s, %+v, %+v.", tag, returnStr, request, response)
logger.Errorf("[%s] Code:%s, %s, %s.", tag, returnStr, toJSONString(request), toJSONString(response))
}
}

Expand Down
Loading