fix imroc#10

khorevaa · Jun 4, 2017 · 02374e7 · 02374e7
1 parent 8168d3b
commit 02374e7
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/resp.go b/resp.go
@@ -64,9 +64,16 @@ func (r *Resp) ToFile(name string) error {
 	if err != nil {
 		return err
 	}
-	_, err = io.Copy(file, r.resp.Body)
-	if err != nil {
-		return err
+	if r.respBody != nil {
+		_, err = file.Write(r.respBody)
+		if err != nil {
+			return err
+		}
+	} else {
+		_, err = io.Copy(file, r.resp.Body)
+		if err != nil {
+			return err
+		}
 	}
 	return nil
 }