Golang接口型函数使用技巧
原文链接:https://www.zhoubotong.site/post/51.html
什么是接口型函数?顾名思义接口函数指的是用函数实现接口,这样在调用的时候就会非常简便,这种方式适用于只有一个函数的接口。
这里以迭代一个map为例,演示这一实现的技巧。
常规接口实现
defer语句用于延迟函数调用,每次会把一个函数压入栈中,函数返回前再把延迟的函数取出并执行。延迟函数可以有参数:
-
延迟函数的参数在defer语句出现时就已确定下来(传值的就是当前值);
-
延迟函数执行按后进先出顺序执行;
-
延迟函数可操作主函数的具名返回值(修改返回值);
-
type Handler interface { DoFunc(k, v interface{}) } func DoEach(m map[interface{}]interface{}, h Handler) { if m != nil && len(m) > 0 { for k, v := range m { h.DoFunc(k, v) } } }
这里我们定义了一个Handler接口,只有一个DoFunc方法,接收k,v两个参数,这就是一个接口了,我们后面会实现他,具体做什么由我们的实现决定。
然后我们定义了一个DoEach函数,该函数的功能就是迭代传递过来的map参数,然后把map的每个key和value值传递给Handler的DoFunc方法,
具体由这个Handler的实现来决定,这也是面向接口编程。
说得再多不如来点实际的例子:用我们刚刚定义的DoEach方法和Handler接口。
package main import "fmt" type Handler interface { DoFunc(k, v interface{}) } func DoEach(m map[interface{}]interface{}, h Handler) { if m != nil && len(m) > 0 { for k, v := range m { h.DoFunc(k, v) } } } type greet string func ( g greet) DoFunc(k, v interface{}) { fmt.Printf("%s,在下%s,我的必杀技是%s\n", g, k, v) } func main() { persons := make(map[interface{}]interface{}) persons["乔峰"] = "龙爪手" persons["鸠摩智"] = "小无相功" persons["慕容复"] = "斗转星移" var g greet = "诸位英雄" DoEach(persons, g) }
输出:
诸位英雄,在下乔峰,我的必杀技是龙爪手 诸位英雄,在下鸠摩智,我的必杀技是小无相功 诸位英雄,在下慕容复,我的必杀技是斗转星移
以上实现,我们定义了一个map来存储几位大佬,map的key是大佬的名字,value是该大佬的绝技。greet是我们新定义的类型,
其对应基本类型string,该greet实现了Handler接口,打印出自我介绍的信息。
接口型函数出场
关于上面的实现,我们可以发现,有两点不太好:
-
因为必须要实现Handler接口,DoFunc这个方法名不能修改,不能定义一个更有意义的名字
-
必须要新定义一个类型,才可以实现Handler接口,才能使用DoEach函数
首先我们先解决第一个问题,根据我们具体做的事情定义一个更有意义的方法名,比如例子中是自我介绍,
那么我们使用selfintroduction是不是要比DoFunc这个语义的方法要好呢。
如果调用者改了方法名,那么就不能实现Handler接口,还要使用DoEach方法怎么办?那就是由提供DoEach函数的负责提供Handler的实现,
我们改造下代码如下:
type HandlerFunc func(k, v interface{}) func (f HandlerFunc) DoFunc(k, v interface{}) { f(k, v) }
上面代码我们定义了一个新的类型HandlerFunc,它是一个func(k, v interface{})类型,然后这个新的HandlerFunc实现了Handler接口(原始实现方式中的
type Handler interface { DoFunc(k, v interface{}) }
),DoFunc方法的实现是调用HandlerFunc本身,因为HandlerFunc类型的变量就是一个方法。现在我们使用这种方式实现同样的效果。
完整代码如下:
package main import "fmt" type Handler interface { DoFunc(k, v interface{}) } type HandlerFunc func(k, v interface{}) func (f HandlerFunc) DoFunc(k, v interface{}) { f(k, v) } type greet string func (g greet) selfintroduction(k, v interface{}) { fmt.Printf("%s,在下%s,我的必杀技是%s\n", g, k, v) } func DoEach(m map[interface{}]interface{}, h Handler) { if m != nil && len(m) > 0 { for k, v := range m { h.DoFunc(k, v) } } } func main() { persons := make(map[interface{}]interface{}) persons["乔峰"] = "龙爪手" persons["鸠摩智"] = "小无相功" persons["慕容复"] = "斗转星移" var g greet = "诸位英雄" DoEach(persons, HandlerFunc(g.selfintroduction)) }
输出:
诸位英雄,在下乔峰,我的必杀技是龙爪手 诸位英雄,在下鸠摩智,我的必杀技是小无相功 诸位英雄,在下慕容复,我的必杀技是斗转星移
还是差不多原来的实现,只是把原接口方法名DoFunc改为selfintroduction。HandlerFunc(g.selfintroduction)不是方法的调用,而是转型,因为selfintroduction和HandlerFunc是同一种类型,
所以可以强制转型。转型后,因为HandlerFunc实现了Handler接口,所以我们就可以继续使用原来的DoEach方法了。
进一步改造
现在解决了命名的问题,但是每次强制转型是不是不太好?我们继续重构下,可以采用新定义一个函数的方式,帮助调用者强制转型。
完整代码如下:
package main import "fmt" type Handler interface { DoFunc(k, v interface{}) } type HandlerFunc func(k, v interface{}) func (f HandlerFunc) DoFunc(k, v interface{}) { f(k, v) } type greet string func (g greet) selfintroduction(k, v interface{}) { fmt.Printf("%s,在下%s,我的必杀技是%s\n", g, k, v) } func DoEach(m map[interface{}]interface{}, h Handler) { if m != nil && len(m) > 0 { for k, v := range m { h.DoFunc(k, v) } } } func EachFunc(m map[interface{}]interface{}, f func(k, v interface{})) { DoEach(m, HandlerFunc(f)) } func main() { persons := make(map[interface{}]interface{}) persons["乔峰"] = "龙爪手" persons["鸠摩智"] = "小无相功" persons["慕容复"] = "斗转星移" var g greet = "诸位英雄" EachFunc(persons, g.selfintroduction) }
上面我们新增了一个EachFunc函数,帮助调用者强制转型,调用者就不用自己做了。
现在我们发现EachFunc函数接收的是一个func(k, v interface{})类型的函数,没有必要实现原Handler接口了,所以我们新的类型可以去掉不用了。
去掉了自定义类型greet之后,整个代码更简洁,可读性是不是更好点?简洁干净的完整代码如下:
package main import "fmt" type Handler interface { DoFunc(k, v interface{}) } type HandlerFunc func(k, v interface{}) func (f HandlerFunc) DoFunc(k, v interface{}) { f(k, v) } func DoEach(m map[interface{}]interface{}, h Handler) { if m != nil && len(m) > 0 { for k, v := range m { h.DoFunc(k, v) } } } func EachFunc(m map[interface{}]interface{}, f func(k, v interface{})) { DoEach(m, HandlerFunc(f)) } func selfintroduction(k, v interface{}) { fmt.Printf("诸位英雄,在下%s,我的必杀技是%s\n", k, v) } func main() { persons := make(map[interface{}]interface{}) persons["乔峰"] = "龙爪手" persons["鸠摩智"] = "小无相功" persons["慕容复"] = "斗转星移" EachFunc(persons, selfintroduction) }
以上关于函数型接口就写完了,如果大家仔细留意,发现和我们自己平时使用的http.Handle方法非常类似,其实接口http.Handler就是这么实现的。
type Handler interface { ServeHTTP(ResponseWriter, *Request) } func Handle(pattern string, handler Handler) { DefaultServeMux.Handle(pattern, handler) } func HandleFunc(pattern string, handler func(ResponseWriter, *Request)) { DefaultServeMux.HandleFunc(pattern, handler) }
这是一种非常好的技巧,提供两种函数,既可以以接口的方式使用,也可以以方法的方式,
对应我们例子中的DoEach和EachFunc这两个函数,灵活方便,也更符合自然语言规则吧。