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

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