泛型编程：统一处理不同结构类型的数据

泛型编程旨在处理不同结构类型的数据，而无需编写特定于每个类型的代码。在 Go 中，可以使用类型参数来定义通用函数，但当前版本不支持直接访问结构体字段。为了解决这个问题，可以使用接口方法来检索所需的数据，并将该方法用作类型约束的一部分。通过这种方法，泛型函数可以统一处理不同结构类型的数据，从而提高代码的可重用性和灵活性。

有两种结构类型：foo 和 bar，带有 int 数据成员 val。我正在尝试编写一个可以处理这两种类型的通用函数。我尝试了以下方法，但没有成功。

package main

import "fmt"

type foo struct {
    val int
}

type bar struct {
    val int
}

func add[t any](slice []t) int {
    var sum int
    for _, elem := range slice {
        sum += elem.val
    }
    return sum
}

func test() {
    f1 := foo{val: 2}
    f2 := foo{val: 2}
    fslice := []foo{f1, f2}
    fsum := add(fslice)
    fmt.printf("fsum = %d\n", fsum)

    b1 := bar{val: 3}
    b2 := bar{val: 3}
    bslice := []bar{b1, b2}
    bsum := add(bslice)
    fmt.printf("bsum = %d\n", bsum)
}

func main() {
    test()
}

编译器抛出以下错误。

$ go run generics1.go
# command-line-arguments
./generics1.go:16:15: elem.val undefined (type T has no field or method val)

go演示链接：https://go.dev/play/p/mdomh3xuwu7

有什么可能的方法来解决这个问题？

正确答案

每golang 1.18 release note

go 编译器不支持访问结构体字段 x.f，其中 x 是类型参数类型，即使类型参数类型集中的所有类型都有字段 f。我们可能会在未来版本中取消此限制。

您可以定义一个 getval() 接口方法来检索 val，并将此方法用作泛型 type constraint 的一部分。

示例代码

type Foo struct {
    val int
}

func (f Foo) GetVal() int {
    return f.val
}

type Bar struct {
    val int
}

func (b Bar) GetVal() int {
    return b.val
}

type MyType interface {
    Foo | Bar
    GetVal() int
}

func Add[T MyType](slice []T) int {
    var sum int
    for _, elem := range slice {
        sum += elem.GetVal()
    }
    return sum
}

https://go.dev/play/p/0eJZpqy7q8f

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~