正确的方法传递接口给方法

最近发现不少小伙伴都对Golang很感兴趣，所以今天继续给大家介绍Golang相关的知识，本文《正确的方法传递接口给方法》主要内容涉及到等等知识点，希望能帮到你！当然如果阅读本文时存在不同想法，可以在评论中表达，但是请勿使用过激的措辞~

偏好：我是 golang 新手，渴望改进。

所以我试图从 dave cheney 的演讲中学习：https://youtu.be/nweuro_w8he?t=812，我们将接口传递给方法以使代码更清晰、更通用。

我在下面实现了一个示例，其中有一个可以输出到 std.out 或文件的结构。然而，我觉得仅仅制作空结构（在我的示例中称为“打印”和“保存”）有点多余，这样我就可以附加方法。我该如何改进？

package main

import (
    "fmt"
    "io"
    "io/ioutil"
    "log"
)

type file struct {
    data string
}

func (f *file) output(w io.Writer) error {
    len, err := w.Write([]byte(f.data))
    if err != nil {
        return err
    } else {
        log.Println("Bytes Out: ", len)
        return nil
    }
}

type print struct{}

func (print *print) Write(p []byte) (n int, err error) {
    return fmt.Printf("%s\n", p)
}

type save struct{}

func (s *save) Write(p []byte) (n int, err error) {
    err = ioutil.WriteFile("./dat1", []byte(p), 0644)
    if err != nil {
        return -1, err
    }
    return len(p), nil
}
func main() {
    f := file{"test"}
    s := save{}
    p := print{}

    f.output(&s)
    f.output(&p)
}

解决方案

此处不需要 print 或 save 类型。您有一个方法 output，它采用 io.writer；重点是您可以将其传递给任何作家。 stdout 和文件都是写入器，因此这里的其他内容都是不必要的。你可以：

func main() {
    f := file{"test"}

    fp,err := os.create("./dat1")
    if err != nil {
        panic(err)
    }
    f.output(fp)  // this writes to a file
    f.output(os.stdout)  // this writes to stdout
}

你的意图不清楚，但回答你原来的问题：如何将函数作为接口值传递？

您可以创建一个类似适配器的类型来实现 io.writer，并且当调用其 write() 方法时，它会转发到您选择的函数。一个典型的例子是 http.HandlerFunc：它是一个实现 http.Handler 的函数类型，因此当需要 http.handler 实现时可以传递函数。

对于 io.writer 适配器，它可能如下所示：

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

func (w writefunc) write(p []byte) (n int, err error) {
    return w(p)
}

如果你有这样的函数：

func printwrite(p []byte) (n int, err error) {
    return fmt.printf("%s\n", p)
}

func savewrite(p []byte) (n int, err error) {
    err = ioutil.writefile("./dat1", []byte(p), 0644)
    if err != nil {
        return -1, err
    }
    return len(p), nil
}

您可以将它们用作 io.writer，如下所示：

f.output(WriteFunc(PrintWrite))
f.output(WriteFunc(SaveWrite))

在 Go Playground 上尝试一下。

终于介绍完啦！小伙伴们，这篇关于《正确的方法传递接口给方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！