首页 > 后端开发 > Golang > 基于go-zero的微服务消息通信解决方案

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

王林
发布: 2023-06-22 17:19:37
原创
1863 人浏览过

随着微服务架构的普及,微服务之间的通信显得越来越重要。过去常用的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中文网其他相关文章!

相关标签:
来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板