fix: fix issue timakin#11

SergeShpak · Apr 19, 2023 · f7423e8 · f7423e8
1 parent e39cf3f
commit f7423e8
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/passes/bodyclose/bodyclose.go b/passes/bodyclose/bodyclose.go
@@ -218,6 +218,10 @@ func (r *runner) getResVal(instr ssa.Instruction) (ssa.Value, bool) {
 		if instr.Type().String() == r.resTyp.String() {
 			return instr, true
 		}
+	case *ssa.Store:
+		if instr.Val.Type().String() == r.resTyp.String() {
+			return instr.Val, true
+		}
 	}
 	return nil, false
 }

diff --git a/passes/bodyclose/testdata/src/a/issue47.go b/passes/bodyclose/testdata/src/a/issue47.go
@@ -0,0 +1,15 @@
+package a
+
+import (
+	"io"
+	"net/http/httptest"
+)
+
+func issues11() {
+	w := httptest.NewRecorder()
+	resp := w.Result()
+	defer func() {
+		_ = resp.Body.Close()
+	}()
+	_, _ = io.ReadAll(resp.Body)
+}