You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@eventmesh.apache.org by ch...@apache.org on 2022/10/23 11:36:16 UTC
[incubator-eventmesh] branch rabbitmq-connector updated: add rabbitmq producer
This is an automated email from the ASF dual-hosted git repository.
chenguangsheng pushed a commit to branch rabbitmq-connector
in repository https://gitbox.apache.org/repos/asf/incubator-eventmesh.git
The following commit(s) were added to refs/heads/rabbitmq-connector by this push:
new 49923650 add rabbitmq producer
new d218c4a0 Merge pull request #1765 from mroccyen/rabbitmq-producer-operation
49923650 is described below
commit 499236501000b61709f1967ad3b93ea1c9134c00
Author: mroccyen <qi...@126.com>
AuthorDate: Sun Oct 23 15:08:23 2022 +0800
add rabbitmq producer
---
.../rabbitmq/producer/RabbitmqProducer.java | 161 +++++++++++++++++++++
1 file changed, 161 insertions(+)
diff --git a/eventmesh-connector-plugin/eventmesh-connector-rabbitmq/src/main/java/org/apache/eventmesh/connector/rabbitmq/producer/RabbitmqProducer.java b/eventmesh-connector-plugin/eventmesh-connector-rabbitmq/src/main/java/org/apache/eventmesh/connector/rabbitmq/producer/RabbitmqProducer.java
new file mode 100644
index 00000000..700c4479
--- /dev/null
+++ b/eventmesh-connector-plugin/eventmesh-connector-rabbitmq/src/main/java/org/apache/eventmesh/connector/rabbitmq/producer/RabbitmqProducer.java
@@ -0,0 +1,161 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.eventmesh.connector.rabbitmq.producer;
+
+import org.apache.eventmesh.api.RequestReplyCallback;
+import org.apache.eventmesh.api.SendCallback;
+import org.apache.eventmesh.api.SendResult;
+import org.apache.eventmesh.api.exception.ConnectorRuntimeException;
+import org.apache.eventmesh.api.exception.OnExceptionContext;
+import org.apache.eventmesh.api.producer.Producer;
+import org.apache.eventmesh.connector.rabbitmq.client.RabbitmqClient;
+import org.apache.eventmesh.connector.rabbitmq.client.RabbitmqConnectionFactory;
+import org.apache.eventmesh.connector.rabbitmq.cloudevent.RabbitmqCloudEvent;
+import org.apache.eventmesh.connector.rabbitmq.cloudevent.RabbitmqCloudEventWriter;
+import org.apache.eventmesh.connector.rabbitmq.config.ConfigurationHolder;
+import org.apache.eventmesh.connector.rabbitmq.utils.ByteArrayUtils;
+
+import java.util.Optional;
+import java.util.Properties;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import io.cloudevents.CloudEvent;
+
+import com.rabbitmq.client.Channel;
+import com.rabbitmq.client.Connection;
+
+public class RabbitmqProducer implements Producer {
+
+ private static final Logger logger = LoggerFactory.getLogger(RabbitmqProducer.class);
+
+ private RabbitmqConnectionFactory rabbitmqConnectionFactory = new RabbitmqConnectionFactory();
+
+ private RabbitmqClient rabbitmqClient;
+
+ private Connection connection;
+
+ private Channel channel;
+
+ private volatile boolean started = false;
+
+ private final ConfigurationHolder configurationHolder = new ConfigurationHolder();
+
+ @Override
+ public boolean isStarted() {
+ return started;
+ }
+
+ @Override
+ public boolean isClosed() {
+ return !isStarted();
+ }
+
+ @Override
+ public void start() {
+ if (!started) {
+ started = true;
+ }
+ }
+
+ @Override
+ public void shutdown() {
+ if (started) {
+ try {
+ rabbitmqClient.closeConnection(connection);
+ rabbitmqClient.closeChannel(channel);
+ } finally {
+ started = false;
+ }
+ }
+ }
+
+ @Override
+ public void init(Properties properties) throws Exception {
+ this.configurationHolder.init();
+ this.rabbitmqClient = new RabbitmqClient(rabbitmqConnectionFactory);
+ this.connection = rabbitmqClient.getConnection(configurationHolder.getHost(), configurationHolder.getUsername(),
+ configurationHolder.getPasswd(), configurationHolder.getPort(), configurationHolder.getVirtualHost());
+ this.channel = rabbitmqConnectionFactory.createChannel(connection);
+ }
+
+ @Override
+ public void publish(CloudEvent cloudEvent, SendCallback sendCallback) throws Exception {
+ try {
+ RabbitmqCloudEventWriter writer = new RabbitmqCloudEventWriter();
+ RabbitmqCloudEvent rabbitmqCloudEvent = writer.writeBinary(cloudEvent);
+ byte[] data = RabbitmqCloudEvent.toByteArray(rabbitmqCloudEvent);
+ if (data != null) {
+ rabbitmqClient.publish(channel, configurationHolder.getExchangeName(), configurationHolder.getRoutingKey(), data);
+
+ SendResult sendResult = new SendResult();
+ sendResult.setTopic(cloudEvent.getSubject());
+ sendResult.setMessageId(cloudEvent.getId());
+ sendCallback.onSuccess(sendResult);
+ }
+ } catch (Exception ex) {
+ logger.error("[RabbitmqProducer] publish happen exception.", ex);
+ sendCallback.onException(
+ OnExceptionContext.builder()
+ .topic(cloudEvent.getSubject())
+ .messageId(cloudEvent.getId())
+ .exception(new ConnectorRuntimeException(ex))
+ .build()
+ );
+ }
+ }
+
+ @Override
+ public void sendOneway(CloudEvent cloudEvent) {
+ try {
+ Optional<byte[]> optionalBytes = ByteArrayUtils.objectToBytes(cloudEvent);
+ if (optionalBytes.isPresent()) {
+ byte[] data = optionalBytes.get();
+ rabbitmqClient.publish(channel, configurationHolder.getExchangeName(),
+ configurationHolder.getRoutingKey(), data);
+ }
+ } catch (Exception ex) {
+ logger.error("[RabbitmqProducer] sendOneway happen exception.", ex);
+ }
+ }
+
+ @Override
+ public void request(CloudEvent cloudEvent, RequestReplyCallback rrCallback, long timeout) throws Exception {
+
+ }
+
+ @Override
+ public boolean reply(CloudEvent cloudEvent, SendCallback sendCallback) throws Exception {
+ return false;
+ }
+
+ @Override
+ public void checkTopicExist(String topic) throws Exception {
+
+ }
+
+ @Override
+ public void setExtFields() {
+
+ }
+
+ public void setRabbitmqConnectionFactory(RabbitmqConnectionFactory rabbitmqConnectionFactory) {
+ this.rabbitmqConnectionFactory = rabbitmqConnectionFactory;
+ }
+}
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@eventmesh.apache.org
For additional commands, e-mail: commits-help@eventmesh.apache.org