feat: add hook dumping HTTP response on error (#19)

Kong · Oct 10, 2024 · 202ef32 · 202ef32
1 parent 077efc6
commit 202ef32
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/internal/hooks/httpdump.go b/internal/hooks/httpdump.go
@@ -38,6 +38,18 @@ var _ afterSuccessHook = (*HTTPDumpResponseHook)(nil)
 
 // AfterSuccess dumps the response to stdout if enabled.
 func (i *HTTPDumpResponseHook) AfterSuccess(hookCtx AfterSuccessContext, res *http.Response) (*http.Response, error) {
+	return i.dumpResponse(res)
+}
+
+var _ afterErrorHook = (*HTTPDumpResponseHook)(nil)
+
+// AfterSuccess dumps the response to stdout if enabled.
+func (i *HTTPDumpResponseHook) AfterError(hookCtx AfterErrorContext, res *http.Response, err error) (*http.Response, error) {
+	fmt.Printf("Error: %v\n", err)
+	return i.dumpResponse(res)
+}
+
+func (i *HTTPDumpResponseHook) dumpResponse(res *http.Response) (*http.Response, error) {
 	if !i.Enabled {
 		return res, nil
 	}

diff --git a/internal/hooks/init.go b/internal/hooks/init.go
@@ -20,4 +20,7 @@ func initHooks(h *Hooks) {
 	h.registerAfterSuccessHook(&HTTPDumpResponseHook{
 		Enabled: os.Getenv("KONNECT_SDK_HTTP_DUMP_RESPONSE") == "true",
 	})
+	h.registerAfterErrorHook(&HTTPDumpResponseHook{
+		Enabled: os.Getenv("KONNECT_SDK_HTTP_DUMP_RESPONSE") == "true",
+	})
 }