bundles/org.simantics.db.impl/src/org/simantics/db/impl/query/CodeGen.java

   1 package org.simantics.db.impl.query;
   2
   3 import java.io.File;
   4 import java.io.IOException;
   5 import java.io.UnsupportedEncodingException;
   6 import java.net.MalformedURLException;
   7 import java.net.URL;
   8 import java.net.URLDecoder;
   9
  10 import org.simantics.db.ObjectResourceIdMap;
  11 import org.simantics.utils.FileUtils;
  12
  13 public class CodeGen {
  14
  15         int indent = 4;
  16
  17
  18
  19         String[] signatureR1RelationInfo = { "int r", "r", "keyR", "long", "InternalProcedure<RelationInfo>", "entry.id" };
  20         String[] signatureR1Bytes = { "int r", "r", "keyR", "long", "InternalProcedure<byte[]>", "entry.id" };
  21         String[] signatureR1IntSet = { "int r", "r", "keyR", "long", "InternalProcedure<IntSet>", "entry.id" };
  22         String[] signatureR1IP = { "int r", "r", "keyR", "long", "IntProcedure", "entry.id" };
  23         String[] signatureR2IP = { "int r1, int r2", "r1,r2", "keyR2", "long", "IntProcedure", "entry.id" };
  24         String[] signatureR2TIP = { "int r1, int r2", "r1,r2", "keyR2", "long", "TripleIntProcedure", "entry.id" };
  25         String[] signatureID1 = { "String id", "id", "keyID", "String", "InternalProcedure<Integer>", "entry.id" };
  26         String[] signatureID2 = { "String id", "id", "keyID", "String", "InternalProcedure<TObjectIntHashMap<String>>", "entry.id" };
  27         String[] signatureChildMap = { "int r", "r", "keyR", "long", "InternalProcedure<ObjectResourceIdMap<String>>", "entry.id" };
  28         String[] signatureRead = { "Read<?> r", "r", "id", "long", "AsyncProcedure", "entry.request" };
  29         String[] signatureAsyncRead = { "AsyncRead<?> r", "r", "id", "long", "AsyncProcedure", "entry.request" };
  30         String[] signatureMultiRead = { "MultiRead<?> r", "r", "id", "long", "AsyncMultiProcedure", "entry.request" };
  31         String[] signatureAsyncMultiRead = { "AsyncMultiRead<?> r", "r", "id", "long", "AsyncMultiProcedure", "entry.request" };
  32         String[] signatureExternalRead = { "ExternalRead<?> r", "r", "id", "long", "AsyncProcedure", "entry.request" };
  33
  34         private void line(StringBuilder content, String line) {
  35                 for(int i=0;i<indent;i++)
  36                         content.append(" ");
  37                 content.append(line);
  38                 content.append("\n");
  39         }
  40
  41         public void generateQuery(StringBuilder content, String clazz, String[] signature, boolean runnerShortcut) {
  42                 generateGetOrCreate(content, clazz, signature);
  43                 generateRemove(content, clazz, signature);
  44                 generateRunner(content, clazz, signature, runnerShortcut);
  45         }
  46
  47         public void generateRunner(StringBuilder content, String clazz, String[] signature, boolean shortcut) {
  48
  49                 line(content, "public static void runner" + clazz + "(ReadGraphImpl graph, " + signature[0] + ", CacheEntry parent, ListenerBase listener, " + signature[4] + " procedure) throws DatabaseException {");
  50                 line(content, "    QueryCache cache  = graph.processor.cache;");
  51                 if(shortcut) {
  52                         line(content, "    if(parent == null && listener == null && !cache.shouldCache(graph.processor, " + signature[1] + ")) {");
  53                         line(content, "        " + clazz + ".computeForEach(graph, " + signature[1] + ", null, procedure);");
  54                         line(content, "        return;");
  55                         line(content, "    }");
  56                 }
  57                 line(content, "    if(procedure == null) procedure = emptyProcedure" + clazz + ";");
  58                 line(content, "    " + clazz + " entry = (" + clazz + ")cache.getOrCreate" + clazz + "(" + signature[1] + ");");
  59                 line(content, "    ListenerEntry listenerEntry = cache.registerDependencies(graph, entry, parent, listener, procedure, false);");
  60                 line(content, "    if(entry.isReady()) entry.performFromCache(graph, procedure);");
  61                 line(content, "    else {");
  62                 if(shortcut) line(content, "        " + clazz + ".computeForEach(graph, " + signature[1] + ", entry, procedure);");
  63                 else line(content, "        entry.compute(graph, procedure);");
  64                 line(content, "        if(listenerEntry != null) cache.primeListenerEntry(listenerEntry, entry.getResult());");
  65                 line(content, "    }");
  66                 line(content, "}");
  67                 line(content, "");
  68
  69         }
  70
  71         public void generateRemove(StringBuilder content, String clazz, String[] signature) {
  72
  73                 String lower = Character.toLowerCase(clazz.charAt(0)) + clazz.substring(1);
  74
  75                 line(content, "void remove(" + clazz + " entry) {");
  76                 line(content, "    synchronized(" + lower + "Map) {");
  77                 line(content, "        " + lower + "Map.remove(" + signature[5] + ");");
  78                 line(content, "    }");
  79                 line(content, "}");
  80                 line(content, "");
  81
  82         }
  83
  84         public void generateGetOrCreate(StringBuilder content, String clazz, String[] signature) {
  85
  86                 String lower = Character.toLowerCase(clazz.charAt(0)) + clazz.substring(1);
  87
  88                 line(content, "" + clazz + " getOrCreate" + clazz + "(" + signature[0] + ") throws DatabaseException {");
  89                 line(content, "    " + clazz + " existing = null;");
  90                 line(content, "    synchronized(" + lower + "Map) {");
  91                 line(content, "        existing = (" + clazz + ")" + lower + "Map.get(" + signature[1] + ");");
  92                 line(content, "        if(existing == null) {");
  93                 line(content, "            existing = new " + clazz + "(" + signature[1] + ");");
  94                 line(content, "            existing.clearResult(querySupport);");
  95                 line(content, "            existing.setPending();");
  96                 line(content, "            " + lower + "Map.put(" + signature[2] + "(" + signature[1] + "), existing);");
  97                 line(content, "            size++;");
  98                 line(content, "            return existing;");
  99                 line(content, "        }");
 100                 line(content, "        if(existing.requiresComputation()) {");
 101                 line(content, "            existing.setPending();");
 102                 line(content, "            return existing;");
 103                 line(content, "        }");
 104                 line(content, "    }");
 105                 line(content, "    if(existing.isPending()) waitPending(existing);");
 106                 line(content, "    return existing;");
 107                 line(content, "}");
 108                 line(content, "");
 109
 110         }
 111
 112         public void generate() {
 113
 114         URL classLocation = CodeGen.class.getResource(".");
 115         if (classLocation != null) {
 116             if (classLocation.getProtocol().equals("file")) {
 117                 try {
 118                     URL resource = new URL(classLocation, ".");
 119                     File path = new File(URLDecoder.decode(resource.getPath(), "UTF-8"));
 120                     String target = path.getAbsolutePath().replace("\\", "/");
 121                     target = target.replace("/bin/", "/src/") + "/QueryCache.java";
 122                     System.err.println("target=" + target);
 123                     File source = new File(target);
 124                     StringBuilder content = new StringBuilder();
 125                     content.append("package org.simantics.db.impl.query;\n");
 126                     content.append("\n");
 127
 128                     content.append("import org.simantics.db.ObjectResourceIdMap;\n");
 129                     content.append("import org.simantics.db.RelationInfo;\n");
 130                     content.append("import org.simantics.db.exception.DatabaseException;\n");
 131                     content.append("import org.simantics.db.impl.graph.ReadGraphImpl;\n");
 132                     content.append("import org.simantics.db.impl.procedure.InternalProcedure;\n");
 133                     content.append("import org.simantics.db.procedure.AsyncMultiProcedure;\n");
 134                     content.append("import org.simantics.db.procedure.AsyncProcedure;\n");
 135                     content.append("import org.simantics.db.procedure.ListenerBase;\n");
 136                     content.append("import org.simantics.db.request.AsyncMultiRead;\n");
 137                     content.append("import org.simantics.db.request.AsyncRead;\n");
 138                     content.append("import org.simantics.db.request.ExternalRead;\n");
 139                     content.append("import org.simantics.db.request.MultiRead;\n");
 140                     content.append("import org.simantics.db.request.Read;\n");
 141                     content.append("\n");
 142                     content.append("import gnu.trove.map.hash.TObjectIntHashMap;\n");
 143                     content.append("\n");
 144
 145                     content.append("public class QueryCache extends QueryCacheBase {\n");
 146                     content.append("\n");
 147                     line(content,"public QueryCache(QuerySupport querySupport, int threads) {");
 148                     line(content,"    super(querySupport, threads);");
 149                     line(content,"}");
 150                     content.append("\n");
 151
 152                     generateQuery(content, "Objects", signatureR2IP, true);
 153                     generateQuery(content, "Statements", signatureR2TIP, true);
 154                     generateQuery(content, "DirectObjects", signatureR2IP, true);
 155                     generateQuery(content, "RelationInfoQuery", signatureR1RelationInfo, true);
 156                     generateQuery(content, "URIToResource", signatureID1, true);
 157                     generateQuery(content, "ValueQuery", signatureR1Bytes, true);
 158                     generateQuery(content, "OrderedSet", signatureR1IP, true);
 159                     generateQuery(content, "PrincipalTypes", signatureR1IP, true);
 160                     generateQuery(content, "DirectPredicates", signatureR1IntSet, true);
 161                     generateQuery(content, "Predicates", signatureR1IntSet, true);
 162                     generateQuery(content, "ReadEntry", signatureRead, true);
 163                     generateQuery(content, "AsyncReadEntry", signatureAsyncRead, true);
 164                     generateQuery(content, "Types", signatureR1IntSet, true);
 165                     //generateQuery(content, "NamespaceIndex", signatureID2, true);
 166                     generateQuery(content, "ChildMap", signatureChildMap, true);
 167
 168                     generateQuery(content, "AssertedStatements", signatureR2TIP, false);
 169                     generateQuery(content, "AssertedPredicates", signatureR1IP, false);
 170                     generateQuery(content, "DirectSuperRelations", signatureR1IP, false);
 171                     generateQuery(content, "SuperTypes", signatureR1IntSet, false);
 172                     generateQuery(content, "TypeHierarchy", signatureR1IntSet, false);
 173                     generateQuery(content, "SuperRelations", signatureR1IntSet, false);
 174                     generateQuery(content, "MultiReadEntry", signatureMultiRead, false);
 175                     generateQuery(content, "AsyncMultiReadEntry", signatureAsyncMultiRead, false);
 176                     generateQuery(content, "ExternalReadEntry", signatureExternalRead, false);
 177                     content.append("}\n");
 178                     FileUtils.writeFile(source, content.toString().getBytes());
 179                 } catch (MalformedURLException e) {
 180                     e.printStackTrace();
 181                 } catch (UnsupportedEncodingException e) {
 182                     e.printStackTrace();
 183                 } catch (IOException e) {
 184                                         e.printStackTrace();
 185                                 }
 186             }
 187         }
 188
 189         }
 190
 191         public static void main(String[] args) {
 192                 new CodeGen().generate();
 193         }
 194
 195 }