类型转换与泛型：Go语言中Map类型转换的替代方案-Golang-PHP中文网

类型转换与泛型：Go语言中Map类型转换的替代方案

霞舞

发布： 2025-09-04 17:53:12

原创

491人浏览过

类型转换与泛型：go语言中map类型转换的替代方案

Go语言不支持隐式类型转换，这意味着直接将map[ID]int转换为map[int]int是不可行的。虽然Go语言本身并不支持泛型，但我们可以通过接口和结构体来实现类似泛型的效果，解决类型转换的问题。

使用接口实现泛型

为了避免编写多个相似的score()函数，我们可以定义一个接口scoreable，该接口定义了score()函数所需的方法。然后，我们创建实现了scoreable接口的结构体，这些结构体分别持有不同类型的ID（例如TeamID和PlayerID）的map。

type scID int

type scoreable interface {
    ids() []scID          // 获取所有ID的列表
    stats(scID) StatLine  // 获取指定ID的统计数据
    score(scID, int)      // 设置指定ID的分数
}

type teamScores struct {
    stats  map[TeamID]StatLine
    scores map[TeamID]int
}

func (s *teamScores) stats(id scID) StatLine {
    return s.stats[TeamID(id)]
}

func (s *teamScores) score(id scID, sc int) {
    s.scores[TeamID(id)] = sc
}

func (s *teamScores) ids() (a []scID) {
    for tid := range s.stats {
        a = append(a, scID(tid))
    }
    return
}

type playerScores struct {
    stats  map[PlayerID]StatLine
    scores map[PlayerID]int
}

func (s *playerScores) stats(id scID) StatLine {
    return s.stats[PlayerID(id)]
}

func (s *playerScores) score(id scID, sc int) {
    s.scores[PlayerID(id)] = sc
}

func (s *playerScores) ids() (a []scID) {
    for pid := range s.stats {
        a = append(a, scID(pid))
    }
    return
}

type StatLine map[StatID]float64
type StatID int
type TeamID int
type PlayerID int

登录后复制

编写通用的score()函数

现在，我们可以编写一个通用的score()函数，该函数接受scoreable接口类型的参数。该函数将遍历所有ID，获取统计数据，计算分数，并将分数设置回对应的结构体中。

func score(s scoreable) {
    // 存储中间值的map
    sum := make(map[scID]float64)
    // 遍历所有ID
    for _, id := range s.ids() {
        // 获取统计数据
        stats := s.stats(id)
        // 计算中间值
        sum[id] = 0.
        for _, statValue := range stats {
            sum[id] += statValue
        }
    }
    // 计算最终分数
    for id, s := range sum {
        score := int(s) // 模拟计算过程
        s.score(id, score)
    }
}

登录后复制

使用示例

为了使用这个通用的score()函数，我们需要创建teamScores或playerScores类型的对象，并将它们传递给score()函数。

天工大模型

中国首个对标ChatGPT的双千亿级大语言模型

115

查看详情

立即学习“go语言免费学习笔记（深入）”；

// 假设已经存在 leagueStats: map[TeamID]StatLine
leagueStats := make(map[TeamID]StatLine)
leagueStats[TeamID(1)] = StatLine{StatID(1): 10.0, StatID(2): 5.0}
leagueStats[TeamID(2)] = StatLine{StatID(1): 8.0, StatID(2): 7.0}

ts := &teamScores{
    stats:  leagueStats,
    scores: make(map[TeamID]int),
}

score(ts)

// 现在 ts.scores 包含了每个 TeamID 对应的分数
fmt.Println(ts.scores)

登录后复制