如何使用 mongo-go-driver 模拟光标

Question

我刚刚学习了 go 语言，然后使用https://github.com/mongodb/mongo-go-driver与 MongoDB 和 Golang 一起制作 REST API，然后我正在进行单元测试，但在模拟 Cursor 时我陷入困境MongoDB，因为 Cursor 是一个结构，是一个想法还是有人创造了它？

Answer 1

在我看来，模拟此类对象的最佳方法是定义一个接口，因为 go 接口是隐式实现的，您的代码可能不需要那么多更改。一旦你有了一个界面，你就可以使用一些第三方库来自动生成模拟，比如嘲笑

有关如何创建界面的示例

type Cursor interface{

  Next(ctx Context)

  Close(ctx Context)  

}

只需更改任何接收 mongodb 游标的函数即可使用自定义接口

Answer 2

我刚刚遇到这个问题。因为mongo.Cursor有一个内部字段保存[]byte-- Current，为了完全模拟，您需要包装mongo.Cursor. 以下是我为此创建的类型：

type MongoCollection interface {

    Find(ctx context.Context, filter interface{}, opts ...*options.FindOptions) (MongoCursor, error)

    FindOne(ctx context.Context, filter interface{}, opts ...*options.FindOneOptions) MongoDecoder

    Aggregate(ctx context.Context, pipeline interface{}, opts ...*options.AggregateOptions) (MongoCursor, error)

}

type MongoDecoder interface {

    DecodeBytes() (bson.Raw, error)

    Decode(val interface{}) error

    Err() error

}

type MongoCursor interface {

    Decode(val interface{}) error

    Err() error

    Next(ctx context.Context) bool

    Close(ctx context.Context) error

    ID() int64

    Current() bson.Raw

}

type mongoCursor struct {

    mongo.Cursor

}

func (m *mongoCursor) Current() bson.Raw {

    return m.Cursor.Current

}

不幸的是，这将是一个移动的目标。MongoCollection随着时间的推移，我将不得不向界面添加新功能。