如何在 Go 中模拟 MongoDB 时添加 Clone 和 Copy 函数?

How to add Clone and Copy functions in mocking MongoDB in Go?

我读了这个 article 并且它有很好的在 Go 中模拟 MongoDB 的指南。但是在Clone()Copy()方法中存在一些问题。 我创建了这个接口和结构:

type ISession interface {
    DB(name string) IDatabase
    Close()
    Clone() ISession
    Copy() ISession
}

type IDatabase interface {
    C(name string) ICollection
}

type MongoSession struct {
    dbSession *mgo.Session
}

func (s MongoSession) DB(name string) IDatabase {
    return &MongoDatabase{Database: s.dbSession.DB(name)}
}

func (s MongoSession) Clone() ISession {
    //return session.clone
    return s.dbSession.Clone()
}

func (s MongoSession) Copy() ISession {
    return s.dbSession.Copy()
}

但是我遇到了这个错误

cannot use s.dbSession.Clone() (type *mgo.Session) as type ISession in return argument: *mgo.Session does not implement ISession (wrong type for Clone method) have Clone() *mgo.Session want Clone() ISession

cannot use s.dbSession.Copy() (type *mgo.Session) as type ISession in return argument: *mgo.Session does not implement ISession (wrong type for Clone method) have Clone() *mgo.Session want Clone() ISession

如何将 Clone()Copy() 方法添加到界面?

MongoSession.Copy()MongoSession.Clone() 必须 return 一个实现 ISession 的值。基本上,您正是为此创建了 MongoSession 类型:实现 ISession.

mgo.Session does not implement your ISession interface (e.g. because its Session.Clone() 方法的 return 类型为 *mgo.Session 而不是 ISession)。您应该创建 return 一个 MongoSession 的新值,您可以在其中包装复制或克隆的 *mgo.Session 值。

像这样:

func (s MongoSession) Clone() ISession {
    return MongoSession{dbSession: s.dbSession.Clone()}
}

func (s MongoSession) Copy() ISession {
    return MongoSession{dbSession: s.dbSession.Copy()}
}