From 146a7b285c70c2f3ee5040fa680dd571dd8898ff Mon Sep 17 00:00:00 2001
From: Mohit Yadav <105265192+mohityadav766@users.noreply.github.com>
Date: Tue, 6 Jun 2023 00:22:33 +0530
Subject: [PATCH] Fix Update Issue in MetadataService in case the ES config is
 updated (#11885)

---
 .../service/jdbi3/MetadataServiceRepository.java            | 3 +++
 .../services/metadata/MetadataServiceResource.java          | 6 +++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/MetadataServiceRepository.java b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/MetadataServiceRepository.java
index 59ae3217c36..1a9be734b6b 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/MetadataServiceRepository.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/MetadataServiceRepository.java
@@ -7,6 +7,8 @@ import org.openmetadata.service.Entity;
 import org.openmetadata.service.resources.services.database.DatabaseServiceResource;
 
 public class MetadataServiceRepository extends ServiceEntityRepository<MetadataService, MetadataConnection> {
+  private static final String UPDATE_FIELDS = "owner,tags,connection";
+
   public MetadataServiceRepository(CollectionDAO dao) {
     super(
         DatabaseServiceResource.COLLECTION_PATH,
@@ -14,6 +16,7 @@ public class MetadataServiceRepository extends ServiceEntityRepository<MetadataS
         dao,
         dao.metadataServiceDAO(),
         MetadataConnection.class,
+        UPDATE_FIELDS,
         ServiceType.METADATA);
   }
 }
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/resources/services/metadata/MetadataServiceResource.java b/openmetadata-service/src/main/java/org/openmetadata/service/resources/services/metadata/MetadataServiceResource.java
index f2b59f8edb9..14693de73c0 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/resources/services/metadata/MetadataServiceResource.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/resources/services/metadata/MetadataServiceResource.java
@@ -1,5 +1,7 @@
 package org.openmetadata.service.resources.services.metadata;
 
+import static org.openmetadata.service.Entity.ADMIN_USER_NAME;
+
 import io.swagger.v3.oas.annotations.ExternalDocumentation;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.Parameter;
@@ -98,8 +100,10 @@ public class MetadataServiceResource
         if (servicesList.size() == 1) {
           MetadataService service = servicesList.get(0);
           service.setConnection(metadataConnection);
+          service.setUpdatedBy(ADMIN_USER_NAME);
+          service.setUpdatedAt(System.currentTimeMillis());
           dao.setFullyQualifiedName(service);
-          dao.initializeEntity(service);
+          dao.createOrUpdate(null, service);
         } else {
           throw new IOException("Only one Openmetadata Service can be initialized from the Data.");
         }