随着微服务架构的普及，微服务之间的通信显得越来越重要。过去常用的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在消息通信技术中具有广泛的使用、支持多种编程语言以及开源等优点，适用于绝大部分的企业级应用场景。在实际应用中，我们需要结合业务场景以及需求特点灵活使用。
