bundles/org.simantics.issues.common/src/org/simantics/issues/common/ConstraintIssueSource.java

   1 package org.simantics.issues.common;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collections;
   5 import java.util.List;
   6 import java.util.Map;
   7 import java.util.Set;
   8
   9 import org.eclipse.core.runtime.IProgressMonitor;
  10 import org.simantics.db.Issue;
  11 import org.simantics.db.ReadGraph;
  12 import org.simantics.db.Resource;
  13 import org.simantics.db.common.QueryMemoryWatcher;
  14 import org.simantics.db.common.utils.Logger;
  15 import org.simantics.db.exception.DatabaseException;
  16 import org.simantics.db.layer0.util.ModelTransferableGraphSourceRequest.DomainOnlyProcessor;
  17 import org.simantics.db.service.CollectionSupport;
  18 import org.simantics.db.service.QueryControl;
  19 import org.simantics.issues.ontology.IssueResource;
  20 import org.simantics.layer0.Layer0;
  21 import org.simantics.scl.db.SCLFunctions;
  22 import org.simantics.scl.runtime.function.Function1;
  23
  24 import gnu.trove.set.hash.THashSet;
  25
  26 /**
  27  * @author Antti Villberg
  28  */
  29 public class ConstraintIssueSource implements BatchIssueSource {
  30
  31     private final Resource resource;
  32
  33     public ConstraintIssueSource(Resource resource) {
  34         this.resource = resource;
  35     }
  36
  37     @Override
  38     public Map<Resource, Set<Issue>> run(IProgressMonitor monitor, ReadGraph graph, BatchIssueValidationContext context) throws DatabaseException {
  39
  40         Layer0 L0 = Layer0.getInstance(graph);
  41         Set<Issue> emptySet = Collections.emptySet();
  42         CollectionSupport cs = graph.getService(CollectionSupport.class);
  43         Map<Resource,Set<Issue>> result = cs.createMap(Set.class);
  44         monitor.setTaskName("Constraint analysis");
  45
  46         DomainOnlyProcessor domain = context.domain;
  47
  48         int entityCount = domain.internals.size();
  49
  50         IssueResource ISSUE = IssueResource.getInstance(graph);
  51         Resource type = graph.getSingleType(resource, ISSUE.IssueSource);
  52         List<Function1<Resource, List<Issue>>> validators = new ArrayList<>();
  53         for(Resource constraint : graph.getObjects(type, ISSUE.IssueSource_HasConstraint)) {
  54             Function1<Resource, List<Issue>> validator = graph.getRelatedValue2(constraint, L0.Constraint_Validator, constraint);
  55             //Resource function = graph.getSingleObject(constraint, L0.Constraint_Validator);
  56             validators.add(validator);
  57         }
  58
  59         QueryControl qc = graph.getService(QueryControl.class);
  60         qc.flush(graph);
  61
  62                 // Allow this process to make 50k queries
  63                 QueryMemoryWatcher memory = new QueryMemoryWatcher(graph, 50000, 0.5, 300);
  64
  65                 SCLFunctions.runWithGraph(graph, () -> {
  66
  67                 int totalExaminedCount = 0;
  68                 int examinedCount = 1000;
  69
  70                 for(Resource r : domain.internals) {
  71
  72                     Set<Issue> set = emptySet;
  73                     if (examinedCount >= 1000) {
  74                         monitor.subTask(contextProgressMessage(totalExaminedCount, entityCount));
  75                         examinedCount = 0;
  76                         if(monitor.isCanceled()) return;
  77                         memory.maintain();
  78                     }
  79                     for(Function1<Resource, List<Issue>> validator : validators) {
  80                         try {
  81                             @SuppressWarnings("unchecked")
  82                             List<Issue> issues = validator.apply(r);//(List<Issue>)Functions.exec(graph, validator, graph, r);
  83                             if (issues != null && !issues.isEmpty()) {
  84                                 if (set == emptySet)
  85                                     set = new THashSet<Issue>();
  86                                 set.addAll(issues);
  87                             }
  88                         } catch (Throwable t) {
  89                             Logger.defaultLogError(t);
  90                         }
  91                     }
  92                     ++totalExaminedCount;
  93                     ++examinedCount;
  94                     if(!set.isEmpty())
  95                         result.put(r, set);
  96
  97                 }
  98
  99                 });
 100
 101         return result;
 102
 103     }
 104
 105     private static String contextProgressMessage(int totalExaminedCount, int entityCount) {
 106         StringBuilder sb = new StringBuilder(80)
 107         .append("Validating resources").append(" ").append(100*totalExaminedCount / entityCount).append("% ready.");
 108         return sb.toString();
 109     }
 110
 111     @Override
 112     public Resource getResource() {
 113         return resource;
 114     }
 115
 116 }