refactor: simplify writerWrapper.enoughContentLength()

nanmu42 · Nov 29, 2020 · 96ed5c7 · 96ed5c7
1 parent b0483ce
commit 96ed5c7
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/writerwrapper.go b/writerwrapper.go
@@ -178,15 +178,10 @@ func (w *writerWrapper) writeBuffer(data []byte) (fit bool) {
 }
 
 func (w *writerWrapper) enoughContentLength() bool {
-	var (
-		header        = w.Header()
-		_, haveCl     = header["Content-Length"]
-		contentLength int64
-	)
-	if haveCl {
-		contentLength, _ = strconv.ParseInt(header.Get("Content-Length"), 10, 64)
+	contentLength, err := strconv.ParseInt(w.Header().Get("Content-Length"), 10, 64)
+	if err != nil {
+		return false
 	}
-
 	if contentLength != 0 && contentLength >= w.MinContentLength {
 		return true
 	}