Update executeModify to close iterators

Signed-off-by: Zach Musgrave <zach@liquidata.co>
This commit is contained in:
Zach Musgrave
2020-09-10 20:03:07 -07:00
parent 3e17e271d2
commit ad43897ac1
+15 -1
View File
@@ -75,8 +75,22 @@ func executeModify(ctx context.Context, dEnv *env.DoltEnv, root *doltdb.RootValu
return nil, err
}
_, _, err = engine.Query(sqlCtx, query)
_, iter, err := engine.Query(sqlCtx, query)
if err != nil {
return nil, err
}
for {
_, err := iter.Next()
if err == io.EOF {
break
}
if err != nil {
return nil, err
}
}
err = iter.Close()
if err != nil {
return nil, err
}