

Amundsen Models¶

Overview¶

These are the python classes that live in databuilder/models/.

Models represent the data structures that live in either neo4j (if the model extends Neo4jSerializable) or in elasticsearch.

Models that extend Neo4jSerializable have methods to create: - the nodes - the relationships

In this way, amundsendatabuilder pipelines can create python objects that can then be loaded into neo4j / elastic search without developers needing to know the internals of the neo4j schema.

The Models¶

TableMetadata ¶

What datasets does my org have?

Description¶

This corresponds to a dataset in amundsen and is the core building block. In addition to ColumnMetadata, tableMetadata is one of the first datasets you should extract as almost everything else depends on these being populated.

Extraction¶

In general, for Table and Column Metadata, you should be able to use one of the pre-made extractors in the extractor package

Watermark ¶

What is the earliest data that this table has? What is the latest data? This is NOT the same as when the data was last updated.

Description¶

Corresponds to the earliest and latest date that a dataset has. Only makes sense if the dataset is timeseries data. For example, a given table may have data from 2019/01/01 -> 2020/01/01 In that case the low watermark is 2019/01/01 and the high watermark is 2020/01/01.

Extraction¶

Depending on the datastore of your dataset, you would extract this by: - a query on the minimum and maximum partition (hive) - a query for the minimum and maximum record of a given timestamp column

TableColumnUsage ¶

Description¶

How many queries is a given column getting? By which users?

Has query counts per a given table per a user. This can help identify who uses given datasets so people can contact them if they have questions on how to use a given dataset or if a dataset is changing. It is also used as a search boost so that the most used tables are put to the top of the search results.

This model also populates the Popular Resources section on the homepage. A table must have at least 10 unique users to appear on the homepage. This configuration is defined here and can be changed if your usage is less.

Extraction¶

For more traditional databases, there should be system tables where you can obtain these sorts of usage statistics.

In other cases, you may need to use audit logs which could require a custom solution.

Finally, for non-traditional data lakes, getting this information exactly maybe difficult and you may need to rely on a heuristic.

User ¶

What users are there out there? Which team is this user on?

Description¶

Represents all of the metadata for a user at your company. This is required if you are going to be having authentication turned on.

Extraction¶

TODO

TableColumnStats ¶

What are the min/max values for this column? How many nulls are in this column? *

Description¶

This represents statistics on the column level (this is not for table level metrics). The idea is that different companies will want to track different things about different columns, so this is highly customizable. It also will probably require a distributed cluster in order to calculate these regularly and in general is probably the least accessible metrics to get at without a custom solution.

Extraction¶

The idea here would be to implement something that does the following: For each table you care about: For each column you care about: Calculate statistics that you care about such as min/max/average etc.

Application ¶

What job/application is writing to this table? *

Description¶

This is used to provide users a way to find out what job/application is responsible for writing to this dataset.

Extraction¶

TODO

Table Owner ¶

What team or user owns this dataset? *

Description¶

A dataset can have one or more owners. These owners are used when requesting table descriptions or could be just a useful point of contact for a user inquiring about how to use a dataset.

Extraction¶

Although the main point of entry for owners is through the WebUI, you could in theory extract this information based on who created a given table.

Table Source ¶

Where is the source code for the application that writes to this dataset? *

Description¶

Generally there is going to be code that your company owns that describes how a dataset is created. This model is what represents the link and type of repository to this source code so it is available to users.

Extraction¶

You will need a github/gitlab/your repository crawler in order to populate this automatically. The idea there would be to search for a given table name or something else that is a unique identifier such that you can be confident that the source correctly matches to this table.

TableLastUpdated ¶

When was the last time this data was updated? Is this table stale or deprecated? *

Description¶

This value is used to describe the last time the table had datapoints inserted into it. It is a very useful value as it can help users identify if there are tables that are no longer being updated.

Extraction¶

There are some extractors available for this like hive_table_last_updated_extractor that you can refer to. But you will need access to history that provides information on when the last data write happened on a given table. If this data isn’t available for your data source, you maybe able to approximate it by looking at the max of some timestamp column.

