All the Collection level methods have context.Context as their first parameter which allows setting a deadline for the operation to occur within. In the private/ops package the runMustUsePrimary and the runMayUseSecondary functions spawn a new goroutine and give it a context.Background which is used when setting the deadline on the TCP connection. The context.Context provided by the user should be used to set the deadline on the TCP connection, update the implementation to do this.