This guide describes how developers can write new connectors for Pulsar IO to move data between Pulsar and other systems. It describes how to create a Pulsar IO connector.

Pulsar IO connectors are specialized Pulsar Functions. So writing a Pulsar IO connector is as simple as writing a Pulsar function. Pulsar IO connectors come in two flavors: Source , which import data from another system, and Sink , which export data to another system. For example, KinesisSink would export the messages of a Pulsar topic to a Kinesis stream, and RabbitmqSource would import the messages of a RabbitMQ queue to a Pulsar topic.

Developing

Develop a source connector

What you need to develop a source connector is to implement Source interface.

First, you need to implement the open method. This method will be called once when the source connector is initialized. In this method, you can retrieve all the connector specific settings through the passed config parameter, and initialize all the necessary resourcess. For example, a Kafka connector can create the Kafka client in this open method.

Beside the passed-in config object, the Pulsar runtime also provides a SourceContext for the connector to access runtime resources for tasks like collecting metrics. The implementation can save the SourceContext for futher usage.

  1.     /**
  2.      * Open connector with configuration
  3.      *
  4.      * @param config initialization config
  5.      * @param sourceContext
  6.      * @throws Exception IO type exceptions when opening a connector
  7.      */
  8.     void open(final Map<String, Object> config, SourceContext sourceContext) throws Exception;

The main task for a Source implementor is to implement read method.

  1.     /**
  2.      * Reads the next message from source.
  3.      * If source does not have any new messages, this call should block.
  4.      * @return next message from source.  The return result should never be null
  5.      * @throws Exception
  6.      */
  7.     Record<T> read() throws Exception;

The implementation should be blocking on this method if nothing to return. It should never return null. The returned Record should encapsulates the information that is needed by Pulsar IO runtime.

These information includes:

  • Topic Name: Optional. If the record is originated from a Pulsar topic, it should be the Pulsar topic name.
  • Key: Optional. If the record has a key associated with it.
  • Value: Required. The actual data of this record.
  • Partition Id: Optional. If the record is originated from a partitioned source, return its partition id. The partition id will be used as part of the unique identifier by Pulsar IO runtime to do message deduplication and achieve exactly-once processing guarantee.
  • Record Sequence: Optional. If the record is originated from a sequential source, return its record sequence. The record sequence will be used as part of the unique identifier by Pulsar IO runtime to do message deduplication and achieve exactly-once processing guarantee.
  • Properties: Optional. If the record carries user-defined properties, return those properties.

Additionally, the implemention of the record should provide two methods: ack and fail. These two methods will be used by Pulsar IO connector to acknowledge the records that it has done processing and fail the records that it has failed to process.

KafkaSource is a good example to follow.

Develop a sink connector

Developing a sink connector is as easy as developing a source connector. You just need to implement Sink interface.

Similarly, you first need to implement the open method to initialize all the necessary resources before implementing the write method.

  1.     /**
  2.      * Open connector with configuration
  3.      *
  4.      * @param config initialization config
  5.      * @param sinkContext
  6.      * @throws Exception IO type exceptions when opening a connector
  7.      */
  8.     void open(final Map<String, Object> config, SinkContext sinkContext) throws Exception;

The main task for a Sink implementor is to implement write method.

  1.     /**
  2.      * Write a message to Sink
  3.      * @param inputRecordContext Context of input record from the source
  4.      * @param record record to write to sink
  5.      * @throws Exception
  6.      */
  7.     void write(Record<T> record) throws Exception;

In the implemention of write method, the implementor can decide how to write the value and the optional key to the actual source, and leverage all the provided information such as Partition Id, Record Sequence for achieving different processing guarantees. The implementor is also responsible for acknowledging records if it has successfully written them or failing records if has failed to write them.

Testing

Testing connectors can be challenging because Pulsar IO connectors interact with two systems that may be difficult to mock - Pulsar and the system the connector is connecting to. It is recommended to write very specificially test the functionalities of the connector classes while mocking the external services.

Once you have written sufficient unit tests for your connector, we also recommend adding separate integration tests to verify end-to-end functionality. In Pulsar, we are using testcontainers for all Pulsar integration tests. Pulsar IO IntegrationTests are good examples to follow on integration testing your connectors.

Packaging

Once you’ve developed and tested your connector, you must package it so that it can be submitted to a Pulsar Functions cluster. There are two approaches described here work with Pulsar Functions’ runtime.

If you plan to package and distribute your connector for others to use, you are obligated to properly license and copyright your own code and to adhere to the licensing and copyrights of all libraries your code uses and that you include in your distribution. If you are using the approach described in “Creating a NAR package”, the NAR plugin will automatically create a DEPENDENCIES file in the generated NAR package, including the proper licensing and copyrights of all libraries of your connector.

Creating a NAR package

The easiest approach to packaging a Pulsar IO connector is to create a NAR package using nifi-nar-maven-plugin.

NAR stands for NiFi Archive. It is a custom packaging mechanism used by Apache NiFi, to provide a bit of Java ClassLoader isolation. For more details, you can read this blog post to understand how NAR works. Pulsar uses the same mechanism for packaging all the builtin connectors.

All what you need is to include this nifi-nar-maven-plugin in your maven project for your connector. 例如:

  1. <plugins>
  2.   <plugin>
  3.     <groupId>org.apache.nifi</groupId>
  4.     <artifactId>nifi-nar-maven-plugin</artifactId>
  5.     <version>1.2.0</version>
  6.   </plugin>
  7. </plugins>

The TwitterFirehose connector is a good example to follow.

Creating an Uber JAR

An alternative approach is to create an uber JAR that contains all of the connector’s JAR files and other resource files. No directory internal structure is necessary.

You can use maven-shade-plugin to create a Uber JAR. 例如:

  1. <plugin>
  2.   <groupId>org.apache.maven.plugins</groupId>
  3.   <artifactId>maven-shade-plugin</artifactId>
  4.   <version>3.1.1</version>
  5.   <executions>
  6.     <execution>
  7.       <phase>package</phase>
  8.       <goals>
  9.         <goal>shade</goal>
  10.       </goals>
  11.       <configuration>
  12.         <filters>
  13.           <filter>
  14.             <artifact>*:*</artifact>
  15.           </filter>
  16.         </filters>
  17.       </configuration>
  18.     </execution>
  19.   </executions>
  20. </plugin>