Skip to content

Commit 53e83c6

Browse files
committed
feat: Add Kafka topic support
1 parent 90b010d commit 53e83c6

3 files changed

Lines changed: 10 additions & 0 deletions

File tree

rust/resource-info-fetcher/src/api.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,11 @@ pub enum ResourceInfoRequestResource {
4848
SupersetDashboard {
4949
id: u64,
5050
},
51+
52+
KafkaTopic {
53+
topic: String,
54+
},
55+
5156
RawDataHubUrn(String),
5257
}
5358

rust/resource-info-fetcher/src/backend/data_hub/mod.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,7 @@ pub struct ResolvedDataHubBackend {
110110
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize)]
111111
#[serde(rename_all = "camelCase")]
112112
pub struct DataHubResourceInfoResponse {
113+
urn: Urn,
113114
owners: BTreeMap<OwnerType, DataHubResourceInfoResponseOwners>,
114115
tags: Vec<Tag>,
115116
}
@@ -271,6 +272,7 @@ impl ResourceInfoBackend for ResolvedDataHubBackend {
271272
let (tags, owners) = futures::try_join!(tags_fut, owners_fut)?;
272273

273274
Ok(DataHubResourceInfoResponse {
275+
urn,
274276
tags,
275277
owners: owners.into_iter().collect(),
276278
})

rust/resource-info-fetcher/src/backend/data_hub/resource_to_urn_mapping.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,9 @@ pub fn urn_for_request(request: &ResourceInfoRequest, env: &str) -> Urn {
3939
ResourceInfoRequestResource::SupersetDashboard { id } => {
4040
format!("urn:li:dashboard:({stacklet},{id})")
4141
}
42+
ResourceInfoRequestResource::KafkaTopic { topic } => {
43+
format!("urn:li:dataset:(urn:li:dataPlatform:{stacklet},{topic},{env})")
44+
}
4245
ResourceInfoRequestResource::RawDataHubUrn(urn) => urn.to_owned(),
4346
};
4447

0 commit comments

Comments
 (0)