From 9f2f668912c80d5a68ec0d5e953b584593289eb7 Mon Sep 17 00:00:00 2001
From: Pallavi Kotkar <pallavikotkar17@gmail.com>
Date: Mon, 24 Feb 2025 14:38:47 -0800
Subject: [PATCH] Removing redundant code

---
 .../java/com/linkedin/metadata/dao/BaseLocalDAO.java | 12 ++++--------
 .../com/linkedin/metadata/dao/utils/RecordUtils.java |  9 ---------
 .../com/linkedin/metadata/dao/BaseLocalDAOTest.java  |  2 +-
 .../metadata/dao/utils/EmbeddedMariaInstance.java    |  8 ++++----
 4 files changed, 9 insertions(+), 22 deletions(-)

diff --git a/dao-api/src/main/java/com/linkedin/metadata/dao/BaseLocalDAO.java b/dao-api/src/main/java/com/linkedin/metadata/dao/BaseLocalDAO.java
index 90c682571..cf8f332e5 100644
--- a/dao-api/src/main/java/com/linkedin/metadata/dao/BaseLocalDAO.java
+++ b/dao-api/src/main/java/com/linkedin/metadata/dao/BaseLocalDAO.java
@@ -691,7 +691,7 @@ private <ASPECT extends RecordTemplate> AddResult<ASPECT> aspectUpdateHelper(URN
    * @return the URN of the entity if the operation is successful, otherwise null
    * @param <ASPECT_UNION> must be a valid aspect union type defined in com.linkedin.metadata.aspect
    */
-  <ASPECT_UNION extends RecordTemplate> URN createAspectWithCallbacks(@Nonnull URN urn,
+  <ASPECT_UNION extends RecordTemplate> URN createAspectsWithCallbacks(@Nonnull URN urn,
       @Nonnull List<? extends RecordTemplate> aspectValues,
       @Nonnull List<AspectCreateLambda<? extends RecordTemplate>> aspectCreateLambdas,
       @Nonnull AuditStamp auditStamp, @Nullable IngestionTrackingContext trackingContext) {
@@ -905,16 +905,12 @@ public <ASPECT extends RecordTemplate> URN create(@Nonnull URN urn,
     aspectCreateLambdas.forEach(aspectCreateLambda -> checkValidAspect(aspectCreateLambda.getAspectClass()));
 
     // create aspects and process callbacks in a single transaction
-    final List<AddResult<ASPECT>> results = runInTransactionWithRetry(() -> {
-      List<AddResult<ASPECT>> createdAspects = new ArrayList<>();
-      createAspectWithCallbacks(urn, aspectValues, aspectCreateLambdas, auditStamp, trackingContext);
-      return createdAspects;
+    final URN createdUrn = runInTransactionWithRetry(() -> {
+      return createAspectsWithCallbacks(urn, aspectValues, aspectCreateLambdas, auditStamp, trackingContext);
       }, maxTransactionRetry
     );
 
-    results.stream().map(x -> unwrapAddResultToUnion(urn, x, auditStamp, trackingContext));
-
-    return urn;
+    return createdUrn;
   }
 
   /**
diff --git a/dao-api/src/main/java/com/linkedin/metadata/dao/utils/RecordUtils.java b/dao-api/src/main/java/com/linkedin/metadata/dao/utils/RecordUtils.java
index 722fd55de..cd5cdf060 100644
--- a/dao-api/src/main/java/com/linkedin/metadata/dao/utils/RecordUtils.java
+++ b/dao-api/src/main/java/com/linkedin/metadata/dao/utils/RecordUtils.java
@@ -78,15 +78,6 @@ public static String toJsonString(@Nonnull RecordTemplate recordTemplate) {
     }
   }
 
-  /**
-   * Serializes a list of {@link RecordTemplate} to JSON string.
-   * @param recordTemplates the list of record templates to serialize
-   * @return the JSON string serialized using {@link JacksonDataTemplateCodec}.
-   */
-  public static String toJsonString(@Nonnull List<RecordTemplate> recordTemplates) {
-    return recordTemplates.stream().map(RecordUtils::toJsonString).collect(Collectors.joining(","));
-  }
-
   /**
    * Creates a {@link RecordTemplate} object from a serialized JSON string.
    *
diff --git a/dao-api/src/test/java/com/linkedin/metadata/dao/BaseLocalDAOTest.java b/dao-api/src/test/java/com/linkedin/metadata/dao/BaseLocalDAOTest.java
index aabdb44e4..260dd8190 100644
--- a/dao-api/src/test/java/com/linkedin/metadata/dao/BaseLocalDAOTest.java
+++ b/dao-api/src/test/java/com/linkedin/metadata/dao/BaseLocalDAOTest.java
@@ -710,7 +710,7 @@ public void testCreateAspectWithCallbacks() throws URISyntaxException {
     aspectValues.add(foo);
     aspectValues.add(bar);
 
-    FooUrn result = _dummyLocalDAO.createAspectWithCallbacks(urn, aspectValues, aspectCreateLambdas, _dummyAuditStamp, null);
+    FooUrn result = _dummyLocalDAO.createAspectsWithCallbacks(urn, aspectValues, aspectCreateLambdas, _dummyAuditStamp, null);
     assertEquals(result, urn);
   }
 
diff --git a/dao-impl/ebean-dao/src/test/java/com/linkedin/metadata/dao/utils/EmbeddedMariaInstance.java b/dao-impl/ebean-dao/src/test/java/com/linkedin/metadata/dao/utils/EmbeddedMariaInstance.java
index 696d67aaa..813e8d878 100644
--- a/dao-impl/ebean-dao/src/test/java/com/linkedin/metadata/dao/utils/EmbeddedMariaInstance.java
+++ b/dao-impl/ebean-dao/src/test/java/com/linkedin/metadata/dao/utils/EmbeddedMariaInstance.java
@@ -95,11 +95,11 @@ private static void initDB() {
 
           /*
           * Add below 3 lines of code if building datahub-gma on a M1 / M2 chip Apple computer.
-          *
-          * configurationBuilder.setBaseDir("/opt/homebrew");
-          * configurationBuilder.setUnpackingFromClasspath(false);
-          * configurationBuilder.setLibDir(System.getProperty("java.io.tmpdir") + "/MariaDB4j/no-libs");
           */
+          configurationBuilder.setBaseDir("/opt/homebrew");
+          configurationBuilder.setUnpackingFromClasspath(false);
+          configurationBuilder.setLibDir(System.getProperty("java.io.tmpdir") + "/MariaDB4j/no-libs");
+
 
           try {
             // ensure the DB directory is deleted before we start to have a clean start