forked from golang/hotime
701 lines
22 KiB
Go
701 lines
22 KiB
Go
// Copyright (C) MongoDB, Inc. 2017-present.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License. You may obtain
|
|
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
package mongo
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/bsoncodec"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/internal"
|
|
"go.mongodb.org/mongo-driver/mongo/description"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"go.mongodb.org/mongo-driver/mongo/readconcern"
|
|
"go.mongodb.org/mongo-driver/mongo/readpref"
|
|
"go.mongodb.org/mongo-driver/x/bsonx/bsoncore"
|
|
"go.mongodb.org/mongo-driver/x/mongo/driver"
|
|
"go.mongodb.org/mongo-driver/x/mongo/driver/operation"
|
|
"go.mongodb.org/mongo-driver/x/mongo/driver/session"
|
|
)
|
|
|
|
var (
|
|
// ErrMissingResumeToken indicates that a change stream notification from the server did not contain a resume token.
|
|
ErrMissingResumeToken = errors.New("cannot provide resume functionality when the resume token is missing")
|
|
// ErrNilCursor indicates that the underlying cursor for the change stream is nil.
|
|
ErrNilCursor = errors.New("cursor is nil")
|
|
|
|
minResumableLabelWireVersion int32 = 9 // Wire version at which the server includes the resumable error label
|
|
networkErrorLabel = "NetworkError"
|
|
resumableErrorLabel = "ResumableChangeStreamError"
|
|
errorCursorNotFound int32 = 43 // CursorNotFound error code
|
|
|
|
// Allowlist of error codes that are considered resumable.
|
|
resumableChangeStreamErrors = map[int32]struct{}{
|
|
6: {}, // HostUnreachable
|
|
7: {}, // HostNotFound
|
|
89: {}, // NetworkTimeout
|
|
91: {}, // ShutdownInProgress
|
|
189: {}, // PrimarySteppedDown
|
|
262: {}, // ExceededTimeLimit
|
|
9001: {}, // SocketException
|
|
10107: {}, // NotPrimary
|
|
11600: {}, // InterruptedAtShutdown
|
|
11602: {}, // InterruptedDueToReplStateChange
|
|
13435: {}, // NotPrimaryNoSecondaryOK
|
|
13436: {}, // NotPrimaryOrSecondary
|
|
63: {}, // StaleShardVersion
|
|
150: {}, // StaleEpoch
|
|
13388: {}, // StaleConfig
|
|
234: {}, // RetryChangeStream
|
|
133: {}, // FailedToSatisfyReadPreference
|
|
}
|
|
)
|
|
|
|
// ChangeStream is used to iterate over a stream of events. Each event can be decoded into a Go type via the Decode
|
|
// method or accessed as raw BSON via the Current field. This type is not goroutine safe and must not be used
|
|
// concurrently by multiple goroutines. For more information about change streams, see
|
|
// https://www.mongodb.com/docs/manual/changeStreams/.
|
|
type ChangeStream struct {
|
|
// Current is the BSON bytes of the current event. This property is only valid until the next call to Next or
|
|
// TryNext. If continued access is required, a copy must be made.
|
|
Current bson.Raw
|
|
|
|
aggregate *operation.Aggregate
|
|
pipelineSlice []bsoncore.Document
|
|
pipelineOptions map[string]bsoncore.Value
|
|
cursor changeStreamCursor
|
|
cursorOptions driver.CursorOptions
|
|
batch []bsoncore.Document
|
|
resumeToken bson.Raw
|
|
err error
|
|
sess *session.Client
|
|
client *Client
|
|
registry *bsoncodec.Registry
|
|
streamType StreamType
|
|
options *options.ChangeStreamOptions
|
|
selector description.ServerSelector
|
|
operationTime *primitive.Timestamp
|
|
wireVersion *description.VersionRange
|
|
}
|
|
|
|
type changeStreamConfig struct {
|
|
readConcern *readconcern.ReadConcern
|
|
readPreference *readpref.ReadPref
|
|
client *Client
|
|
registry *bsoncodec.Registry
|
|
streamType StreamType
|
|
collectionName string
|
|
databaseName string
|
|
crypt driver.Crypt
|
|
}
|
|
|
|
func newChangeStream(ctx context.Context, config changeStreamConfig, pipeline interface{},
|
|
opts ...*options.ChangeStreamOptions) (*ChangeStream, error) {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
|
|
cs := &ChangeStream{
|
|
client: config.client,
|
|
registry: config.registry,
|
|
streamType: config.streamType,
|
|
options: options.MergeChangeStreamOptions(opts...),
|
|
selector: description.CompositeSelector([]description.ServerSelector{
|
|
description.ReadPrefSelector(config.readPreference),
|
|
description.LatencySelector(config.client.localThreshold),
|
|
}),
|
|
cursorOptions: config.client.createBaseCursorOptions(),
|
|
}
|
|
|
|
cs.sess = sessionFromContext(ctx)
|
|
if cs.sess == nil && cs.client.sessionPool != nil {
|
|
cs.sess, cs.err = session.NewClientSession(cs.client.sessionPool, cs.client.id, session.Implicit)
|
|
if cs.err != nil {
|
|
return nil, cs.Err()
|
|
}
|
|
}
|
|
if cs.err = cs.client.validSession(cs.sess); cs.err != nil {
|
|
closeImplicitSession(cs.sess)
|
|
return nil, cs.Err()
|
|
}
|
|
|
|
cs.aggregate = operation.NewAggregate(nil).
|
|
ReadPreference(config.readPreference).ReadConcern(config.readConcern).
|
|
Deployment(cs.client.deployment).ClusterClock(cs.client.clock).
|
|
CommandMonitor(cs.client.monitor).Session(cs.sess).ServerSelector(cs.selector).Retry(driver.RetryNone).
|
|
ServerAPI(cs.client.serverAPI).Crypt(config.crypt).Timeout(cs.client.timeout)
|
|
|
|
if cs.options.Collation != nil {
|
|
cs.aggregate.Collation(bsoncore.Document(cs.options.Collation.ToDocument()))
|
|
}
|
|
if comment := cs.options.Comment; comment != nil {
|
|
cs.aggregate.Comment(*comment)
|
|
|
|
commentVal, err := transformValue(cs.registry, comment, true, "comment")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cs.cursorOptions.Comment = commentVal
|
|
}
|
|
if cs.options.BatchSize != nil {
|
|
cs.aggregate.BatchSize(*cs.options.BatchSize)
|
|
cs.cursorOptions.BatchSize = *cs.options.BatchSize
|
|
}
|
|
if cs.options.MaxAwaitTime != nil {
|
|
cs.cursorOptions.MaxTimeMS = int64(*cs.options.MaxAwaitTime / time.Millisecond)
|
|
}
|
|
if cs.options.Custom != nil {
|
|
// Marshal all custom options before passing to the initial aggregate. Return
|
|
// any errors from Marshaling.
|
|
customOptions := make(map[string]bsoncore.Value)
|
|
for optionName, optionValue := range cs.options.Custom {
|
|
bsonType, bsonData, err := bson.MarshalValueWithRegistry(cs.registry, optionValue)
|
|
if err != nil {
|
|
cs.err = err
|
|
closeImplicitSession(cs.sess)
|
|
return nil, cs.Err()
|
|
}
|
|
optionValueBSON := bsoncore.Value{Type: bsonType, Data: bsonData}
|
|
customOptions[optionName] = optionValueBSON
|
|
}
|
|
cs.aggregate.CustomOptions(customOptions)
|
|
}
|
|
if cs.options.CustomPipeline != nil {
|
|
// Marshal all custom pipeline options before building pipeline slice. Return
|
|
// any errors from Marshaling.
|
|
cs.pipelineOptions = make(map[string]bsoncore.Value)
|
|
for optionName, optionValue := range cs.options.CustomPipeline {
|
|
bsonType, bsonData, err := bson.MarshalValueWithRegistry(cs.registry, optionValue)
|
|
if err != nil {
|
|
cs.err = err
|
|
closeImplicitSession(cs.sess)
|
|
return nil, cs.Err()
|
|
}
|
|
optionValueBSON := bsoncore.Value{Type: bsonType, Data: bsonData}
|
|
cs.pipelineOptions[optionName] = optionValueBSON
|
|
}
|
|
}
|
|
|
|
switch cs.streamType {
|
|
case ClientStream:
|
|
cs.aggregate.Database("admin")
|
|
case DatabaseStream:
|
|
cs.aggregate.Database(config.databaseName)
|
|
case CollectionStream:
|
|
cs.aggregate.Collection(config.collectionName).Database(config.databaseName)
|
|
default:
|
|
closeImplicitSession(cs.sess)
|
|
return nil, fmt.Errorf("must supply a valid StreamType in config, instead of %v", cs.streamType)
|
|
}
|
|
|
|
// When starting a change stream, cache startAfter as the first resume token if it is set. If not, cache
|
|
// resumeAfter. If neither is set, do not cache a resume token.
|
|
resumeToken := cs.options.StartAfter
|
|
if resumeToken == nil {
|
|
resumeToken = cs.options.ResumeAfter
|
|
}
|
|
var marshaledToken bson.Raw
|
|
if resumeToken != nil {
|
|
if marshaledToken, cs.err = bson.Marshal(resumeToken); cs.err != nil {
|
|
closeImplicitSession(cs.sess)
|
|
return nil, cs.Err()
|
|
}
|
|
}
|
|
cs.resumeToken = marshaledToken
|
|
|
|
if cs.err = cs.buildPipelineSlice(pipeline); cs.err != nil {
|
|
closeImplicitSession(cs.sess)
|
|
return nil, cs.Err()
|
|
}
|
|
var pipelineArr bsoncore.Document
|
|
pipelineArr, cs.err = cs.pipelineToBSON()
|
|
cs.aggregate.Pipeline(pipelineArr)
|
|
|
|
if cs.err = cs.executeOperation(ctx, false); cs.err != nil {
|
|
closeImplicitSession(cs.sess)
|
|
return nil, cs.Err()
|
|
}
|
|
|
|
return cs, cs.Err()
|
|
}
|
|
|
|
func (cs *ChangeStream) createOperationDeployment(server driver.Server, connection driver.Connection) driver.Deployment {
|
|
return &changeStreamDeployment{
|
|
topologyKind: cs.client.deployment.Kind(),
|
|
server: server,
|
|
conn: connection,
|
|
}
|
|
}
|
|
|
|
func (cs *ChangeStream) executeOperation(ctx context.Context, resuming bool) error {
|
|
var server driver.Server
|
|
var conn driver.Connection
|
|
var err error
|
|
|
|
if server, cs.err = cs.client.deployment.SelectServer(ctx, cs.selector); cs.err != nil {
|
|
return cs.Err()
|
|
}
|
|
if conn, cs.err = server.Connection(ctx); cs.err != nil {
|
|
return cs.Err()
|
|
}
|
|
defer conn.Close()
|
|
cs.wireVersion = conn.Description().WireVersion
|
|
|
|
cs.aggregate.Deployment(cs.createOperationDeployment(server, conn))
|
|
|
|
if resuming {
|
|
cs.replaceOptions(cs.wireVersion)
|
|
|
|
csOptDoc, err := cs.createPipelineOptionsDoc()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pipIdx, pipDoc := bsoncore.AppendDocumentStart(nil)
|
|
pipDoc = bsoncore.AppendDocumentElement(pipDoc, "$changeStream", csOptDoc)
|
|
if pipDoc, cs.err = bsoncore.AppendDocumentEnd(pipDoc, pipIdx); cs.err != nil {
|
|
return cs.Err()
|
|
}
|
|
cs.pipelineSlice[0] = pipDoc
|
|
|
|
var plArr bsoncore.Document
|
|
if plArr, cs.err = cs.pipelineToBSON(); cs.err != nil {
|
|
return cs.Err()
|
|
}
|
|
cs.aggregate.Pipeline(plArr)
|
|
}
|
|
|
|
// If no deadline is set on the passed-in context, cs.client.timeout is set, and context is not already
|
|
// a Timeout context, honor cs.client.timeout in new Timeout context for change stream operation execution
|
|
// and potential retry.
|
|
if _, deadlineSet := ctx.Deadline(); !deadlineSet && cs.client.timeout != nil && !internal.IsTimeoutContext(ctx) {
|
|
newCtx, cancelFunc := internal.MakeTimeoutContext(ctx, *cs.client.timeout)
|
|
// Redefine ctx to be the new timeout-derived context.
|
|
ctx = newCtx
|
|
// Cancel the timeout-derived context at the end of executeOperation to avoid a context leak.
|
|
defer cancelFunc()
|
|
}
|
|
if original := cs.aggregate.Execute(ctx); original != nil {
|
|
retryableRead := cs.client.retryReads && cs.wireVersion != nil && cs.wireVersion.Max >= 6
|
|
if !retryableRead {
|
|
cs.err = replaceErrors(original)
|
|
return cs.err
|
|
}
|
|
|
|
cs.err = original
|
|
switch tt := original.(type) {
|
|
case driver.Error:
|
|
if !tt.RetryableRead() {
|
|
break
|
|
}
|
|
|
|
server, err = cs.client.deployment.SelectServer(ctx, cs.selector)
|
|
if err != nil {
|
|
break
|
|
}
|
|
|
|
conn.Close()
|
|
conn, err = server.Connection(ctx)
|
|
if err != nil {
|
|
break
|
|
}
|
|
defer conn.Close()
|
|
cs.wireVersion = conn.Description().WireVersion
|
|
|
|
if cs.wireVersion == nil || cs.wireVersion.Max < 6 {
|
|
break
|
|
}
|
|
|
|
cs.aggregate.Deployment(cs.createOperationDeployment(server, conn))
|
|
cs.err = cs.aggregate.Execute(ctx)
|
|
}
|
|
|
|
if cs.err != nil {
|
|
cs.err = replaceErrors(cs.err)
|
|
return cs.Err()
|
|
}
|
|
|
|
}
|
|
cs.err = nil
|
|
|
|
cr := cs.aggregate.ResultCursorResponse()
|
|
cr.Server = server
|
|
|
|
cs.cursor, cs.err = driver.NewBatchCursor(cr, cs.sess, cs.client.clock, cs.cursorOptions)
|
|
if cs.err = replaceErrors(cs.err); cs.err != nil {
|
|
return cs.Err()
|
|
}
|
|
|
|
cs.updatePbrtFromCommand()
|
|
if cs.options.StartAtOperationTime == nil && cs.options.ResumeAfter == nil &&
|
|
cs.options.StartAfter == nil && cs.wireVersion.Max >= 7 &&
|
|
cs.emptyBatch() && cs.resumeToken == nil {
|
|
cs.operationTime = cs.sess.OperationTime
|
|
}
|
|
|
|
return cs.Err()
|
|
}
|
|
|
|
// Updates the post batch resume token after a successful aggregate or getMore operation.
|
|
func (cs *ChangeStream) updatePbrtFromCommand() {
|
|
// Only cache the pbrt if an empty batch was returned and a pbrt was included
|
|
if pbrt := cs.cursor.PostBatchResumeToken(); cs.emptyBatch() && pbrt != nil {
|
|
cs.resumeToken = bson.Raw(pbrt)
|
|
}
|
|
}
|
|
|
|
func (cs *ChangeStream) storeResumeToken() error {
|
|
// If cs.Current is the last document in the batch and a pbrt is included, cache the pbrt
|
|
// Otherwise, cache the _id of the document
|
|
var tokenDoc bson.Raw
|
|
if len(cs.batch) == 0 {
|
|
if pbrt := cs.cursor.PostBatchResumeToken(); pbrt != nil {
|
|
tokenDoc = bson.Raw(pbrt)
|
|
}
|
|
}
|
|
|
|
if tokenDoc == nil {
|
|
var ok bool
|
|
tokenDoc, ok = cs.Current.Lookup("_id").DocumentOK()
|
|
if !ok {
|
|
_ = cs.Close(context.Background())
|
|
return ErrMissingResumeToken
|
|
}
|
|
}
|
|
|
|
cs.resumeToken = tokenDoc
|
|
return nil
|
|
}
|
|
|
|
func (cs *ChangeStream) buildPipelineSlice(pipeline interface{}) error {
|
|
val := reflect.ValueOf(pipeline)
|
|
if !val.IsValid() || !(val.Kind() == reflect.Slice) {
|
|
cs.err = errors.New("can only transform slices and arrays into aggregation pipelines, but got invalid")
|
|
return cs.err
|
|
}
|
|
|
|
cs.pipelineSlice = make([]bsoncore.Document, 0, val.Len()+1)
|
|
|
|
csIdx, csDoc := bsoncore.AppendDocumentStart(nil)
|
|
|
|
csDocTemp, err := cs.createPipelineOptionsDoc()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
csDoc = bsoncore.AppendDocumentElement(csDoc, "$changeStream", csDocTemp)
|
|
csDoc, cs.err = bsoncore.AppendDocumentEnd(csDoc, csIdx)
|
|
if cs.err != nil {
|
|
return cs.err
|
|
}
|
|
cs.pipelineSlice = append(cs.pipelineSlice, csDoc)
|
|
|
|
for i := 0; i < val.Len(); i++ {
|
|
var elem []byte
|
|
elem, cs.err = transformBsoncoreDocument(cs.registry, val.Index(i).Interface(), true, fmt.Sprintf("pipeline stage :%v", i))
|
|
if cs.err != nil {
|
|
return cs.err
|
|
}
|
|
|
|
cs.pipelineSlice = append(cs.pipelineSlice, elem)
|
|
}
|
|
|
|
return cs.err
|
|
}
|
|
|
|
func (cs *ChangeStream) createPipelineOptionsDoc() (bsoncore.Document, error) {
|
|
plDocIdx, plDoc := bsoncore.AppendDocumentStart(nil)
|
|
|
|
if cs.streamType == ClientStream {
|
|
plDoc = bsoncore.AppendBooleanElement(plDoc, "allChangesForCluster", true)
|
|
}
|
|
|
|
if cs.options.FullDocument != nil {
|
|
// Only append a default "fullDocument" field if wire version is less than 6 (3.6). Otherwise,
|
|
// the server will assume users want the default behavior, and "fullDocument" does not need to be
|
|
// specified.
|
|
if *cs.options.FullDocument != options.Default || (cs.wireVersion != nil && cs.wireVersion.Max < 6) {
|
|
plDoc = bsoncore.AppendStringElement(plDoc, "fullDocument", string(*cs.options.FullDocument))
|
|
}
|
|
}
|
|
|
|
if cs.options.FullDocumentBeforeChange != nil {
|
|
plDoc = bsoncore.AppendStringElement(plDoc, "fullDocumentBeforeChange", string(*cs.options.FullDocumentBeforeChange))
|
|
}
|
|
|
|
if cs.options.ResumeAfter != nil {
|
|
var raDoc bsoncore.Document
|
|
raDoc, cs.err = transformBsoncoreDocument(cs.registry, cs.options.ResumeAfter, true, "resumeAfter")
|
|
if cs.err != nil {
|
|
return nil, cs.err
|
|
}
|
|
|
|
plDoc = bsoncore.AppendDocumentElement(plDoc, "resumeAfter", raDoc)
|
|
}
|
|
|
|
if cs.options.ShowExpandedEvents != nil {
|
|
plDoc = bsoncore.AppendBooleanElement(plDoc, "showExpandedEvents", *cs.options.ShowExpandedEvents)
|
|
}
|
|
|
|
if cs.options.StartAfter != nil {
|
|
var saDoc bsoncore.Document
|
|
saDoc, cs.err = transformBsoncoreDocument(cs.registry, cs.options.StartAfter, true, "startAfter")
|
|
if cs.err != nil {
|
|
return nil, cs.err
|
|
}
|
|
|
|
plDoc = bsoncore.AppendDocumentElement(plDoc, "startAfter", saDoc)
|
|
}
|
|
|
|
if cs.options.StartAtOperationTime != nil {
|
|
plDoc = bsoncore.AppendTimestampElement(plDoc, "startAtOperationTime", cs.options.StartAtOperationTime.T, cs.options.StartAtOperationTime.I)
|
|
}
|
|
|
|
// Append custom pipeline options.
|
|
for optionName, optionValue := range cs.pipelineOptions {
|
|
plDoc = bsoncore.AppendValueElement(plDoc, optionName, optionValue)
|
|
}
|
|
|
|
if plDoc, cs.err = bsoncore.AppendDocumentEnd(plDoc, plDocIdx); cs.err != nil {
|
|
return nil, cs.err
|
|
}
|
|
|
|
return plDoc, nil
|
|
}
|
|
|
|
func (cs *ChangeStream) pipelineToBSON() (bsoncore.Document, error) {
|
|
pipelineDocIdx, pipelineArr := bsoncore.AppendArrayStart(nil)
|
|
for i, doc := range cs.pipelineSlice {
|
|
pipelineArr = bsoncore.AppendDocumentElement(pipelineArr, strconv.Itoa(i), doc)
|
|
}
|
|
if pipelineArr, cs.err = bsoncore.AppendArrayEnd(pipelineArr, pipelineDocIdx); cs.err != nil {
|
|
return nil, cs.err
|
|
}
|
|
return pipelineArr, cs.err
|
|
}
|
|
|
|
func (cs *ChangeStream) replaceOptions(wireVersion *description.VersionRange) {
|
|
// Cached resume token: use the resume token as the resumeAfter option and set no other resume options
|
|
if cs.resumeToken != nil {
|
|
cs.options.SetResumeAfter(cs.resumeToken)
|
|
cs.options.SetStartAfter(nil)
|
|
cs.options.SetStartAtOperationTime(nil)
|
|
return
|
|
}
|
|
|
|
// No cached resume token but cached operation time: use the operation time as the startAtOperationTime option and
|
|
// set no other resume options
|
|
if (cs.sess.OperationTime != nil || cs.options.StartAtOperationTime != nil) && wireVersion.Max >= 7 {
|
|
opTime := cs.options.StartAtOperationTime
|
|
if cs.operationTime != nil {
|
|
opTime = cs.sess.OperationTime
|
|
}
|
|
|
|
cs.options.SetStartAtOperationTime(opTime)
|
|
cs.options.SetResumeAfter(nil)
|
|
cs.options.SetStartAfter(nil)
|
|
return
|
|
}
|
|
|
|
// No cached resume token or operation time: set none of the resume options
|
|
cs.options.SetResumeAfter(nil)
|
|
cs.options.SetStartAfter(nil)
|
|
cs.options.SetStartAtOperationTime(nil)
|
|
}
|
|
|
|
// ID returns the ID for this change stream, or 0 if the cursor has been closed or exhausted.
|
|
func (cs *ChangeStream) ID() int64 {
|
|
if cs.cursor == nil {
|
|
return 0
|
|
}
|
|
return cs.cursor.ID()
|
|
}
|
|
|
|
// Decode will unmarshal the current event document into val and return any errors from the unmarshalling process
|
|
// without any modification. If val is nil or is a typed nil, an error will be returned.
|
|
func (cs *ChangeStream) Decode(val interface{}) error {
|
|
if cs.cursor == nil {
|
|
return ErrNilCursor
|
|
}
|
|
|
|
return bson.UnmarshalWithRegistry(cs.registry, cs.Current, val)
|
|
}
|
|
|
|
// Err returns the last error seen by the change stream, or nil if no errors has occurred.
|
|
func (cs *ChangeStream) Err() error {
|
|
if cs.err != nil {
|
|
return replaceErrors(cs.err)
|
|
}
|
|
if cs.cursor == nil {
|
|
return nil
|
|
}
|
|
|
|
return replaceErrors(cs.cursor.Err())
|
|
}
|
|
|
|
// Close closes this change stream and the underlying cursor. Next and TryNext must not be called after Close has been
|
|
// called. Close is idempotent. After the first call, any subsequent calls will not change the state.
|
|
func (cs *ChangeStream) Close(ctx context.Context) error {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
|
|
defer closeImplicitSession(cs.sess)
|
|
|
|
if cs.cursor == nil {
|
|
return nil // cursor is already closed
|
|
}
|
|
|
|
cs.err = replaceErrors(cs.cursor.Close(ctx))
|
|
cs.cursor = nil
|
|
return cs.Err()
|
|
}
|
|
|
|
// ResumeToken returns the last cached resume token for this change stream, or nil if a resume token has not been
|
|
// stored.
|
|
func (cs *ChangeStream) ResumeToken() bson.Raw {
|
|
return cs.resumeToken
|
|
}
|
|
|
|
// Next gets the next event for this change stream. It returns true if there were no errors and the next event document
|
|
// is available.
|
|
//
|
|
// Next blocks until an event is available, an error occurs, or ctx expires. If ctx expires, the error
|
|
// will be set to ctx.Err(). In an error case, Next will return false.
|
|
//
|
|
// If Next returns false, subsequent calls will also return false.
|
|
func (cs *ChangeStream) Next(ctx context.Context) bool {
|
|
return cs.next(ctx, false)
|
|
}
|
|
|
|
// TryNext attempts to get the next event for this change stream. It returns true if there were no errors and the next
|
|
// event document is available.
|
|
//
|
|
// TryNext returns false if the change stream is closed by the server, an error occurs when getting changes from the
|
|
// server, the next change is not yet available, or ctx expires. If ctx expires, the error will be set to ctx.Err().
|
|
//
|
|
// If TryNext returns false and an error occurred or the change stream was closed
|
|
// (i.e. cs.Err() != nil || cs.ID() == 0), subsequent attempts will also return false. Otherwise, it is safe to call
|
|
// TryNext again until a change is available.
|
|
//
|
|
// This method requires driver version >= 1.2.0.
|
|
func (cs *ChangeStream) TryNext(ctx context.Context) bool {
|
|
return cs.next(ctx, true)
|
|
}
|
|
|
|
func (cs *ChangeStream) next(ctx context.Context, nonBlocking bool) bool {
|
|
// return false right away if the change stream has already errored or if cursor is closed.
|
|
if cs.err != nil {
|
|
return false
|
|
}
|
|
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
|
|
if len(cs.batch) == 0 {
|
|
cs.loopNext(ctx, nonBlocking)
|
|
if cs.err != nil {
|
|
cs.err = replaceErrors(cs.err)
|
|
return false
|
|
}
|
|
if len(cs.batch) == 0 {
|
|
return false
|
|
}
|
|
}
|
|
|
|
// successfully got non-empty batch
|
|
cs.Current = bson.Raw(cs.batch[0])
|
|
cs.batch = cs.batch[1:]
|
|
if cs.err = cs.storeResumeToken(); cs.err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (cs *ChangeStream) loopNext(ctx context.Context, nonBlocking bool) {
|
|
for {
|
|
if cs.cursor == nil {
|
|
return
|
|
}
|
|
|
|
if cs.cursor.Next(ctx) {
|
|
// non-empty batch returned
|
|
cs.batch, cs.err = cs.cursor.Batch().Documents()
|
|
return
|
|
}
|
|
|
|
cs.err = replaceErrors(cs.cursor.Err())
|
|
if cs.err == nil {
|
|
// Check if cursor is alive
|
|
if cs.ID() == 0 {
|
|
return
|
|
}
|
|
|
|
// If a getMore was done but the batch was empty, the batch cursor will return false with no error.
|
|
// Update the tracked resume token to catch the post batch resume token from the server response.
|
|
cs.updatePbrtFromCommand()
|
|
if nonBlocking {
|
|
// stop after a successful getMore, even though the batch was empty
|
|
return
|
|
}
|
|
continue // loop getMore until a non-empty batch is returned or an error occurs
|
|
}
|
|
|
|
if !cs.isResumableError() {
|
|
return
|
|
}
|
|
|
|
// ignore error from cursor close because if the cursor is deleted or errors we tried to close it and will remake and try to get next batch
|
|
_ = cs.cursor.Close(ctx)
|
|
if cs.err = cs.executeOperation(ctx, true); cs.err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (cs *ChangeStream) isResumableError() bool {
|
|
commandErr, ok := cs.err.(CommandError)
|
|
if !ok || commandErr.HasErrorLabel(networkErrorLabel) {
|
|
// All non-server errors or network errors are resumable.
|
|
return true
|
|
}
|
|
|
|
if commandErr.Code == errorCursorNotFound {
|
|
return true
|
|
}
|
|
|
|
// For wire versions 9 and above, a server error is resumable if it has the ResumableChangeStreamError label.
|
|
if cs.wireVersion != nil && cs.wireVersion.Includes(minResumableLabelWireVersion) {
|
|
return commandErr.HasErrorLabel(resumableErrorLabel)
|
|
}
|
|
|
|
// For wire versions below 9, a server error is resumable if its code is on the allowlist.
|
|
_, resumable := resumableChangeStreamErrors[commandErr.Code]
|
|
return resumable
|
|
}
|
|
|
|
// Returns true if the underlying cursor's batch is empty
|
|
func (cs *ChangeStream) emptyBatch() bool {
|
|
return cs.cursor.Batch().Empty()
|
|
}
|
|
|
|
// StreamType represents the cluster type against which a ChangeStream was created.
|
|
type StreamType uint8
|
|
|
|
// These constants represent valid change stream types. A change stream can be initialized over a collection, all
|
|
// collections in a database, or over a cluster.
|
|
const (
|
|
CollectionStream StreamType = iota
|
|
DatabaseStream
|
|
ClientStream
|
|
)
|