Skip to content

Commit 5cb4398

Browse files
author
Alice Thum
committed
Close connections in executeOperation.
GODRIVER-1247 Change-Id: Icf35cbb89050bec1b08833d57942cd08d4e9460e
1 parent 4094924 commit 5cb4398

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

mongo/change_stream.go

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -161,12 +161,17 @@ func newChangeStream(ctx context.Context, config changeStreamConfig, pipeline in
161161
func (cs *ChangeStream) executeOperation(ctx context.Context, resuming bool) error {
162162
var server driver.Server
163163
var conn driver.Connection
164+
var err error
165+
164166
if server, cs.err = cs.client.topology.SelectServer(ctx, cs.selector); cs.err != nil {
165167
return cs.Err()
166168
}
167169
if conn, cs.err = server.Connection(ctx); cs.err != nil {
168170
return cs.Err()
169171
}
172+
173+
defer conn.Close()
174+
170175
cs.aggregate.Deployment(driver.SingleConnectionDeployment{
171176
C: conn,
172177
})
@@ -204,12 +209,15 @@ func (cs *ChangeStream) executeOperation(ctx context.Context, resuming bool) err
204209
break
205210
}
206211

207-
server, err := cs.client.topology.SelectServer(ctx, cs.selector)
212+
server, err = cs.client.topology.SelectServer(ctx, cs.selector)
208213
if err != nil {
209214
break
210215
}
211216

212-
conn, err := server.Connection(ctx)
217+
conn.Close()
218+
conn, err = server.Connection(ctx)
219+
defer conn.Close()
220+
213221
if err != nil {
214222
break
215223
}

0 commit comments

Comments
 (0)