SqaleRepositoryService.java: added basic countObjects implementation

Evolveum · Feb 16, 2021 · 9a85a68 · 9a85a68
1 parent 2e2c249
commit 9a85a68
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/repo/repo-sqale/src/main/java/com/evolveum/midpoint/repo/sqale/SqaleRepositoryService.java b/repo/repo-sqale/src/main/java/com/evolveum/midpoint/repo/sqale/SqaleRepositoryService.java
@@ -515,8 +515,27 @@ private void logTraceModifications(@NotNull Collection<? extends ItemDelta<?, ?>
     public <T extends ObjectType> int countObjects(Class<T> type, ObjectQuery query,
             Collection<SelectorOptions<GetOperationOptions>> options, OperationResult parentResult)
             throws SchemaException {
-        return 0;
-        // TODO
+        Objects.requireNonNull(type, "Object type must not be null.");
+        Objects.requireNonNull(parentResult, "Operation result must not be null.");
+
+        OperationResult operationResult = parentResult.subresult(OP_NAME_PREFIX + "countObjects")
+                .addQualifier(type.getSimpleName())
+                .addParam("type", type.getName())
+                .addParam("query", query)
+                .build();
+
+        try {
+            var queryContext = SqaleQueryContext.from(type, transformerContext, sqlRepoContext);
+            return sqlQueryExecutor.count(queryContext, query, options);
+        } catch (QueryException | RuntimeException e) {
+            handleGeneralException(e, operationResult);
+            throw new SystemException(e);
+        } catch (Throwable t) {
+            operationResult.recordFatalError(t);
+            throw t;
+        } finally {
+            operationResult.computeStatusIfUnknown();
+        }
     }
 
     @Override