foomo/gotsrpc
1
0
Fork 0
mirror of https://github.com/foomo/gotsrpc.git synced 2025-10-16 12:35:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: add better handling of response stats

Browse Source
This commit is contained in:
Stefan Martinov 2025-01-20 15:17:20 +01:00
parent 86d0dd5399
commit e8bb451a50
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
response.go
View File

@ -68,26 +68,31 @@ func Reply(response []interface{}, stats *CallStats, r *http.Request, w http.Res
}
func recordStats(stats *CallStats, response []interface{}, responseWriter *responseWriterWithLength) func() {
if stats == nil {
return func() {}
}
start := time.Now()
return func() {
if stats != nil {
stats.ResponseSize = responseWriter.length
stats.Marshalling = time.Since(start)
if len(response) > 0 {
errResp := response[len(response)-1]
if v, ok := errResp.(error); ok && v != nil {
if !reflect.ValueOf(v).IsNil() {
stats.ErrorCode = 1
stats.ErrorType = fmt.Sprintf("%T", v)
stats.ErrorMessage = v.Error()
if v, ok := v.(interface {
ErrorCode() int
}); ok {
stats.ErrorCode = v.ErrorCode()
}
}
stats.ResponseSize = responseWriter.length
stats.Marshalling = time.Since(start)
if len(response) == 0 {
return
}
errResp := response[len(response)-1]
if v, ok := errResp.(error); ok && v != nil {
if !reflect.ValueOf(v).IsNil() {
stats.ErrorCode = 1
stats.ErrorType = fmt.Sprintf("%T", v)
stats.ErrorMessage = v.Error()
if v, ok := v.(interface {
ErrorCode() int
}); ok {
stats.ErrorCode = v.ErrorCode()
}
}
}
}
}