Golang 函数调用中如何使用发布订阅模式-Golang-PHP中文网

Golang 函数调用中如何使用发布订阅模式

王林

发布： 2024-09-11 16:12:02

原创

722人浏览过

在 go 中，我们可以使用 pub/sub 模式进行函数调用：导入模块：导入必要的 pub/sub 模块。创建客户端和主题：创建 pub/sub 客户端并创建主题。函数调用实现：定义要执行的函数，以便在收到消息后调用。创建订阅：为主题创建一个订阅，允许接收消息。接收消息：使用 goroutine 接收来自订阅的消息。发布消息：将消息发布到主题，其中包含要调用的函数名称。

Golang 函数调用中如何使用发布订阅模式

Go 中使用发布-订阅模式进行函数调用

发布-订阅（Pub/Sub）模式是一种消息传递范例，其中发布者将消息发送到一个主题，而订阅者只能接收已订阅主题的消息。在 Go 中，我们可以使用 Pub/Sub 实现函数调用。

导入必要的模块

立即学习“go语言免费学习笔记（深入）”；

第一步是导入 cloud.google.com/go/pubsub 模块。

import (
    "context"
    "fmt"
    "time"

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

登录后复制

创建客户端和主题

然后，创建 Pub/Sub 客户端并创建主题。

ctx := context.Background()
client, err := pubsub.NewClient(ctx, "my-project")
if err != nil {
    // 处理错误
}

topicName := "my-topic"
topic, err := client.CreateTopic(ctx, topicName)
if err != nil {
    // 处理错误
}

登录后复制

函数调用实现

以下是函数调用实现：

// functionToCall 包含要执行的函数
func functionToCall(ctx context.Context, m *pubsub.Message) {
    fmt.Printf("收到消息：%s\n", string(m.Data))
    // 执行业务逻辑

    m.Ack()
}

登录后复制

创建订阅

接下来，为主题创建订阅。订阅是与主题关联的对象，它允许接收由发布者发送的消息。

subscriptionName := "my-subscription"
// 此处使用默认配置创建订阅
subscription, err := client.CreateSubscription(ctx, subscriptionName, pubsub.SubscriptionConfig{Topic: topic})
if err != nil {
    // 处理错误
}

登录后复制

启动一个 goroutine 来接收消息

现在，使用 goroutine 启动一个 loop 来接收消息。

go func() {
    ctx := context.Background()
    err := subscription.Receive(ctx, func(ctx context.Context, m *pubsub.Message) {
        functionToCall(ctx, m)
    })
    if err != nil {
        // 处理错误
    }
}

登录后复制

发布消息

要调用函数，我们需要发布一条消息。

result := &pubsub.PublishResult{}
// 设置消息属性
result, err = topic.Publish(ctx, &pubsub.Message{
    Data: []byte("调用函数"),
    Attributes: map[string]string{
        "function": "functionToCall",
    },
})
if err != nil {
    // 处理错误
}

// 阻塞等待发布结果
id, err := result.Get(ctx)
if err != nil {
    // 处理错误
}
fmt.Printf("发布消息成功：%s\n", id)

登录后复制

注意事项