Load Cosmos DB Data to a Database Using Embulk

Embulk is an open source bulk data loader. When paired with the CData JDBC Driver for Cosmos DB, Embulk easily loads data from Cosmos DB to any supported destination. In this article, we explain how to use the CData JDBC Driver for Cosmos DB in Embulk to load Cosmos DB data to a MySQL dtabase.

With built-in optimized data processing, the CData JDBC Driver offers unmatched performance for interacting with live Cosmos DB data. When you issue complex SQL queries to Cosmos DB, the driver pushes supported SQL operations, like filters and aggregations, directly to Cosmos DB and utilizes the embedded SQL engine to process unsupported operations client-side (often SQL functions and JOIN operations).

Configure a JDBC Connection to Cosmos DB Data

Before creating a bulk load job in Embulk, note the installation location for the JAR file for the JDBC Driver (typically C:\Program Files\CData\CData JDBC Driver for Cosmos DB\lib).

Embulk supports JDBC connectivity, so you can easily connect to Cosmos DB and execute SQL queries. Before creating a bulk load job, create a JDBC URL for authenticating with Cosmos DB.

To obtain the connection string needed to connect to a Cosmos DB account using the SQL API, log in to the Azure Portal, select Azure Cosmos DB, and select your account. In the Settings section, click Connection String and set the following values:

• AccountEndpoint: The Cosmos DB account URL from the Keys blade of the Cosmos DB account
• AccountKey: In the Azure portal, navigate to the Cosmos DB service and select your Azure Cosmos DB account. From the resource menu, go to the Keys page. Find the PRIMARY KEY value and set AccountKey to this value.

Built-in Connection String Designer

For assistance in constructing the JDBC URL, use the connection string designer built into the Cosmos DB JDBC Driver. Either double-click the JAR file or execute the jar file from the command-line.

java -jar cdata.jdbc.cosmosdb.jar

Fill in the connection properties and copy the connection string to the clipboard.

👁 Using the built-in connection string designer to generate a JDBC URL (Salesforce is shown.)

Below is a typical JDBC connection string for Cosmos DB:

jdbc:cosmosdb:AccountEndpoint=myAccountEndpoint;AccountKey=myAccountKey;

Load Cosmos DB Data in Embulk

After installing the CData JDBC Driver and creating a JDBC connection string, install the required Embulk plugins.

Install Embulk Input & Output Plugins

1. Install the JDBC Input Plugin in Embulk.
   https://github.com/embulk/embulk-input-jdbc/tree/master/embulk-input-jdbc

embulk gem install embulk-input-jdbc

2. In this article, we use MySQL as the destination database. You can also choose SQL Server, PostgreSQL, or Google BigQuery as the destination using the output Plugins.
   https://github.com/embulk/embulk-output-jdbc/tree/master/embulk-output-mysql

   embulk gem install embulk-output-mysql
   

With the input and output plugins installed, we are ready to load Cosmos DB data into MySQL using Embulk.

Create a Job to Load Cosmos DB Data

Start by creating a config file in Embulk, using a name like cosmosdb-mysql.yml.

1. For the input plugin options, use the CData JDBC Driver for Cosmos DB, including the path to the driver JAR file, the driver class (e.g. cdata.jdbc.cosmosdb.CosmosDBDriver), and the JDBC URL from above
2. For the output plugin options, use the values and credentials for the MySQL database

Sample Config File (cosmosdb-mysql.yml)

in:
	type: jdbc
	driver_path: C:\Program Files\CData[product_name] 20xx\lib\cdata.jdbc.cosmosdb.jar
	driver_class: cdata.jdbc.cosmosdb.CosmosDBDriver
	url: jdbc:cosmosdb:AccountEndpoint=myAccountEndpoint;AccountKey=myAccountKey;
	table: "Customers"
out: 
	type: mysql
	host: localhost
	database: DatabaseName
	user: UserId
	password: UserPassword
	table: "Customers"
	mode: insert

After creating the file, run the Embulk job.

embulk run cosmosdb-mysql.yml

After running the the Embulk job, find the Salesforce data in the MySQL table.

Load Filtered Cosmos DB Data

In addition to loading data directly from a table, you can use a custom SQL query to have more granular control of the data loaded. You can also perform increment loads by setting a last updated column in a SQL WHERE clause in the query field.

in:
	type: jdbc
	driver_path: C:\Program Files\CData[product_name] 20xx\lib\cdata.jdbc.cosmosdb.jar
	driver_class: cdata.jdbc.cosmosdb.CosmosDBDriver
	url: jdbc:cosmosdb:AccountEndpoint=myAccountEndpoint;AccountKey=myAccountKey;
	query: "SELECT City, CompanyName FROM Customers WHERE [RecordId] = 1"
out: 
	type: mysql
	host: localhost
	database: DatabaseName
	user: UserId
	password: UserPassword
	table: "Customers"
	mode: insert

More Information & Free Trial

By using CData JDBC Driver for Cosmos DB as a connector, Embulk can integrate Cosmos DB data into your data load jobs. And with drivers for more than 200+ other enterprise sources, you can integrate any enterprise SaaS, big data, or NoSQL source as well. Download a 30-day free trial and get started today.