View Javadoc
1   package de.dlr.shepard.context.references.basicreference.daos;
2   
3   import de.dlr.shepard.common.util.CypherQueryHelper;
4   import de.dlr.shepard.common.util.QueryParamHelper;
5   import de.dlr.shepard.context.references.basicreference.entities.BasicReference;
6   import de.dlr.shepard.context.version.daos.VersionableEntityDAO;
7   import jakarta.enterprise.context.RequestScoped;
8   import java.util.ArrayList;
9   import java.util.Collections;
10  import java.util.HashMap;
11  import java.util.List;
12  import java.util.Map;
13  import java.util.stream.StreamSupport;
14  
15  @RequestScoped
16  public class BasicReferenceDAO extends VersionableEntityDAO<BasicReference> {
17  
18    @Override
19    public Class<BasicReference> getEntityType() {
20      return BasicReference.class;
21    }
22  
23    /**
24     * Searches the database for references.
25     *
26     * @param dataObjectId identifies the dataObject
27     * @param params       encapsulates possible parameters
28     * @return a List of references
29     */
30    public List<BasicReference> findByDataObjectNeo4jId(long dataObjectId, QueryParamHelper params) {
31      String query;
32      Map<String, Object> paramsMap = new HashMap<>();
33      paramsMap.put("name", params.getName());
34      if (params.hasPagination()) {
35        paramsMap.put("offset", params.getPagination().getOffset());
36        paramsMap.put("size", params.getPagination().getSize());
37      }
38      query = "MATCH (d:DataObject)-[hr:has_reference]->%s WHERE ID(d)=%d WITH r".formatted(
39          CypherQueryHelper.getObjectPart("r", "BasicReference", params.hasName()),
40          dataObjectId
41        );
42      if (params.hasOrderByAttribute()) {
43        query += " " + CypherQueryHelper.getOrderByPart("r", params.getOrderByAttribute(), params.getOrderDesc());
44      }
45      if (params.hasPagination()) {
46        query += " " + CypherQueryHelper.getPaginationPart();
47      }
48      query += " " + CypherQueryHelper.getReturnPart("r");
49      var result = new ArrayList<BasicReference>();
50      for (var ref : findByQuery(query, paramsMap)) {
51        if (matchDataObject(ref, dataObjectId) && matchName(ref, params.getName())) {
52          result.add(ref);
53        }
54      }
55      return result;
56    }
57  
58    /**
59     * Searches the database for references.
60     *
61     * @param dataObjectShepardId identifies the dataObject
62     * @param params              encapsulates possible parameters
63     * @return a List of references
64     */
65    public List<BasicReference> findByDataObjectShepardId(long dataObjectShepardId, QueryParamHelper params) {
66      String query;
67      Map<String, Object> paramsMap = new HashMap<>();
68      paramsMap.put("name", params.getName());
69      if (params.hasPagination()) {
70        paramsMap.put("offset", params.getPagination().getOffset());
71        paramsMap.put("size", params.getPagination().getSize());
72      }
73      query = "MATCH (d:DataObject)-[hr:has_reference]->%s WHERE d.shepardId=%d WITH br".formatted(
74          CypherQueryHelper.getObjectPart("br", "BasicReference", params.hasName()),
75          dataObjectShepardId
76        );
77      if (params.hasOrderByAttribute()) {
78        query += " " + CypherQueryHelper.getOrderByPart("br", params.getOrderByAttribute(), params.getOrderDesc());
79      }
80      if (params.hasPagination()) {
81        query += " " + CypherQueryHelper.getPaginationPart();
82      }
83      query += " " + CypherQueryHelper.getReturnPart("br");
84      var result = new ArrayList<BasicReference>();
85      for (var ref : findByQuery(query, paramsMap)) {
86        if (matchDataObjectByShepardId(ref, dataObjectShepardId) && matchName(ref, params.getName())) {
87          result.add(ref);
88        }
89      }
90      return result;
91    }
92  
93    private boolean matchDataObject(BasicReference ref, long dataObjectId) {
94      return ref.getDataObject() != null && ref.getDataObject().getId().equals(dataObjectId);
95    }
96  
97    private boolean matchDataObjectByShepardId(BasicReference ref, long dataObjectShepardId) {
98      return ref.getDataObject() != null && ref.getDataObject().getShepardId().equals(dataObjectShepardId);
99    }
100 
101   private boolean matchName(BasicReference ref, String name) {
102     return name == null || ref.getName().equalsIgnoreCase(name);
103   }
104 
105   public List<BasicReference> getBasicReferencesByQuery(String query) {
106     var queryResult = findByQuery(query, Collections.emptyMap());
107     List<BasicReference> ret = StreamSupport.stream(queryResult.spliterator(), false).toList();
108     return ret;
109   }
110 }