foo
公开类型A
的库,该库中Fn
的函数返回*A
。
我为A
定义了一个名为B
的"包装器":
type B foo.A
我可以在不取消引用A
的情况下将*A
转换为*B
吗?
换句话说,如果我有
a := foo.Fn() // a is a *A
b := B(*a)
return &b
如何在不使用*a
的情况下将*a
转换为*b
?
我问的原因是,在我正在使用的库中,github.com/coreos/bbolt
,从Open
函数返回的*DB
值包括一个sync.Mutex
,因此当我尝试复制Mutex
时,编译器会抱怨。
更新以解释我将如何使用它
我有一个
type Datastore struct {
*bolt.DB
}
我还有一个函数(众多函数之一),如下所示:
func (ds *Datastore) ReadOne(bucket, id string, data interface{}) error {
return ds.View(func(tx *bolt.Tx) error {
b, err := tx.CreateBucketIfNotExists([]byte(bucket))
if err != nil {
return fmt.Errorf("opening bucket %s: %v", bucket, err)
}
bytes := b.Get([]byte(id))
if bytes == nil {
return fmt.Errorf("id %s not found", id)
}
if err := json.Unmarshal(bytes, data); err != nil {
return fmt.Errorf("unmarshalling item: %v", err)
}
return nil
})
}
我想使用哈希映射模拟底层 BoltDB 数据库。我在嘲笑这个问题时遇到了一个问题,因为View
期望一个需要bolt.Tx
的功能。然后,该tx
用于在CreateBucketIfNotExists
中创建新存储桶。我不能用调用我的哈希映射模拟版本的CreateBucketIfNotExists
替换该匿名函数参数。
我想出了这个:
package boltdb
import (
"github.com/coreos/bbolt"
)
type (
bucket bolt.Bucket
// Bucket is a wrapper for bolt.Bucket to facilitate mocking.
Bucket interface {
ForEach(fn func([]byte, []byte) error) error
Get(key []byte) []byte
NextSequence() (uint64, error)
Put(key, value []byte) error
}
db bolt.DB
// DB is a wrapper for bolt.DB to facilitate mocking.
DB interface {
Close() error
Update(fn func(*Tx) error) error
View(fn func(*Tx) error) error
}
transaction bolt.Tx
// Tx is a wrapper for bolt.Tx to facilitate mocking.
Tx interface {
CreateBucketIfNotExists(name []byte) (Bucket, error)
}
)
// ForEach executes a function for each key/value pair in a bucket.
func (b *bucket) ForEach(fn func([]byte, []byte) error) error {
return ((*bolt.Bucket)(b)).ForEach(fn)
}
// Get retrieves the value for a key in the bucket.
func (b *bucket) Get(key []byte) []byte {
return ((*bolt.Bucket)(b)).Get(key)
}
// NextSequence returns an autoincrementing integer for the bucket.
func (b *bucket) NextSequence() (uint64, error) {
return ((*bolt.Bucket)(b)).NextSequence()
}
// Put sets the value for a key in the bucket.
func (b *bucket) Put(key, value []byte) error {
return ((*bolt.Bucket)(b)).Put(key, value)
}
// Close releases all database resources.
func (db *db) Close() error {
return ((*bolt.DB)(db)).Close()
}
// Update executes a function within the context of a read-write managed transaction.
func (db *db) Update(fn func(Tx) error) error {
return ((*bolt.DB)(db)).Update(func(tx *bolt.Tx) error {
t := transaction(*tx)
return fn(&t)
})
}
// View executes a function within the context of a managed read-only transaction.
func (db *db) View(fn func(Tx) error) error {
return ((*bolt.DB)(db)).View(func(tx *bolt.Tx) error {
t := transaction(*tx)
return fn(&t)
})
}
// CreateBucketIfNotExists creates a new bucket if it doesn't already exist.
func (tx *transaction) CreateBucketIfNotExists(name []byte) (Bucket, error) {
b, err := ((*bolt.Tx)(tx)).CreateBucketIfNotExists(name)
if err != nil {
return nil, err
}
w := bucket(*b)
return &w, nil
}
到目前为止,在我的代码中,我只使用上面显示的函数。如果需要新代码,我可以添加更多。
我将在实际代码中用DB
替换每个bolt.DB
,bolt.Tx
用Tx
替换,bolt.Bucket
用Bucket
替换。模拟程序将使用所有三种类型的替换,这些类型使用基础哈希映射而不是存储到磁盘。然后,我可以测试我的所有代码,直到数据库调用。
您可以简单/直接将类型*A
的值转换为类型为*B
的值,只需将*B
括起来:
a := foo.Fn() // a is a *A
b := (*B)(a)
return b
您甚至可以转换函数调用的返回值:
return (*B)(foo.Fn())
在Go Playground上尝试一下。
这是可能的,因为规范:转换:
在以下任何一种情况下,可以将非常量值
x
转换为类型T
:
x
可分配给T
。。
和规格:可分配性:
如果满足以下条件之一,则值
x
可分配给类型为T
的变量("x
可分配给T
"):
。
x
的类型V
和T
具有相同的基础类型,并且至少有一个V
或T
不是定义的类型。
*B
和*A
类型都没有定义,并且*B
的基础类型与*A
的基础类型相同(它是指向A
类型声明中任何类型的基础类型的指针)。