Reuse id object

AAFC-BICoE · Oct 8, 2024 · 6363a85 · 6363a85
1 parent f522029
commit 6363a85
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/dina-base-api/src/main/java/ca/gc/aafc/dina/repository/DinaRepository.java b/dina-base-api/src/main/java/ca/gc/aafc/dina/repository/DinaRepository.java
@@ -303,6 +303,7 @@ private QuerySpec resolveFilterAdapters(QuerySpec querySpec) {
    */
   @Transactional
   @Override
+  @SuppressWarnings("unchecked")
   public <S extends D> S save(S resource) {
 
     // make sure data is safe to manipulate
@@ -323,9 +324,7 @@ public <S extends D> S save(S resource) {
     dinaService.update(entity);
 
     // reload since the data may be changed by the service
-    D dto = findOne(
-      (Serializable) PropertyUtils.getProperty(entity, idFieldName),
-      new QuerySpec(resourceClass));
+    D dto = findOne((Serializable) id, new QuerySpec(resourceClass));
 
     auditService.ifPresent(service -> service.audit(dto));
     return (S) dto;