隨著微服務架構的普及，微服務之間的通訊顯得越來越重要。過去常用的REST API通訊方式在微服務之間互相呼叫的情況下，有以下缺點：
因此，基於訊息佇列（Message Queue）實作微服務之間的非同步通訊已經成為了不錯的選擇。訊息佇列可以將訊息傳送到訊息伺服器，接受方可以非同步地接收和處理訊息，節省了服務之間的通訊耗時，提高了系統效能的同時也確保了通訊的可靠性。
然而，在架構實作方面，如何快速地將一個強大、高可擴展的訊息通訊方案落地呢？今天我們來聊聊基於go-zero的微服務訊息通訊解決方案。
實現微服務通信，我們需要工具來處理伺服器上的消息佇列。當下比較流行的有RabbitMQ、Kafka、RocketMQ等。本文將介紹RabbitMQ訊息佇列的使用。所以我們首先需要安裝和設定RabbitMQ。
相關安裝套件和安裝方法可以前往[RabbitMQ官網](https://www.rabbitmq.com/download.html)進行查看。這裡不做詳細的解釋。
安裝完成後，在RabbitMQ的web管理控制台上建立所需的vhost和使用者。這裡我們設定一個vhost為myhost，一個使用者名稱和密碼為admin/admin。
go-zero是一個用於建立可擴展的可維護的微服務工具包。它整合了單體應用中所有必要的工具，並提供了微服務、RPC和API網關等最常用的工具。我們需要在本機安裝go-zero開發環境。
go-zero基於微服務架構，提供了微服務、RPC、API閘道等常用的開發工具。使用go-zero來實作一個支援RabbitMQ訊息佇列的微服務架構非常簡單。
我們使用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表示服務的設定檔。
開啟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時必需的資訊。
我們需要加入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部分為處理應用邏輯的程式碼，具體程式碼實作可以根據具體業務需求進行編寫。
建立好handler和service之後，需要使用go-zero的命令列工具來為我們自動產生RPC程式碼。我們在命令列中，進入到與message.api檔案相同的目錄下，執行如下程式碼：
goctl rpc proto -src message.api -dir .
此命令將為我們自動產生與message.api檔案相關的gRPC協定文件，並將其放到目前目錄下。
在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中文網其他相關文章！
