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