使用指针接收器方法创建包装现有类型的接口

golang学习网今天将给大家带来《使用指针接收器方法创建包装现有类型的接口》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

我需要测试一个使用 google cloud pubsub 的应用，因此必须包装其类型 pubsub.client 和 pubsub.subscriber 以便进行测试。然而，尽管进行了多次尝试，我还是无法获得可编译的接口。

我尝试包装的方法的定义是：

func (s *subscription) receive(
    ctx context.context, f func(context.context, *message)) error

func (c *client) subscription(id string) *subscription

这是当前的代码。 receiver 接口（subscriber 的包装）似乎可以工作，但我怀疑它可能需要更改才能修复 subscriptionmaker，因此我将两者都包含在内。

注意：我已经尝试了几种引用和取消引用指针的变体，所以请不要告诉我更改它，除非您解释了为什么您建议的配置是正确的或者您亲自验证了它编译。

import (
    "context"

    "cloud.google.com/go/pubsub"
)

type receiver interface {
    receive(context.context, func(ctx context.context, msg *pubsub.message)) (err error)
}

// pubsub subscriptions implement receiver
var _ receiver = &pubsub.subscription{}

type subscriptionmaker interface {
    subscription(name string) (s receiver)
}

// pubsub clients implement subscriptionmaker
var _ subscriptionmaker = pubsub.client{}

当前错误消息：

common_types.go:21:5: cannot use "cloud.google.com/go/pubsub".Client literal (type "cloud.google.com/go/pubsub".Client) as type SubscriptionMaker in assignment:
    "cloud.google.com/go/pubsub".Client does not implement SubscriptionMaker (wrong type for Subscription method)
        have Subscription(string) *"cloud.google.com/go/pubsub".Subscription
        want Subscription(string) Receiver

解决方案

这是不可能的。

在接口上定义方法的类型签名时，它必须完全匹配。 func (c *client) subscription(id string) *subscription 返回一个 *subscription，而 *subscription 是一个有效的 receiver，但它不算符合接口方法 subscription(string) receiver。 go 需要函数签名的精确匹配，而不是通常用于接口的鸭子类型风格。

首先，对于大多数用途，使用 ptest 包可能是测试 pubsub 的更简单的方法。但是当然，您的具体问题可以适用于任何库，并且以下方法对很多事情都有用，而不仅仅是嘲笑 pubsub。

使用接口来模拟这样的库的更广泛目标是可行的。但是，当您希望模拟的库返回您无法模拟的具体类型（可能是由于未报告的字段）时，情况会很复杂。要采取的方法比通常值得的要复杂得多，因为可能有更简单的方法来测试您的代码。

但是，如果您打算这样做，则必须采取的方法是不要将整个包包装在接口中，而不仅仅是您希望模拟的特定方法。

您还需要包装您希望模拟的任何类型，这些类型也由您的界面返回或接受。这通常意味着您还需要修改生产代码（而不仅仅是测试代码），因此这有时可能会破坏现有代码库的交易。

我以前通常这样做是在模拟标准库的 sql 驱动程序之类的东西时，但这里可以应用相同的方法。本质上，您需要为 pubsub 库创建一个包装器包，甚至可以在生产代码中使用它。同样，这可能会对现有代码库造成很大的干扰，但只是为了说明目的。使用您定义的接口：

package mypubsub

import "cloud.google.com/go/pubsub"

type receiver interface {
    recieve(context.context, func(context.context, *pubsub.message) error)
}

type subscriptionmaker interface {
    subscription(string) receiver
}

然后您可以包装默认实现，以便在生产代码中使用：

// defaultClient wraps the default pubsub Client functionality.
type defaultClient struct {
    *pubsub.Client
}

func (d defaultImplementation) Subscription(name string) Receiver {
    return d.Client.Subscription()
}

当然，您需要扩展此包以包装您正在使用的大部分或全部 pubsub 包。这可能有点令人畏惧。

但是完成此操作后，请在代码中的任何位置使用 mypubsub 包，而不是直接依赖 pubsub 包。现在，您可以在需要测试的任何地方轻松更换模拟实现。

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