模拟mgo链接函数

Question

在测试我的一些代码库时，我发现用接口模拟"gopkg.in/mgo.v2“中的对象是有用的。我遇到了一个问题，mgo.Query没有实现我的接口查询。

import mgo "gopkg.in/mgo.v2"

type Collection interface {
    FindId(interface{}) Query
    Find(interface{}) Query
    UpdateId(interface{}, interface{}) error
    Update(interface{}, interface{}) error
    UpsertId(interface{}, interface{}) (interface{}, error)
    Insert(...interface{}) error
    RemoveId(interface{}) error
}

type Query interface {
    One(interface{}) error
    All(interface{}) error
    Select(interface{}) Query
}

var _ Query = (*mgo.Query)(nil)

查询强制转换启动错误：

cannot use (*mgo.Query)(nil) (type *mgo.Query) as type Query in assignment:
    *mgo.Query does not implement Query (wrong type for Select method)
        have Select(interface {}) *mgo.Query
        want Select(interface {}) Query

这是否是链函数无法在接口中定义的问题？我不知道如何使Select与mgo实现相匹配。

Answer 1

函数签名是不同的，这就是您遇到编译错误的原因。Select的接口版本返回Query类型，而mgo的Select返回*mgo.Query，这是一种不同的类型。即使该类型实现了您的接口，函数签名仍然是不同的。您将需要添加另一层，在这里，您可以包装mgo包。

type MgoQuery struct {
    *mgo.Query
}

func (q *MgoQuery) Select(selector interface{}) Query {
    return q.Query.Select(selector)
}

...