mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-24 18:10:11 +00:00
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
|
import logging
|
||
|
|
from typing import Union
|
||
|
|
|
||
|
|
from datahub.configuration.kafka import KafkaProducerConnectionConfig
|
||
|
|
from datahub.emitter.kafka_emitter import DatahubKafkaEmitter, KafkaEmitterConfig
|
||
|
|
from datahub.emitter.mce_builder import make_dataset_urn
|
||
|
|
from datahub.emitter.rest_emitter import DataHubRestEmitter
|
||
|
|
from datahub.specific.dataset import DatasetPatchBuilder
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
|
||
|
|
|
||
|
|
# Get an emitter, either REST or Kafka, this example shows you both
|
||
|
|
def get_emitter() -> Union[DataHubRestEmitter, DatahubKafkaEmitter]:
|
||
|
|
USE_REST_EMITTER = True
|
||
|
|
if USE_REST_EMITTER:
|
||
|
|
gms_endpoint = "http://localhost:8080"
|
||
|
|
return DataHubRestEmitter(gms_server=gms_endpoint)
|
||
|
|
else:
|
||
|
|
kafka_server = "localhost:9092"
|
||
|
|
schema_registry_url = "http://localhost:8081"
|
||
|
|
return DatahubKafkaEmitter(
|
||
|
|
config=KafkaEmitterConfig(
|
||
|
|
connection=KafkaProducerConnectionConfig(
|
||
|
|
bootstrap=kafka_server, schema_registry_url=schema_registry_url
|
||
|
|
)
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
dataset_urn = make_dataset_urn(platform="hive", name="fct_users_created", env="PROD")
|
||
|
|
|
||
|
|
with get_emitter() as emitter:
|
||
|
|
for patch_mcp in (
|
||
|
|
DatasetPatchBuilder(dataset_urn)
|
||
|
|
.add_custom_property("cluster_name", "datahubproject.acryl.io")
|
||
|
|
.remove_custom_property("retention_time")
|
||
|
|
.build()
|
||
|
|
):
|
||
|
|
emitter.emit(patch_mcp)
|
||
|
|
|
||
|
|
|
||
|
|
log.info(
|
||
|
|
f"Added cluster_name property, removed retention_time property from dataset {dataset_urn}"
|
||
|
|
)
|