Dashboard models¶

Dashboard models are normalized which means that the model is separated so that it can be easily decoupled with how data is extracted. (If model is denormalized, all metadata is in model, then one extraction needs to able to pull all the data which makes extraction hard and complex) There’s trade off in this decision of normalized design where it can be inefficient in the case that some ingestion can be done in one job for metadata source happen to provide all data it need. However, to make model flexible for most of metadata, it is normalized.

DashboardMetadata ¶

Description¶

A baseline of Dashboard metadata that consists of dashboard group name, dashboard group description, dashboard description, etc. This model needs to be ingested first as other model builds relation to this.

Extraction¶

ModeDashboardExtractor

DashboardOwner ¶

Description¶

A model that encapsulate Dashboard’s owner. Note that it does not create new user as it has insufficient information about user but it builds relation between User and Dashboard

Extraction¶

ModeDashboardOwnerExtractor

DashboardTable ¶

A model that link Dashboard with the tables used in various charts of the dashboard. Note that it does not create new dashboard, table as it has insufficient information but it builds relation between Tables and Dashboard.

Supporting extractor: Currently there’s no open sourced extractor for this. In Lyft, there’s audit table that records SQL query, where it came from with identifier, along with tables that is used in SQL query. We basically query this table via DBAPIExtractor

DashboardUsage ¶

Description¶

A model that encapsulate Dashboard usage between Dashboard and User

Extraction¶

You can use ModeDashboardUsageExtractor . However, currently Mode only provides accumulated view count where we need recent view counts (past 30, 60, or 90 days). To get recent view count, in Lyft, we use ModeDashboardUsageExtractor to extract accumulated view count and GenericLoader to load its record (no publisher here and publisher is not mandatory in DefaultJob) as a event where event materialized as daily snapshot. Once it captures daily accumulated view count, ingest recent view count by querying the datastore. In Lyft, we query via DBAPIExtractor through Presto.

DashboardLastModifiedTimestamp ¶

Description¶

A model that encapsulate Dashboard’s last modified timestamp in epoch

Extraction¶

ModeDashboardLastModifiedTimestampExtractor

DashboardExecution ¶

A model that encapsulate Dashboard’s execution timestamp in epoch and execution state. Note that this model supports last_execution and last_successful_execution by using different identifier in the URI.

Extraction¶

ModeDashboardExecutionsExtractor which extracts last_execution.

ModeDashboardLastSuccessfulExecutionExtractor

DashboardQuery ¶

Description¶

A model that encapsulate Dashboard’s query information.

Supporting extractor: ModeDashboardQueriesExtractor

DashboardChart ¶

Description¶

A model that encapsulate Dashboard’s charts where chart is associated with query.

Extraction¶

ModeDashboardChartsExtractor

Feature models¶

Feature models include FeatureMetadata, which encapsulates the basic feature details, and supplemental models Feature_Generation_Code and Feature_Watermark for adding extra metadata. In addition, the Tag, Badge, Owner, and Programmatic_Description models work with features.

FeatureMetadata ¶

Description¶

A baseline of Feature metadata. This model needs to be ingested first as other models build relations to it.

Extraction¶

No specific extractors are provided at this time. We expect users will either write custom extractors, or use generic extractors (e.g. SQLAlchemyExtractor).

Feature_Generation_Code ¶

Description¶

Allows ingesting the text of the generation code (SQL or otherwise) which was used to create a feature.

Extraction¶

No specific extractors are provided at this time. We expect users will either write custom extractors, or use generic extractors (e.g. SQLAlchemyExtractor).

Feature_Watermark ¶

Description¶

Allows ingesting the high and low data range of a feature. Unlike Watermark, which is specific to tables (requires a partition, for example), Feature_Watermark is more general and does not care about how the feature is stored.

Extraction¶

No specific extractors are provided at this time. We expect users will either write custom extractors, or use generic extractors (e.g. SQLAlchemyExtractor).