Go 入门 - 使用interface的map函数

使用interface的map函数,让它至少能同时工作于int和string

package main

import "fmt"

type e interface {

}

func mult2(f e) e {
	switch f.(type) {
	case int:
		return f.(int) * 2
	case string:
		return f.(string) + f.(string) + f.(string) + f.(string)
	}
	return f
}

func Map(n []e, f func(e) e) []e {
	m := make([]e, len(n))
	for k, v :=range n {
		m[k] = f(v)
	}
	return m
}

func main()  {
	m := []e{1,2,3,4}
	s := []e{"a","b","c","d"}
	mf := Map(m, mult2)
	sf := Map(s, mult2)
	fmt.Printf("%v\n",mf)
	fmt.Printf("%v\n",sf)
}

猜你喜欢

转载自blog.csdn.net/zhengwish/article/details/87694173