如何通过组合不同包的接口来创建接口？

大家好，我们又见面了啊~本文《如何通过组合不同包的接口来创建接口？》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

情况

对于代码库，我使用 pion/rtpio 包。

我试图通过添加 clos​​e() 函数来扩展接口 rtpwriter 。目标是创建一个 newrtpwritcloser() 函数，该函数返回具有组合函数的 writecloser。

从包中，我看到作者已经创建了接口 rtpwritecloser

type rtpwritecloser interface {
    rtpwriter
    io.closer
}

尝试

我这样做是为了重用该函数，这是错误的，但我还不知道替代方案：

func newrtpwritecloser(wc io.writecloser) rtpio.rtpwritecloser {
    writecloser := rtpio.newrtpwriter(wc)
    return writecloser
}

并期望wc有自己的clos​​e()函数准备好，因此返回的writecloser将满足接口rtpwritecloser。但是，我得到了 (missing method close) 错误。

函数newrtpwriter()是这样的：

func NewRTPWriter(w io.Writer) RTPWriter {
    return &RawRTPWriter{
        dst: w,
    }
}

问题

• 我们如何同时从多个嵌入接口创建一个具有所有所需功能的实例以满足嵌入接口？
• 在本例中，我们想要为 rtpwritecloser 接口创建 newrtpwritecloser 函数，但我们不能先创建一个 writer，然后向其添加 close 函数？
• 我是否必须创建一个 rtpwritecloser 结构并重写接受该结构的所有必要函数？ （看起来效率很低）

已搜索

我自己进行了搜索，例如：接口内部的接口以及组合或扩展接口，但它们并没有让我最终理解我的问题。

正确答案

您应该定义一个适配器结构，将 io.closer 语义添加到您的基本类型中：

type wrappingrtpwritecloser struct {
    w rtpwriter
    c io.closer
}

然后您应该定义 clos​​e 方法来满足该接口：

func (w *wrappingrtpwritecloser) close() error {
    return w.c.close()
}

然后，您应该在创建实例时创建包装结构的新引用：

func newrtpwritecloser(wc io.writecloser) rtpio.rtpwritecloser {
    writecloser := wrappingrtpwritecloser{
            w: rtpio.newrtpwriter(wc),
            c: wd,
    }
    return writecloser
}

另一种解决方案是使用包中提供的开箱即用的 rtppipe 函数，该函数返回 rtpreadcloser 和 rtpwritecloser 实例（将 rtpreadcloser 输入通过管道传输到 rtpwritecloserz qbendczqb 输出）：

// rtppipe creates a new rtppipe and returns the reader and writer.
func rtppipe() (rtpreadcloser, rtpwritecloser) {
    r, w := io.pipe()
    return &pipertpreader{closer: r, rtpreader: newrtpreader(r, 1500)}, &pipertpwriter{closer: w, rtpwriter: newrtpwriter(w)}
}

我相信您需要按照您的建议创建一个新结构来实现您的接口，我们将其命名为 rawrtpwritercloser 并添加 clos​​e() 的实现。

但是，您可以将结构嵌入到其他结构中，因此 clos​​e() 方法是您需要额外定义的唯一函数：

type RawRTPWriterCloser struct {
    rtpio.RawRTPWriter
}

func (rw *RawRTPWriterCloser) Close() error {
    panic("add implementation here")
}

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~