implemented count command for ninja

Evolveum · Mar 14, 2018 · 93cd957 · 93cd957
1 parent abb6866
commit 93cd957
Show file tree

Hide file tree

Showing 4 changed files with 70 additions and 6 deletions.
diff --git a/tools/ninja/src/main/java/com/evolveum/midpoint/ninja/action/CountRepositoryAction.java b/tools/ninja/src/main/java/com/evolveum/midpoint/ninja/action/CountRepositoryAction.java
@@ -16,15 +16,76 @@
 
 package com.evolveum.midpoint.ninja.action;
 
+import com.evolveum.midpoint.ninja.impl.NinjaException;
 import com.evolveum.midpoint.ninja.opts.CountOptions;
+import com.evolveum.midpoint.ninja.util.FileReference;
+import com.evolveum.midpoint.ninja.util.NinjaUtils;
+import com.evolveum.midpoint.prism.query.ObjectQuery;
+import com.evolveum.midpoint.repo.api.RepositoryService;
+import com.evolveum.midpoint.schema.constants.ObjectTypes;
+import com.evolveum.midpoint.schema.result.OperationResult;
+import com.evolveum.midpoint.xml.ns._public.common.common_3.ObjectType;
+
+import java.lang.reflect.Modifier;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
 
 /**
  * Created by Viliam Repan (lazyman).
  */
 public class CountRepositoryAction extends RepositoryAction<CountOptions> {
 
+    private static final String DOT_CLASS = CountRepositoryAction.class.getName() + ".";
+
+    private static final String OPERATION_COUNT = DOT_CLASS + "count";
+
     @Override
     public void execute() throws Exception {
-        // todo implement
+        RepositoryService repository = context.getRepository();
+
+        FileReference fileReference = options.getFilter();
+        if (fileReference != null && options.getFilter() == null) {
+            throw new NinjaException("Type must be defined");
+        }
+
+        ObjectQuery query = NinjaUtils.createObjectQuery(options.getFilter(), context);
+
+        List<ObjectTypes> types = getTypes();
+
+        int total = 0;
+        OperationResult result = new OperationResult(OPERATION_COUNT);
+        for (ObjectTypes type : types) {
+            Class<? extends ObjectType> clazz = type.getClassDefinition();
+            if (Modifier.isAbstract(clazz.getModifiers())) {
+                continue;
+            }
+
+            int count = repository.countObjects(clazz, query, new ArrayList<>(), result);
+            if (count == 0 && options.getType() == null) {
+                continue;
+            }
+            log.info("{}:\t{}", type.name(), count);
+
+            total += count;
+        }
+
+        log.info("===\nTotal:\t{}", total);
+    }
+
+    private List<ObjectTypes> getTypes() {
+        List<ObjectTypes> types = new ArrayList<>();
+
+        ObjectTypes type = options.getType();
+        if (type != null) {
+            types.add(type);
+        } else {
+            types.addAll(Arrays.asList(ObjectTypes.values()));
+        }
+
+        Collections.sort(types);
+
+        return types;
     }
 }
diff --git a/tools/ninja/src/main/java/com/evolveum/midpoint/ninja/impl/Command.java b/tools/ninja/src/main/java/com/evolveum/midpoint/ninja/impl/Command.java
@@ -14,7 +14,7 @@ public enum Command {
 
     DELETE("delete", DeleteOptions.class, DeleteRepositoryAction.class, null),
 
-//    COUNT("count", CountOptions.class, CountRepositoryAction.class, null),
+    COUNT("count", CountOptions.class, CountRepositoryAction.class, null),
 
 //    PASSWORD_RESET("password", PasswordResetOptions.class, PasswordResetRepositoryAction.class, null),
 //

diff --git a/tools/ninja/src/main/java/com/evolveum/midpoint/ninja/opts/CountOptions.java b/tools/ninja/src/main/java/com/evolveum/midpoint/ninja/opts/CountOptions.java
@@ -17,6 +17,7 @@
 package com.evolveum.midpoint.ninja.opts;
 
 import com.beust.jcommander.Parameter;
+import com.beust.jcommander.Parameters;
 import com.evolveum.midpoint.ninja.util.FileReference;
 import com.evolveum.midpoint.ninja.util.FileReferenceConverter;
 import com.evolveum.midpoint.ninja.util.ObjectTypesConverter;
@@ -25,6 +26,7 @@
 /**
  * Created by Viliam Repan (lazyman).
  */
+@Parameters(resourceBundle = "messages", commandDescriptionKey = "count")
 public class CountOptions {
 
     public static final String P_TYPE = "-t";
@@ -33,11 +35,11 @@ public class CountOptions {
     public static final String P_FILTER = "-f";
     public static final String P_FILTER_LONG = "--filter";
 
-    @Parameter(names = {P_TYPE, P_TYPE_LONG}, descriptionKey = "countOptions.type",
+    @Parameter(names = {P_TYPE, P_TYPE_LONG}, descriptionKey = "count.type",
             validateWith = ObjectTypesConverter.class, converter = ObjectTypesConverter.class)
     private ObjectTypes type;
 
-    @Parameter(names = {P_FILTER, P_FILTER_LONG}, descriptionKey = "countOptions.filter",
+    @Parameter(names = {P_FILTER, P_FILTER_LONG}, descriptionKey = "count.filter",
             converter = FileReferenceConverter.class, validateWith = FileReferenceConverter.class)
     private FileReference filter;
 

diff --git a/tools/ninja/src/main/resources/messages.properties b/tools/ninja/src/main/resources/messages.properties
@@ -60,5 +60,6 @@ listKeys.askKeyPassword=Please write key password
 schema=DB schema operations
 schema.test=Validate DB schema
 schema.init=Initialize DB schema
-countOptions.type=
-countOptions.filter=
+count=
+count.type=
+count.filter=