Question

要求：

type A struct {//some code}
type B struct {//some code}

func getData(db string) interface{} {
    if db == "dbA" {     // problem with this if condition
        type C A
    } else {
        type C B
    }
    var byteData []byte
    query := //cassandra query object
    iter := query.Iter()
    for iter.Scan(&byteData) {
        myObject := new(C)
        err := proto.Unmarshal(byteData, myObject)
        objects = append(objects, myObject)
    }
    return objects
}

基本上，我不想在循环内编写if条件。但是myObject在C的范围之外声明。

备用方式：我已经尝试过将myObject的类型设置为proto.Message，但这给了我错误“无效的内存地址或nil指针取消引用”

myObject := new(proto.Message)
err := proto.Unmarshal(byteData, myObject)
objects = append(objects, myObject)

另一种替代方法：我也不确定是否可以使用相同的变量（因此每次循环内我都尝试创建一个新变量）

PS：不需要太多Cassandra知识

帮助表示赞赏。谢谢！

编辑1：我要完成的工作是从数据库中获取几行。但是由于我有多个包含非常相似数据的表，因此我想以一种非常优化的方式在一个函数中进行操作。我想获取的值以字节存储，我正在使用proto将其转换为Golang对象。

编辑2： proto.Unmarshal需要第二个参数为proto.Message类型。因此，我不能使用空接口。（类型A和类型B均实现proto.Message）

编辑3：您可以在“ github.com/golang/protobuf”中找到原型。我用它将对象转换成字节，然后又转换回对象。

Answer 1

在Go的类型系统中，您想要实现的目标无法直接完成。

但是，借助interface{}和一流的函数值，我们可以轻松地完成它。

为此，我们声明一个新的构造方法，而不是声明一个新的type C：

var newC func() interface{}
if true {
    newC = func() interface{} {
        c := new(A)
        return c
    }
} else {
    newC = func() interface{} {
        c := new(B)
        return c
    }
}

然后，在循环中，将myObject := new(C)更改为myObject := newC()。完成了。

示例：https://play.golang.org/p/dBKBYrLqi_P

编辑：

由于您需要将参数设为proto.Message（我猜是interface），因此可以将其转换为proto.Message。

因此基本上，您可以将代码重新编写为：

type A struct {//some code}
type B struct {//some code}

func getData(db string) interface{} {
var newC func() interface{}
if true {
    newC = func() proto.Message {
        c := new(A)
        return c
    }
} else {
    newC = func() proto.Message {
        c := new(B)
        return c
    }
}
var byteData []byte
    query := //cassandra query object
    iter := query.Iter()
    for iter.Scan(&byteData) {
        myObject := newC()
        err := proto.Unmarshal(byteData, myObject)
        objects = append(objects, myObject)
    }
    return objects
}

如果条件在内部创建类型

1 个答案: