Spring Cloud Stream rocketmq 函数式编程实现

1.maven配置

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.6.4</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>

    <groupId>com.example</groupId>
    <artifactId>demo</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <packaging>jar</packaging>
    <properties>
        <java.version>11</java.version>
        <spring-cloud.version>2021.0.1</spring-cloud.version>
        <spring-cloud-alibaba.version>2021.1</spring-cloud-alibaba.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-stream</artifactId>
        </dependency>
        <dependency>
            <groupId>com.alibaba.cloud</groupId>
            <artifactId>spring-cloud-starter-stream-rocketmq</artifactId>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
    </dependencies>

    <dependencyManagement>
        <dependencies>
            <dependency>
                <groupId>org.springframework.cloud</groupId>
                <artifactId>spring-cloud-dependencies</artifactId>
                <version>${spring-cloud.version}</version>
                <type>pom</type>
                <scope>import</scope>
            </dependency>
            <dependency>
                <groupId>com.alibaba.cloud</groupId>
                <artifactId>spring-cloud-alibaba-dependencies</artifactId>
                <version>${spring-cloud-alibaba.version}</version>
                <type>pom</type>
                <scope>import</scope>
            </dependency>
        </dependencies>

    </dependencyManagement>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>
</project>

2.yml配置

spring:
  cloud:
    function:
      definition: example;demo1;test1
    stream:
      rocketmq:
        binder:
          name-server: 127.0.0.1:9876
          vipChannelEnabled: false
        bindings:
          demo1-in-0:
            consumer:
              tags: s1||s2  #消费tag
              broadcasting: false # 是否使用广播消费，默认为 false 使用集群消费
          demo1-out-0:
            producer:
              #开启同步发送
              sync: true
          example-in-0:
            consumer:
              tags:   #消费tag
              broadcasting: true # 是否使用广播消费，默认为 false 使用集群消费
          example-out-0:
            producer:
              sync: false #默认异步
          test1-in-0:
            consumer:
              tags:   #消费tag
              broadcasting: false # 是否使用广播消费，默认为 false 使用集群消费
          test1-out-0:
            producer:
              sync: true

      bindings:
        #消费者
        demo1-in-0:
          destination: dtopic #topic
          content-type: application/json
          group: demo1-group #group
        demo1-out-0:
          destination: dtopic #topic
          group: pd-group
        example-in-0:
          destination: extopic #topic
          content-type: application/json
          group: example-group #group
        example-out-0:
          destination: extopic #topic
          group: px-group
        test1-in-0:
          destination: ttopic #topic
          content-type: application/json
          group: t-group #group
        test1-out-0:
          destination: ttopic #topic
          group: pt-group #group

消费者和生产者的使用方式为：名称-out-数字的方式用于生产者，名称-in-数字的方式用于消费者。

3.发送消息

package com.example.streamrockmq.controller;

import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.common.message.MessageConst;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cloud.stream.function.StreamBridge;
import org.springframework.integration.support.MessageBuilder;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHeaders;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;
import java.util.Map;

@Slf4j
@RestController
@RequestMapping("/t")
public class TestController {

    @Autowired
    StreamBridge streamBridge;

    @GetMapping("t1")
    public void run1() {
        boolean s1 = streamBridge.send("example-out-0", "topic example");
        log.info("example send msg:{}", s1);
    }

    @GetMapping("/t2")
    public void run2() throws Exception {
        //tag 发送
        String payload = "消息体demo1发送到tag s1";
        Map<String, Object> headers = new HashMap<>();
        headers.put(MessageConst.PROPERTY_TAGS, "s1");
        MessageHeaders messageHeaders = new MessageHeaders(headers);
        Message<String> message = MessageBuilder.withPayload(payload).copyHeadersIfAbsent(messageHeaders).build();
        boolean s3 = streamBridge.send("demo1-out-0", message);
        log.info("demo1 send msg:{}", s3);

    }

    @GetMapping("/t3")
    public void run3() throws Exception {
        String payload = "消息体tag s3 延迟消息 topic test";
        Map<String, Object> headers = new HashMap<>();
        //延迟消费 延迟10秒
        headers.put(MessageConst.PROPERTY_DELAY_TIME_LEVEL, "3");
        MessageHeaders messageHeaders = new MessageHeaders(headers);
        Message<String> message = MessageBuilder.withPayload(payload).copyHeadersIfAbsent(messageHeaders).build();
        boolean s3 = streamBridge.send("test1-out-0", message);
        log.info("test send msg:{}", s3);
    }


}

4.消费

package com.example.streamrockmq.consumer;

import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.spring.support.RocketMQHeaders;
import org.springframework.context.annotation.Bean;
import org.springframework.messaging.Message;
import org.springframework.stereotype.Service;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.util.function.Consumer;
import java.util.function.Function;

/**
 * 三种消费任选
 */
@Slf4j
@Service
public class ConsumerTest {

    @Bean
    public Consumer<Message<String>> demo1() {
        return message -> {
            log.info("demo1获取消息tag:{}", message.getHeaders().get(RocketMQHeaders.PREFIX + RocketMQHeaders.TAGS));
            log.info("demo1接收数据:{}", message.getPayload());
        };
    }


    @Bean
    public Function<Flux<Message<String>>, Mono<Void>> example() {
        return flux -> flux.map(message -> {
            log.info("example接收数据:{}", message.getPayload());
            return message;
        }).then();
    }

    @Bean
    public Consumer<Flux<Message<String>>> test1() {
        return flux -> flux.map(message -> {
            log.info("test接收数据:{}", message.getPayload());
            return message;
        }).subscribe();
    }

}

5.执行生产者和消费者执行结果

 

6.文档可参照Preface