You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@hudi.apache.org by fo...@apache.org on 2023/01/04 03:32:18 UTC

[hudi] 08/45: fix zhiyan reporter for metadata

This is an automated email from the ASF dual-hosted git repository.

forwardxu pushed a commit to branch release-0.12.1
in repository https://gitbox.apache.org/repos/asf/hudi.git

commit 6dbe53e6232de8b85c7548fefda670d6f4359ec1
Author: XuQianJin-Stars <fo...@apache.com>
AuthorDate: Sun Jun 5 15:06:30 2022 +0800

    fix zhiyan reporter for metadata
---
 .../main/java/org/apache/hudi/config/HoodieWriteConfig.java   |  8 ++++++++
 .../apache/hudi/metadata/HoodieBackedTableMetadataWriter.java | 11 +++++++++++
 2 files changed, 19 insertions(+)

diff --git a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/config/HoodieWriteConfig.java b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/config/HoodieWriteConfig.java
index 23bc0ee329..9610ad382b 100644
--- a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/config/HoodieWriteConfig.java
+++ b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/config/HoodieWriteConfig.java
@@ -2234,6 +2234,8 @@ public class HoodieWriteConfig extends HoodieConfig {
     private boolean isPreCommitValidationConfigSet = false;
     private boolean isMetricsJmxConfigSet = false;
     private boolean isMetricsGraphiteConfigSet = false;
+
+    private boolean isMetricsZhiyanConfig = false;
     private boolean isLayoutConfigSet = false;
     private boolean isTdbankConfigSet = false;
 
@@ -2429,6 +2431,12 @@ public class HoodieWriteConfig extends HoodieConfig {
       return this;
     }
 
+    public Builder withMetricsZhiyanConfig(HoodieMetricsZhiyanConfig metricsZhiyanConfig) {
+      writeConfig.getProps().putAll(metricsZhiyanConfig.getProps());
+      isMetricsZhiyanConfig = true;
+      return this;
+    }
+
     public Builder withPreCommitValidatorConfig(HoodiePreCommitValidatorConfig validatorConfig) {
       writeConfig.getProps().putAll(validatorConfig.getProps());
       isPreCommitValidationConfigSet = true;
diff --git a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/metadata/HoodieBackedTableMetadataWriter.java b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/metadata/HoodieBackedTableMetadataWriter.java
index 962875fb92..405db43a51 100644
--- a/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/metadata/HoodieBackedTableMetadataWriter.java
+++ b/hudi-client/hudi-client-common/src/main/java/org/apache/hudi/metadata/HoodieBackedTableMetadataWriter.java
@@ -65,6 +65,7 @@ import org.apache.hudi.config.HoodieWriteConfig;
 import org.apache.hudi.config.metrics.HoodieMetricsConfig;
 import org.apache.hudi.config.metrics.HoodieMetricsGraphiteConfig;
 import org.apache.hudi.config.metrics.HoodieMetricsJmxConfig;
+import org.apache.hudi.config.metrics.HoodieMetricsZhiyanConfig;
 import org.apache.hudi.exception.HoodieException;
 import org.apache.hudi.exception.HoodieIndexException;
 import org.apache.hudi.exception.HoodieMetadataException;
@@ -316,6 +317,16 @@ public abstract class HoodieBackedTableMetadataWriter implements HoodieTableMeta
               .toJmxHost(writeConfig.getJmxHost())
               .build());
           break;
+        case ZHIYAN:
+          builder.withMetricsZhiyanConfig(HoodieMetricsZhiyanConfig.newBuilder()
+              .withReportServiceUrl(writeConfig.getZhiyanReportServiceURL())
+              .withApiTimeout(writeConfig.getZhiyanApiTimeoutSeconds())
+              .withAppMask(writeConfig.getZhiyanAppMask())
+              .withReportPeriodSeconds(writeConfig.getZhiyanReportPeriodSeconds())
+              .withSeclvlEnvName(writeConfig.getZhiyanSeclvlEnvName())
+              .withJobName(writeConfig.getZhiyanHoodieJobName())
+              .build());
+          break;
         case DATADOG:
         case PROMETHEUS:
         case PROMETHEUS_PUSHGATEWAY: