developers

No menu items for this category

Develop the Ingestion Code

We recommend you to take some time to understand how the Ingestion Framework works by reading this small article.

The main takes for developing a new connector are:

  • To understand that each of our Source Types (Databases, Dashboards, etc) have a Topology attached.
  • To understand that the process flow is implemented as a generator chain, going through each step.

When developing a new database ingestion connector in OpenMetadata, ensure all necessary components are correctly configured. This guide outlines the steps required to define the connector's ingestion capabilities using a service_spec.py file.

The service_spec.py file centralizes the definitions of sources, profilers, lineage, and other ingestion-related components for a connector. This approach helps standardize implementations across connectors, making it easier to manage ingestion workflows.

Add a service_spec.py file within the connector's directory. This file will define the components needed for ingestion, such as metadata sources, lineage sources, profilers, and samplers.

The DefaultDatabaseSpec class simplifies the definition of connectors by bundling the required components. Import the DefaultDatabaseSpec and reference the appropriate classes for your connector.

Customize the ServiceSpec object based on the features of your connector. Below is an example configuration:

Replace the example classes (e.g., BigquerySource, BigqueryLineageSource, etc.) with those specific to your connector. Depending on the connector's features, you may include or exclude certain components like usage or profiling.

  • metadata_source_class: Defines the class for metadata ingestion.
  • lineage_source_class: Defines the class for lineage extraction.
  • usage_source_class: Tracks data usage patterns.
  • profiler_class: Profiles data for quality and insights.
  • sampler_class: Samples data for efficient ingestion.

Place the file in the connector’s directory.

Define the ServiceSpec using the required classes, adjusting for your connector’s capabilities.

Run the ingestion workflow to test the connector and ensure all components are functioning correctly.

The Topology defines a series of Nodes and Stages that get executed in a hierarchical way and describe how we extract the needed data from the sources.

Starting from the Root node we process the entities in a depth first approach, following the topology tree through the node's children.

From the Service Topology you can understand what methods you need to implement:

  • producer: Methods that will fetch the entities we need to process
  • processor: Methods that will yield a given Entity
  • post_process: Methods that will yield a given Entity but are ran after all entities from that node were processed.

Can be found in ingestion/src/metadata/ingestion/source/database/database_service.py

OpenMetadata 1.6.0 or later

Starting from 1.6.0 the OpenMetadata Ingestion Framewotk is using a ServiceSpec specificaiton in order to define the entrypoints for the ingestion process.

Now that you understand how the Ingestion Process works, you need to understand the Service Source.

A Service Source is an abstract class that is the base for any Connector from that Source Type. They tend to have a lot of methods and are pretty overwhelming at first glance but you don't need to worry. You'll need to check which abstract methods you need to implement in your connector.

Can be found in ingestion/src/metadata/ingestion/source/database/database_service.py

With the Code ready to go, we can now proceed to make a small change in the UI to be able to configure the Connector properly from there.

Apply the UI Changes

Learn what you need to do to be able see the Connector properly in the UI