import com.rabbitmq.client.AMQP.BasicProperties; import org.apache.flink.api.common.serialization.SimpleStringSchema; import org.apache.flink.streaming.api.datastream.DataStreamSource; import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment; import org.apache.flink.streaming.connectors.kafka.FlinkKafkaConsumer; import org.apache.flink.streaming.connectors.rabbitmq.RMQSink; import org.apache.flink.streaming.connectors.rabbitmq.RMQSinkPublishOptions; import org.apache.flink.streaming.connectors.rabbitmq.common.RMQConnectionConfig; import org.apache.kafka.clients.consumer.ConsumerConfig; import java.util.Properties; public class KafkaToMqExchange { /** * args[0]: kafka server address * args[1]: kafka consumer group * args[2]: kafka topic * args[3]: RabbitMQ address * args[4]: RabbitMQ username * args[5]: RabbitMQ password * args[6]: RabbitMQ exchange * args[7]: RabbitMQ routing key * * @param args * @throws Exception */ public static void main(String[] args) throws Exception { StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment(); Properties properties = new Properties(); properties.put(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG, args[0] + ":9092"); properties.put(ConsumerConfig.GROUP_ID_CONFIG, args[1]); DataStreamSource<String> data = env.addSource(new FlinkKafkaConsumer<String>(args[2], new SimpleStringSchema(), properties)); RMQConnectionConfig connectionConfig = new RMQConnectionConfig .Builder().setHost(args[3]).setVirtualHost("/") .setPort(5672).setUserName(args[4]).setPassword(args[5]) .build(); data.addSink(new RMQSink<>(connectionConfig, new SimpleStringSchema(), new RMQSinkPublishOptions<String>() { @Override public String computeRoutingKey(String var1) { return args[7]; } @Override public String computeExchange(String var1) { return args[6]; } @Override public BasicProperties computeProperties(String var1) { return null; } })); env.enableCheckpointing(10000); env.execute("flink connectors rabbitmq exchange"); } }
<dependencies> <dependency> <groupId>org.apache.flink</groupId> <artifactId>flink-connector-rabbitmq_${scala.version}</artifactId> <version>${flink.version}</version> </dependency> <dependency> <groupId>org.apache.flink</groupId> <artifactId>flink-java</artifactId> <version>${flink.version}</version> <scope>provided</scope> </dependency> <dependency> <groupId>org.apache.flink</groupId> <artifactId>flink-streaming-java_${scala.version}</artifactId> <version>${flink.version}</version> <scope>provided</scope> </dependency> <dependency> <groupId>org.apache.flink</groupId> <artifactId>flink-connector-kafka_2.11</artifactId> <version>${flink.version}</version> </dependency> <dependency> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId> <version>${lombok.version}</version> </dependency> </dependencies> <build> <plugins> <plugin> <artifactId>maven-compiler-plugin</artifactId> <version>2.3.2</version> <configuration> <source>1.8</source> <target>1.8</target> </configuration> </plugin> <plugin> <artifactId>maven-assembly-plugin</artifactId> <configuration> <descriptorRefs> <descriptorRef>jar-with-dependencies</descriptorRef> </descriptorRefs> <archive> <manifest> <mainClass>TestFlinkToMq</mainClass> </manifest> </archive> </configuration> <executions> <execution> <id>make-assembly</id> <phase>package</phase> <goals> <goal>single</goal> </goals> </execution> </executions> </plugin> <plugin> <groupId>org.apache.maven.plugins</groupId> <artifactId>maven-shade-plugin</artifactId> <version>2.3</version> <executions> <execution> <phase>package</phase> <goals> <goal>shade</goal> </goals> <configuration> <transformers> <transformer implementation="org.apache.maven.plugins.shade.resource.ManifestResourceTransformer"> <mainClass>com.dahua.KafkaToMq</mainClass> </transformer> <transformer implementation="org.apache.maven.plugins.shade.resource.AppendingTransformer"> <resource>reference.conf</resource> </transformer> </transformers> </configuration> </execution> </executions> </plugin> </plugins> </build>