创建扩展类型数组

Question

我有一个结构体 A，用结构体 B 扩展（“子类化”）它，如下所示：

package main

type A struct {
    x int
}

type B struct {
    A
    y int
}

我想创建一个数组，在其中可以将 A 或 B 附加到它，以便此代码可以工作：

func main() {
    var m [2]B

    m[0] = B { A { 1 }, 2 }
    m[0].x = 0
    m[0].y = 0

    m[1] = A { 3 }
    m[1].x = 0
}

但它不能。如果我创建 B 类型的数组，我会得到“不能在赋值中使用结构体文字（A 类型）作为 B 类型”。如果我尝试创建类型 A 的数组，我会得到相同的错误（只是类型相反）。

所以我的问题是：数组应该是什么类型？

Answer 1

您可以使用 struct 值。例如，

package main

import "fmt"

type A struct {
    x int
}

type B struct {
    A
    y int
}

func main() {
    var m []interface{}
    m = append(m, B{A{1}, 2})
    m = append(m, A{3})
    fmt.Println(m[0], m[1])
    if b, ok := m[0].(B); ok {
        b.x = 0
        b.y = 0
        m[0] = b
    }
    if a, ok := m[1].(A); ok {
        a.x = 0
        m[1] = a
    }
    fmt.Println(m[0], m[1])
}

Output:
{{1} 2} {3}
{{0} 0} {0}

或者，您可以使用 struct 指针。例如，

package main

import "fmt"

type A struct {
    x int
}

type B struct {
    A
    y int
}

func main() {
    var m []interface{}
    m = append(m, &B{A{1}, 2})
    m = append(m, &A{3})
    fmt.Println(m[0], m[1])
    if b, ok := m[0].(*B); ok {
        b.x = 0
        b.y = 0
    }
    if a, ok := m[1].(*A); ok {
        a.x = 0
    }
    fmt.Println(m[0], m[1])
}

Output:
&{{1} 2} &{3}
&{{0} 0} &{0}

Answer 2

您需要将数组类型定义为 interface{} 而不是 B。然后你可以将这两种类型存储在那里。这是实现这一目标的唯一方法。如果两种类型都实现特定接口，那么您可以键入该接口，而不是通用接口{}