登录
首页 >  Golang >  Go问答

能不能在 Go 中使用泛型来实现"map"和"reduce"操作?

来源:stackoverflow

时间:2024-02-14 21:36:24 178浏览 收藏

哈喽!大家好,很高兴又见面了,我是golang学习网的一名作者,今天由我给大家带来一篇《能不能在 Go 中使用泛型来实现"map"和"reduce"操作?》,本文主要会讲到等等知识点,希望大家一起学习进步,也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧!

问题内容

我决定既然泛型已经被引入 go,像 map/reduce 这样的东西应该是可能的。因此,我天真地尝试了一下,但得到了错误: ./prog.go:18:36:无法在 mapper 的参数中使用事物(int 类型的变量)作为类型 i

这并不能解释问题是否是根本性的,或者我只是在语法上做错了什么。 go 中可以实现通用的 map/reduce 吗?

package main

import "fmt"

func main() {
    things := []int{1, 2, 3, 4}
    results := Map(things, func(t int) int {
        return t + 1
    })
    fmt.Printf("%v", results)
}

func Map[I interface{}, O interface{}](things []I, mapper func(thing I) O) []O {
    results := make([]O, 0, len(things))
    for thing := range things {
        results = append(results, mapper(thing))
    }
    return results
}

正确答案


这可以很容易地完成。您的代码中有一个错误,尽管就在这里:

for thing := range things {

您正在迭代索引值 (int),而不是 i 类型的值。您还指定了 2 个约束(类型 io),均设置为 interface{}。您可以直接使用 any(它是 interface{} 的简写)

所以简单地写:

func map[t any, o any](things []t, mapper func(thing t) o) []o {
    result := make([]o, 0, len(things))
    for _, thing := range things {
        result = append(result, mapper(thing))
    }
    return result
}

Demo

这与我在 codereview exchange here 上审阅的一些代码密切相关。在浏览完代码并编写带有大量建议的片段后,我决定创建一个包并将其放在 github 上。您可以找到存储库 here

其中有一些示例可能会派上用场,或者帮助您解决 golang 中 wrt 泛型的一些其他问题。我特别考虑了这一点,您可以使用回调来过滤通用地图类型,如下所示:

// given the smap type
type smap[k comparable, v any] struct {
    mu *sync.rwmutex
    m  map[k]v
}

// filter returns a map containing the elements that matched the filter callback argument
func (s *smap[k, v]) filter(cb func(k, v) bool) map[k]v {
    s.mu.rlock()
    defer s.mu.runlock()
    ret := make(map[k]v, len(s.m))
    for k, v := range s.m {
        if cb(k, v) {
            ret[k] = v
        }
    }
    return ret
}

您对 range 的使用不正确。从 range 中提取的单个变量将是索引(类型为 int),而不是值(类型为 i,在本例中恰好是 int)。

尝试

for _, thing := range things{...}

文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《能不能在 Go 中使用泛型来实现"map"和"reduce"操作?》文章吧,也可关注golang学习网公众号了解相关技术文章。

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>