基于go-zero的微服务消息通信解决方案

随着微服务架构的普及，微服务之间的通信显得越来越重要。过去常用的REST API通信方式在微服务之间互相调用的情况下，存在以下缺点：

• 频繁的网络请求会带来延迟和性能瓶颈；
• 对于高频次的请求，短时间内的大量请求可能会导致服务崩溃；
• 对于数据传输量较大的场景，基于HTTP协议的传输方式也极易产生低效的问题。

因此，基于消息队列（Message Queue）实现微服务之间的异步通信已经成为了不错的选择。消息队列可以将消息发送到消息服务器，接受方可以异步地接收和处理消息，节省了服务之间的通讯耗时，提高了系统性能的同时也保证了通信的可靠性。

然而，在架构实现方面，如何快速地将一个强大、高可扩展的消息通信方案落地呢？今天我们来聊聊基于go-zero的微服务消息通信解决方案。

环境配置

实现微服务通信，我们需要工具来处理服务器上的消息队列。当下比较流行的有RabbitMQ、Kafka、RocketMQ等。本文将介绍RabbitMQ消息队列的使用。所以我们首先需要安装和配置RabbitMQ。

安装RabbitMQ

相关安装包和安装方法可以前往[RabbitMQ官网](https://www.rabbitmq.com/download.html)进行查看。这里不做详细的解释。

安装完成后，在RabbitMQ的web管理控制台上创建所需的vhost和用户。这里我们设置一个vhost为myhost，一个用户名和密码为admin/admin。

安装go-zero

go-zero是一个用于构建可扩展的可维护的微服务工具包。它集成了单体应用中所有必需的工具，并提供了微服务、RPC和API网关等最常用的工具。我们需要在本地安装go-zero开发环境。

go-zero基础架构

go-zero基于微服务架构，提供了微服务、RPC、API网关等常用的开发工具。使用go-zero来实现一个支持RabbitMQ消息队列的微服务架构非常简单。

1. 创建服务

我们使用goctl工具（go-zero的命令行工具）来快速创建服务。首先，在go-zero命令行中进入到创建服务的目录中，然后运行如下命令：

goctl api new message

运行的结果如下：

- api
    - etc
        - message-api.yaml
    - internal
        - config
            - config.go
        - handler
            - message_handler.go
        - service
            - message_service.go
    - message.api
- Dockerfile
- go.mod
- go.sum
- main.go

通过goctl api new命令创建的服务中包含了我们所需的基础目录和默认实现。其中- api表示服务的名称，- etc/message-api.yaml表示服务的配置文件。

2. 配置文件

打开etc/message-api.yaml，配置我们的RabbitMQ信息：

Name: message-api
Host: 0.0.0.0
Port: 8080
JwtSecret: # 在API网关中使用

MessageQueue:
  Host: 127.0.0.1
  Port: 5672 # RabbitMQ端口
  User: admin # RabbitMQ用户名
  Password: admin # RabbitMQ密码
  VirtualHost: myhost # RabbitMQ vhost

其中MessageQueue部分指定了使用RabbitMQ时必需的信息。

3. handler和service实现

我们需要添加handler和service实现来处理RabbitMQ消息队列中的消息。在handler目录下添加message_handler.go文件，并添加如下代码：

package handler

import (
    "context"

    "github.com/tal-tech/go-zero/core/logx"
    "github.com/tal-tech/go-zero/rest/httpx"
    "github.com/tal-tech/go-zero/zmq"
)

type MessageHandler struct {
    messageService *zmq.Service
}

func NewMessageHandler(ctx context.Context) *MessageHandler {
    return &MessageHandler{
        messageService: zmq.NewService("my-exchange", "my-key", false), // 确定队列信息
    }
}

func (h *MessageHandler) SendMessage(ctx context.Context, req *MessageRequest) (*MessageResponse, error) {
    logx.Infof("handler receive message %v", req.Message)

    err := h.messageService.SendMessage(zmq.NewMessage(req.Message)) // 发送消息到队列

    if err != nil {
        logx.Error(err)
        return nil, httpx.NewDefaultError(httpx.InternalServerErrorStatus)
    }

    return &MessageResponse{}, nil
}

其中，我们使用了go-zero提供的zmq工具来实现与RabbitMQ消息队列的交互（详见[go-zero官方文档](https://go-zero.dev/cn/tools/zmq)）。

在service目录中添加message_service.go文件，并添加如下代码：

package service

import (
    "context"

    "message/internal/biz"
)

type MessageApi struct {
    bc *biz.BizClient
}

func NewMessageApi(bc *biz.BizClient) *MessageApi {
    return &MessageApi{
        bc: bc,
    }
}

func (s *MessageApi) SendMessage(ctx context.Context, req *MessageRequest) (*MessageResponse, error) {
    _, err := s.bc.SendMessage(ctx, req.Message)

    if err != nil {
        return nil, err
    }

    return &MessageResponse{}, nil
}

其中，biz部分为处理应用逻辑的代码，具体代码实现可以根据具体业务需求进行编写。

4. 代码生成

创建好handler和service之后，需要使用go-zero的命令行工具来为我们自动生成RPC代码。我们在命令行中，进入到与message.api文件相同的目录下，运行如下代码：

goctl rpc proto -src message.api -dir .

此命令将为我们自动生成与message.api文件相关的gRPC协议文件，并将其放到当前目录下。

5. 注册服务

在internal/config/config.go中增加如下代码：

// register MessageApi
group := server.Group("/")
messageSvc := service.NewMessageApi(biz.NewBizClient())
server.POST(group+"/send-message", httpx.NewHandler(messageSvc.SendMessage))

go func() {
    err = zmq.NewSubscriber("my-exchange", "my-key", false).Subscribe(func(msg zmq.Message) {
        logx.Infof("[RabbitMQ Subscriber] receive message %s", string(msg.Body))
    }) // 订阅队列

    if err != nil {
        panic(err)
    }
}()

其中，我们对应用中的MessageApi服务进行了注册，在zmq.NewSubscriber调用中进行了订阅。在订阅到消息后，我们通过logx.Infof来直接处理RabbitMQ队列中的消息。

最后，在main.go中的Run方法中添加如下代码：

s := &http.Server{
    Handler: server,
    Addr:    ":"+cfg.Port,
}

上述代码将服务配置为可运行的HTTP服务器，并绑定到指定的端口上。最后我们就可以测试我们的服务是否正常了。

总结

本文我们介绍了基于go-zero的微服务消息通信解决方案，并通过示例代码实现了支持RabbitMQ的微服务架构。RabbitMQ在消息通信技术中具有广泛的使用、支持多种编程语言以及开源等优点，适用于绝大部分的企业级应用场景。在实际应用中，我们需要结合业务场景以及需求特点灵活使用。

以上是基于go-zero的微服务消息通信解决方案的详细内容。更多信息请关注PHP中文网其他相关文章！