将go函数指针转为接口

将go函数指针转为接口

(金庆的专栏 2020.2)

golang 中的接口如下：

type Writer interface {
    Write func(p []byte) (n int, err error)
}

一般API参数要求一个接口，而不是一个函数指针，如 io.Copy() 需要输入一个 Writer 和 Reader：

func Copy(dst Writer, src Reader) (written int64, err error)

而不是这样2个函数指针：

func CopyWithFunc(writeFunc func([]byte) (int, error), readRunc func([]byte) (int, error)) (written int64, err error)

大家统一使用接口，而不是接口和函数指针混用，可以避免API复杂化。
如 io.Copy() 有2个参数，如果要支持接口和函数指针混用，就会变成4个 Copy() 重载。
golang 没有重载，就只能用4个不同的函数名。

在实际使用中，需要将函数转化成接口，才能调用 io.Copy().
如有一个函数:

func MyWriteFunction(p []byte) (n int, err error) { 
    fmt.Print("%v",p)
    return len(p),nil
}

调用 io.Copy() 时需要创建一个 Writer，并将该函数指针转型为Writer后使用。
这里用 WriteFunc 类型实现 Writer。

type WriteFunc func(p []byte) (n int, err error)

func (wf WriteFunc) Write(p []byte) (n int, err error) {
    return wf(p)
}

WriteFunc 本身是个与 MyWriteFunction 同类型的函数类型，同时实现了 Writer 接口。
所以 MyWriteFunction 可以直接转成WriteFunc类型成为一个 Writer.
这样就可以调用 io.Copy() 了：

io.Copy(WriteFunc(MyWriteFunction), strings.NewReader("Hello world"))

参考：https://stackoverflow.com/questions/20728965/golang-function-pointer-as-a-part-of-a-struct