Truncate request response body if journal memory limit is set

Author: tommysitu

core/journal/journal.go

@@ -107,9 +107,14 @@ func (this *Journal) NewEntry(request *http.Request, response *http.Response, mo
 
 	respBody, _ := util.GetResponseBody(response)
 
+	if this.BodyMemoryLimit.ToBytes() > 0 {
+		payloadRequest.Body = util.TruncateStringWithEllipsis(payloadRequest.Body, this.BodyMemoryLimit.ToBytes())
+		respBody = util.TruncateStringWithEllipsis(respBody, this.BodyMemoryLimit.ToBytes())
+	}
+
 	payloadResponse := &models.ResponseDetails{
 		Status:  response.StatusCode,
-		Body:    respBody,
+		Body:    util.TruncateStringWithEllipsis(respBody, this.BodyMemoryLimit.ToBytes()),
 		Headers: response.Header,
 	}
 

core/util/util.go

@@ -558,8 +558,12 @@ func ResolveAndValidatePath(absBasePath, relativePath string) (string, error) {
 	return resolvedPath, nil
 }
 
-func truncateStringWithEllipsis(input string, maxSize int) string {
-	ellipsis := "..."
+func TruncateStringWithEllipsis(input string, maxSize int) string {
+	ellipsis := "..." // 3 bytes in UTF-8
+
+	if maxSize <= 3 {
+		return ellipsis
+	}
 
 	if len(input) <= maxSize{
 		return input

core/util/util_test.go

@@ -432,6 +432,12 @@ func TestTruncateStringWithEllipsis(t *testing.T) {
 			maxSize:  10,
 			expected: "Hello",
 		},
+		{
+			name:     "Zero max size",
+			input:    "Hello",
+			maxSize:  0,
+			expected: "...",
+		},
 		{
 			name:     "Truncate with ellipsis",
 			input:    "Hello, World!",
@@ -473,7 +479,7 @@ func TestTruncateStringWithEllipsis(t *testing.T) {
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
 			g := NewWithT(t)
-			result := truncateStringWithEllipsis(test.input, test.maxSize)
+			result := TruncateStringWithEllipsis(test.input, test.maxSize)
 			g.Expect(result).To(Equal(test.expected), "Expected %q but got %q for input %q with maxSize %d", test.expected, result, test.input, test.maxSize)
 		})
 	}