金庆的专栏

  C++博客 :: 首页 :: 新随笔 :: 联系 :: 聚合  :: 管理 ::
  423 随笔 :: 0 文章 :: 454 评论 :: 0 Trackbacks

# 将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
posted on 2020-02-13 14:38 金庆 阅读(377) 评论(0)  编辑 收藏 引用 所属分类: 3. Golang

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理