Add deferred call to close DocumentResult's cursor

GODRIVER-373 Change-Id: I7addfb359952a5abfccd40c59bf294d3e30be16a
zsimon2016 · Sep 13, 2018 · e0b7d07 · e0b7d07
1 parent fdda312
commit e0b7d07
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.errcheck-excludes b/.errcheck-excludes
@@ -4,6 +4,7 @@
 (*github.com/mongodb/mongo-go-driver/core/topology.Server).Close
 (*github.com/mongodb/mongo-go-driver/core/connection.pool).closeConnection
 (github.com/mongodb/mongo-go-driver/core/wiremessage.ReadWriteCloser).Close
+(github.com/mongodb/mongo-go-driver/mongo.Cursor).Close
 (net.Conn).Close
 encoding/pem.Encode
 fmt.Fprintf

diff --git a/mongo/document_result.go b/mongo/document_result.go
@@ -40,6 +40,7 @@ func (dr *DocumentResult) Decode(v interface{}) error {
 		}
 		return bson.Unmarshal(dr.rdr, v)
 	case dr.cur != nil:
+		defer dr.cur.Close(context.TODO())
 		if !dr.cur.Next(context.TODO()) {
 			if err := dr.cur.Err(); err != nil {
 				return err