소스 검색

Log response even if raw streaming failed

DarthSim 3 년 전
부모
커밋
12f17a56bc
1개의 변경된 파일5개의 추가작업 그리고 3개의 파일을 삭제
  1. 5 3
      stream.go

+ 5 - 3
stream.go

@@ -129,9 +129,7 @@ func streamOriginImage(ctx context.Context, reqID string, r *http.Request, rw ht
 	buf := streamBufPool.Get().(*[]byte)
 	defer streamBufPool.Put(buf)
 
-	if _, err := io.CopyBuffer(rw, res.Body, *buf); err != nil {
-		panic(http.ErrAbortHandler)
-	}
+	_, copyerr := io.CopyBuffer(rw, res.Body, *buf)
 
 	router.LogResponse(
 		reqID, r, res.StatusCode, nil,
@@ -140,4 +138,8 @@ func streamOriginImage(ctx context.Context, reqID string, r *http.Request, rw ht
 			"processing_options": po,
 		},
 	)
+
+	if copyerr != nil {
+		panic(http.ErrAbortHandler)
+	}
 }