From 3008ab5843f567f624bd4c8db84876060c4860ac Mon Sep 17 00:00:00 2001
From: simon987 <me@simon987.net>
Date: Sat, 28 Dec 2019 11:47:13 -0500
Subject: [PATCH] return response even on error (WIP)

---
 main.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 2f309bb..a227fb8 100644
--- a/main.go
+++ b/main.go
@@ -35,7 +35,11 @@ func New() *Architeuthis {
 
 			if err != nil {
 				logrus.WithError(err).Trace("Could not complete request")
-				return nil, goproxy.NewResponse(r, "text/plain", http.StatusInternalServerError, err.Error())
+				if resp != nil {
+					return nil, resp
+				} else {
+					return nil, goproxy.NewResponse(r, "text/plain", http.StatusInternalServerError, err.Error())
+				}
 			}
 
 			return nil, resp
@@ -115,7 +119,7 @@ func (a *Architeuthis) processRequest(r *http.Request) (*http.Response, error) {
 		}
 
 		if !responseCtx.ShouldRetry {
-			return nil, responseCtx.Error
+			return responseCtx.Response, responseCtx.Error
 		}
 	}
 }