fixTest

Netflix · Jun 2, 2024 · 15f652a · 15f652a
1 parent c2adfdf
commit 15f652a
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/.../src/main/java/com/netflix/metacat/metadata/mysql/MySqlParentChildRelMetaDataService.java b/.../src/main/java/com/netflix/metacat/metadata/mysql/MySqlParentChildRelMetaDataService.java
@@ -100,22 +100,26 @@ public void createParentChildRelation(final QualifiedName parentName,
     @Override
     public void rename(final QualifiedName oldName, final QualifiedName newName) {
         try {
-            retryTemplate.execute((RetryCallback<Void, RuntimeException>) context -> {
-                jdbcTemplate.update(connection -> {
+            // Retry for renaming the Parent Entity
+            retryTemplate.execute((RetryCallback<Integer, RuntimeException>) context -> {
+                return jdbcTemplate.update(connection -> {
                     // Rename Parent Entity
                     final PreparedStatement renameParentPS = connection.prepareStatement(SQL_RENAME_PARENT_ENTITY);
                     renameParentPS.setString(1, newName.toString());
                     renameParentPS.setString(2, oldName.toString());
-                    renameParentPS.executeUpdate();
+                    return renameParentPS;
+                });
+            });
 
+            // Retry for renaming the Child Entity
+            retryTemplate.execute((RetryCallback<Integer, RuntimeException>) context -> {
+                return jdbcTemplate.update(connection -> {
                     // Rename Child Entity
                     final PreparedStatement renameChildPS = connection.prepareStatement(SQL_RENAME_CHILD_ENTITY);
                     renameChildPS.setString(1, newName.toString());
                     renameChildPS.setString(2, oldName.toString());
-                    renameChildPS.executeUpdate();
-                    return null;
+                    return renameChildPS;
                 });
-                return null;
             });
         } catch (RuntimeException e) {
             log.error("Failed to rename entity", e);