在 go-zero 中使用 rocketmq gozero rocketmq

使用 docker-compose.yaml 创建并运行容器

准备仓库

version: '3.5'

services:
  rmqnamesrv:
    image: foxiswho/rocketmq:4.8.0
#    image: registry.cn-hangzhou.aliyuncs.com/foxiswho/rocketmq:4.7.0
    container_name: rmqnamesrv
    ports:
      - 9876:9876
    volumes:
      - ./rmqs/logs:/home/rocketmq/logs
      - ./rmqs/store:/home/rocketmq/store
    environment:
      JAVA_OPT_EXT: "-Duser.home=/home/rocketmq -Xms512M -Xmx512M -Xmn128m"
    command: ["sh","mqnamesrv"]
    networks:
        rmq:
          aliases:
            - rmqnamesrv
  rmqbroker:
    image: foxiswho/rocketmq:4.8.0
#    image: registry.cn-hangzhou.aliyuncs.com/foxiswho/rocketmq:4.7.0
    container_name: rmqbroker
    ports:
      - 10909:10909
      - 10911:10911
    volumes:
      - ./rmq/logs:/home/rocketmq/logs
      - ./rmq/store:/home/rocketmq/store
      - ./rmq/brokerconf/broker.conf:/etc/rocketmq/broker.conf
    environment:
        JAVA_OPT_EXT: "-Duser.home=/home/rocketmq -Xms512M -Xmx512M -Xmn128m"
    command: ["sh","mqbroker","-c","/etc/rocketmq/broker.conf","-n","rmqnamesrv:9876","autoCreateTopicEnable=true"]
    depends_on:
      - rmqnamesrv
    networks:
      rmq:
        aliases:
          - rmqbroker

  rmqconsole:
    image: styletang/rocketmq-console-ng
    container_name: rmqconsole
    ports:
      - 8180:8080
    environment:
        JAVA_OPTS: "-Drocketmq.namesrv.addr=rmqnamesrv:9876 -Dcom.rocketmq.sendMessageWithVIPChannel=false"
    depends_on:
      - rmqnamesrv
    networks:
      rmq:
        aliases:
          - rmqconsole

networks:
  rmq:
    name: rmq
    driver: bridge

执行以下命令

# 创建目录
mkdir -p ./rmqs/logs
mkdir -p ./rmqs/store
mkdir -p ./rmq/logs
mkdir -p ./rmq/store

# 设置目录权限
chmod -R 777 ./rmqs/logs
chmod -R 777 ./rmqs/store
chmod -R 777 ./rmq/logs
chmod -R 777 ./rmq/store

# 下载并启动容器，且为 后台 自动启动
docker-compose up -d

# 显示 rocketmq 容器
docker ps |grep rocketmq

编写代码

使用的是 apache 开源的 rocketmq-client-go 库
"github.com/apache/rocketmq-client-go/v2"

初始化 Protocer 跟 PushConsumer 较多为重复代码所以打包进入 common

// 目录格式
common
├── xrocketmq
│   └── rocket.go
└── xxx...

新建 Producer

func NewProducer(groupName , topic string, str string, delay int) error {
	res, err := rocketmq.NewProducer(
		producer.WithGroupName(groupName),
		producer.WithNsResolver(primitive.NewPassthroughResolver([]string{address})),
		producer.WithRetry(2),
	)
	if err != nil {
		return err
	}

	err = res.Start()
	if err != nil {
		return err
	}

	msg := &primitive.Message{
		Topic: topic,
		Body:  []byte(str),
	}
	msg.WithDelayTimeLevel(delay)
	// 发送
	_, err = res.SendSync(context.Background(), msg)
	if err != nil {
		return errors.New(fmt.Sprintf("发送消息失败 : %s\n", err))
	}

	err = res.Shutdown()
	if err != nil {
		return errors.New(fmt.Sprintf("producer shutdown errpr : %s\n", err))
	}
	return nil
}

新建消费者 PushConsumer

func NewPushConsumer(groupName string) (rocketmq.PushConsumer, error) {
	res, err := rocketmq.NewPushConsumer(
		consumer.WithGroupName(groupName),
		consumer.WithNsResolver(primitive.NewPassthroughResolver([]string{address})),
	)
	if err != nil {
		return nil, err
	}

	return res, nil
}

先编写消费端

rmq 需要保持启动，这里偷懒，使用 goctl api 指令进行生成相关代码

hello.api

syntax = "v1"

type (
	HelloReq{
	}
	HelloRes{
	}
)

service hello{
	@handler Hello
	post /hello (HelloReq) returns (HelloRes)
}

生成代码
goctl api go -api hello.api -dir ./service/hello/rmq/ -style gozero

进入 /service/hello/rmq/hello.go

func main() {
	flag.Parse()

	var c config.Config
	conf.MustLoad(*configFile, &c)

	ctx := svc.NewServiceContext(c)
	server := rest.MustNewServer(c.RestConf)
	defer server.Stop()

	handler.RegisterHandlers(server, ctx)

	// 启动queue服务
	queue(ctx)

	fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
	server.Start()
}

func queue(svc *svc.ServiceContext) {
	// 启动queue服务
	l := logic.NewHelloLogic(context.Background(), svc)
	go func() {
		_, _ = l.Hello(&types.HelloReq{})
	}()
}

进入 /service/hello/rmq/internal/logic/hello.go

func (l *NewHelloLogic) Hello(req *types.HelloReq) (resp *types.HelloRes, err error) {
	// todo: add your logic here and delete this line

	c, _ := xrocketmq.NewPushConsumer("test")

	err = c.Subscribe("topic_test", consumer.MessageSelector{}, func(ctx context.Context, msgs ...*primitive.MessageExt) (consumer.ConsumeResult, error) {
		for i := range msgs {
			fmt.Printf("Hello subscribe callback: %v \n", msgs[i])
		}
		return consumer.ConsumeSuccess, nil
	})
	if err != nil {
		logx.Error(err)
	}

	// 开启消费
	err = c.Start()
	if err != nil {
		logx.Error(err)
	}
	
	time.Sleep(time.Hour)

	err = c.Shutdown()
	if err != nil {
		logx.Error(err)
	}

	return
}

运行测试
cd service\hello\rmq
go run hello.go

测试发送消息

其他代码任意地方，或新建main.go 进行测试都行

// 测试发送延迟队列
	// 定时 10s
	err := xrocketmq.NewProducer("test", "topic_test", "测试发送 hello", 3)
	if err != nil {
		fmt.Println(err)
	}

接收到咯~~~~