[Fix partially kbss-cvut/fta-fmea-ui#428] Extend dao layer to support…

… FaultTree status - add FaultTreeDao.updateStatus - add ManagedEntityDao.findSummary
kbss-cvut · Jun 26, 2024 · 3c0fafa · 3c0fafa
1 parent a7b1eaa
commit 3c0fafa
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 16 deletions.
diff --git a/src/main/java/cz/cvut/kbss/analysis/dao/FaultTreeDao.java b/src/main/java/cz/cvut/kbss/analysis/dao/FaultTreeDao.java
@@ -21,6 +21,8 @@
 @Repository
 public class FaultTreeDao extends ManagedEntityDao<FaultTree> {
 
+    public static final URI STATUS_PROP = URI.create(Vocabulary.s_p_status);
+
     @Autowired
     protected FaultTreeDao(EntityManager em, PersistenceConf config, IdentifierService identifierService, SecurityUtils securityUtils) {
         super(FaultTree.class, em, config, identifierService, securityUtils);
@@ -168,4 +170,8 @@ public Query getSummariesQuery() {
                     .setParameter("pCreator", P_CREATOR)
                     .setParameter("pLastEditor", P_LAST_EDITOR);
     }
+
+    public void updateStatus(URI faultTree, Status status){
+        super.addOrReplaceValue(faultTree, STATUS_PROP, status, faultTree);
+    }
 }
diff --git a/src/main/java/cz/cvut/kbss/analysis/dao/ManagedEntityDao.java b/src/main/java/cz/cvut/kbss/analysis/dao/ManagedEntityDao.java
@@ -11,6 +11,7 @@
 import cz.cvut.kbss.jopa.model.descriptors.EntityDescriptor;
 import cz.cvut.kbss.jopa.model.metamodel.Attribute;
 import cz.cvut.kbss.jopa.model.metamodel.EntityType;
+import cz.cvut.kbss.jopa.model.query.Query;
 import cz.cvut.kbss.jopa.vocabulary.DC;
 
 import java.net.URI;
@@ -75,8 +76,30 @@ public void setChangedByContext(URI context, Date date){
 
     public List<T> findAllSummaries(){
         try {
-            List<ManagedEntity> ret = em.createNativeQuery("""
+            List<ManagedEntity> ret = getSummariesQuery()
+                    .getResultList();
+
+            return ret.stream().map(s -> s.asEntity(type)).toList();
+        } catch (RuntimeException e) {
+            throw new PersistenceException(e);
+        }
+    }
+
+    public T findSummary(URI managedEntityUri){
+        try {
+            ManagedEntity ret = (ManagedEntity)getSummariesQuery()
+                    .setParameter("_uri", managedEntityUri)
+                    .getSingleResult();
+            return ret.asEntity(type);
+        } catch (RuntimeException e) {
+            throw new PersistenceException(e);
+        }
+    }
+
+    public Query getSummariesQuery(){
+        return em.createNativeQuery("""
                             SELECT * WHERE {
+                                BIND(?_uri as ?uri)
                                 ?uri a ?type.
                                 ?uri ?pName ?name.
                                 OPTIONAL{?uri ?pDescription ?description.}
@@ -85,21 +108,13 @@ public List<T> findAllSummaries(){
                                 OPTIONAL{?uri ?pCreator ?creator.}
                                 OPTIONAL{?uri ?pLastEditor ?lastEditor.}
                             }""", "ManagedEntitySummary")
-                    .setParameter("type", typeUri)
-                    .setParameter("pName", P_HAS_NAME)
-                    .setParameter("pDescription", P_HAS_DESCRIPTION)
-                    .setParameter("pCreated", P_CREATED)
-                    .setParameter("pModified", P_MODIFIED)
-                    .setParameter("pCreator", P_CREATOR)
-                    .setParameter("pLastEditor", P_LAST_EDITOR)
-                    .getResultList();
-
-            return ret.stream().map(s -> s.asEntity(type)).toList();
-        } catch (RuntimeException e) {
-            throw new PersistenceException(e);
-        }
+                .setParameter("type", typeUri)
+                .setParameter("pName", P_HAS_NAME)
+                .setParameter("pDescription", P_HAS_DESCRIPTION)
+                .setParameter("pCreated", P_CREATED)
+                .setParameter("pModified", P_MODIFIED)
+                .setParameter("pCreator", P_CREATOR)
+                .setParameter("pLastEditor", P_LAST_EDITOR);
     }
 
-
-
 }