From 52ea9ea81e9a848ca5f868a64e692e6800a3ff76 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Fri, 4 Oct 2024 16:57:15 +0530
Subject: [PATCH] Fix #18005: Mask Greenplum Password (#18047)

---
 .../converter/GreenplumConnectionClassConverter.java       | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/GreenplumConnectionClassConverter.java b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/GreenplumConnectionClassConverter.java
index aeea95cad0e..34bd1d76422 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/GreenplumConnectionClassConverter.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/GreenplumConnectionClassConverter.java
@@ -16,6 +16,8 @@ package org.openmetadata.service.secrets.converter;
 import java.util.List;
 import org.openmetadata.schema.security.ssl.ValidateSSLClientConfig;
 import org.openmetadata.schema.services.connections.database.GreenplumConnection;
+import org.openmetadata.schema.services.connections.database.common.IamAuthConfig;
+import org.openmetadata.schema.services.connections.database.common.basicAuth;
 import org.openmetadata.service.util.JsonUtils;
 
 /**
@@ -24,6 +26,8 @@ import org.openmetadata.service.util.JsonUtils;
 public class GreenplumConnectionClassConverter extends ClassConverter {
 
   private static final List<Class<?>> SSL_SOURCE_CLASS = List.of(ValidateSSLClientConfig.class);
+  private static final List<Class<?>> CONFIG_SOURCE_CLASSES =
+      List.of(basicAuth.class, IamAuthConfig.class);
 
   public GreenplumConnectionClassConverter() {
     super(GreenplumConnection.class);
@@ -34,6 +38,9 @@ public class GreenplumConnectionClassConverter extends ClassConverter {
     GreenplumConnection greenplumConnection =
         (GreenplumConnection) JsonUtils.convertValue(object, this.clazz);
 
+    tryToConvert(greenplumConnection.getAuthType(), CONFIG_SOURCE_CLASSES)
+        .ifPresent(greenplumConnection::setAuthType);
+
     tryToConvert(greenplumConnection.getSslConfig(), SSL_SOURCE_CLASS)
         .ifPresent(greenplumConnection::setSslConfig);