Go Generics

[kagxin]

示例

package main

import "fmt"

type Number interface {
    int64 | float64
}

// SumIntsOrFloats sums the values of map m. It supports both int64 and float64
// as types for map values.
func SumIntsOrFloats[K comparable, V Number](m map[K]V) V {
    var s V
    for _, v := range m {
        s += v
    }
    return s
}

// IF 模仿三目运算
func IF[V any](flag bool, a, b V) V {
    if flag {
        return a
    }
    return b
}

func main() {
    // Initialize a map for the integer values
    ints := map[string]int64{
        "first":  34,
        "second": 12,
    }
    // Initialize a map for the float values
    floats := map[string]float64{
        "first":  35.98,
        "second": 26.99,
    }
    fmt.Printf("Generic Sums: %v and %v\n",
        SumIntsOrFloats[string, int64](ints),
        SumIntsOrFloats[string, float64](floats))

    fmt.Printf("Generic Sums, type parameters inferred: %v and %v\n",
        SumIntsOrFloats(ints),
        SumIntsOrFloats(floats))

    fmt.Printf("IF: %+v", IF[int](true, 1, 2))
}

如何理解泛型

有几个关键点：

• 我们要在 [ ] 中为泛型函数声明它支持的类型(类型约束) func SumIntsOrFloats[K comparable, V Number](m map[K]V) V
• 我们使用这个函数的时候指定它使用的参数类型，或者让编译器自己推断 IF[int](true, 1, 2), IF(true, 1, 2)
• 我们可以声明 类型约束

  type Number interface {
  int64 | float64
  }

新增类型

为了方便泛型使用，新增了两个类型 any 和 comparable，不过这两种类型都只能用于参数类型约束 不能用于声明变量。

vscode 如何尝试 generics

1、到官网下载安装 go1.18 2、vscode cmd + shift + p 3、选择 Go: Install/Update Tools 勾选更新所有工具

REF

https://go.dev/doc/tutorial/generics https://go.dev/blog/go1.18