mirror of
https://github.com/datahub-project/datahub.git
synced 2025-10-31 02:37:05 +00:00
65 lines
2.4 KiB
Java
65 lines
2.4 KiB
Java
/**
|
|
* Copyright 2015 LinkedIn Corp. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
*/
|
|
package models.kafka;
|
|
|
|
import java.util.Map;
|
|
import java.util.regex.Matcher;
|
|
import java.util.regex.Pattern;
|
|
|
|
import models.daos.DatasetInfoDao;
|
|
import org.apache.avro.generic.GenericData;
|
|
import wherehows.common.schemas.Record;
|
|
import wherehows.common.utils.StringUtil;
|
|
|
|
|
|
public class GobblinTrackingAuditProcessor extends KafkaConsumerProcessor {
|
|
|
|
final private static String DALI_LIMITED_RETENTION_AUDITOR = "DaliLimitedRetentionAuditor";
|
|
final private static String DALI_AUTOPURGED_AUDITOR = "DaliAutoPurgeAuditor";
|
|
final private static String DS_IGNORE_IDPC_AUDITOR = "DsIgnoreIDPCAuditor";
|
|
final private static String DATASET_URN_PREFIX = "hdfs://";
|
|
final private static String DATASET_OWNER_SOURCE = "IDPC";
|
|
|
|
/**
|
|
* Process a Gobblin tracking event audit record
|
|
* @param record
|
|
* @param topic
|
|
* @return null
|
|
* @throws Exception
|
|
*/
|
|
public Record process(GenericData.Record record, String topic)
|
|
throws Exception {
|
|
|
|
if (record != null && record.get("name") != null) {
|
|
final String name = record.get("name").toString();
|
|
// only handle "DaliLimitedRetentionAuditor","DaliAutoPurgeAuditor" and "DsIgnoreIDPCAuditor"
|
|
if (name.equals(DALI_LIMITED_RETENTION_AUDITOR)
|
|
|| name.equals(DALI_AUTOPURGED_AUDITOR)
|
|
|| name.equals(DS_IGNORE_IDPC_AUDITOR)) {
|
|
Long timestamp = (Long) record.get("timestamp");
|
|
Map<String, String> metadata = StringUtil.convertObjectMapToStringMap(record.get("metadata"));
|
|
|
|
String hasError = metadata.get("HasError");
|
|
if (!hasError.equalsIgnoreCase("true")) {
|
|
String datasetPath = metadata.get("DatasetPath");
|
|
String datasetUrn = DATASET_URN_PREFIX + (datasetPath.startsWith("/") ? "" : "/") + datasetPath;
|
|
String ownerUrns = metadata.get("OwnerURNs");
|
|
DatasetInfoDao.updateKafkaDatasetOwner(datasetUrn, ownerUrns, DATASET_OWNER_SOURCE, timestamp);
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
}
|