From 98ef90d0cb7c877cdcc982b80f297796a7c34893 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Wed, 24 Jul 2024 18:47:26 +0530
Subject: [PATCH] MINOR: Fix Trino Azure config sceret not hidden (#17158)

---
 .../secrets/converter/MysqlConnectionClassConverter.java       | 3 ++-
 .../secrets/converter/PostgresConnectionClassConverter.java    | 3 ++-
 .../secrets/converter/TrinoConnectionClassConverter.java       | 3 ++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/MysqlConnectionClassConverter.java b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/MysqlConnectionClassConverter.java
index 9f792f4f86a2..9f4a21a14752 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/MysqlConnectionClassConverter.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/MysqlConnectionClassConverter.java
@@ -16,6 +16,7 @@
 import java.util.List;
 import org.openmetadata.schema.security.ssl.ValidateSSLClientConfig;
 import org.openmetadata.schema.services.connections.database.MysqlConnection;
+import org.openmetadata.schema.services.connections.database.common.AzureConfig;
 import org.openmetadata.schema.services.connections.database.common.IamAuthConfig;
 import org.openmetadata.schema.services.connections.database.common.basicAuth;
 import org.openmetadata.service.util.JsonUtils;
@@ -26,7 +27,7 @@
 public class MysqlConnectionClassConverter extends ClassConverter {
 
   private static final List<Class<?>> CONFIG_SOURCE_CLASSES =
-      List.of(basicAuth.class, IamAuthConfig.class);
+      List.of(basicAuth.class, IamAuthConfig.class, AzureConfig.class);
 
   private static final List<Class<?>> SSL_SOURCE_CLASS = List.of(ValidateSSLClientConfig.class);
 
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/PostgresConnectionClassConverter.java b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/PostgresConnectionClassConverter.java
index 43760c13965c..588123639795 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/PostgresConnectionClassConverter.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/PostgresConnectionClassConverter.java
@@ -16,6 +16,7 @@
 import java.util.List;
 import org.openmetadata.schema.security.ssl.ValidateSSLClientConfig;
 import org.openmetadata.schema.services.connections.database.PostgresConnection;
+import org.openmetadata.schema.services.connections.database.common.AzureConfig;
 import org.openmetadata.schema.services.connections.database.common.IamAuthConfig;
 import org.openmetadata.schema.services.connections.database.common.basicAuth;
 import org.openmetadata.service.util.JsonUtils;
@@ -28,7 +29,7 @@ public class PostgresConnectionClassConverter 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);
+      List.of(basicAuth.class, IamAuthConfig.class, AzureConfig.class);
 
   public PostgresConnectionClassConverter() {
     super(PostgresConnection.class);
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/TrinoConnectionClassConverter.java b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/TrinoConnectionClassConverter.java
index e8d35ed54605..cb07e4df21cf 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/TrinoConnectionClassConverter.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/secrets/converter/TrinoConnectionClassConverter.java
@@ -15,6 +15,7 @@
 
 import java.util.List;
 import org.openmetadata.schema.services.connections.database.TrinoConnection;
+import org.openmetadata.schema.services.connections.database.common.AzureConfig;
 import org.openmetadata.schema.services.connections.database.common.basicAuth;
 import org.openmetadata.schema.services.connections.database.common.jwtAuth;
 import org.openmetadata.service.util.JsonUtils;
@@ -23,7 +24,7 @@
 public class TrinoConnectionClassConverter extends ClassConverter {
 
   private static final List<Class<?>> CONFIG_SOURCE_CLASSES =
-      List.of(basicAuth.class, jwtAuth.class);
+      List.of(basicAuth.class, jwtAuth.class, AzureConfig.class);
 
   public TrinoConnectionClassConverter() {
     super(TrinoConnection.class);