Example #1
0
 private CypherResult doExecuteQuery(String query, boolean canProfile) {
   long time = System.currentTimeMillis();
   Transaction tx = gdb.beginTx();
   javax.transaction.Transaction resumeTx;
   try {
     resumeTx = suspendTx(query);
     ExecutionResult result =
         canProfile ? executionEngine.profile(query) : executionEngine.execute(query);
     final Collection<Map<String, Object>> data = IteratorUtil.asCollection(result);
     time = System.currentTimeMillis() - time;
     resumeTransaction(resumeTx);
     CypherResult cypherResult =
         new CypherResult(
             result.columns(),
             data,
             result.getQueryStatistics(),
             time,
             canProfile ? result.executionPlanDescription() : null,
             prettify(query));
     tx.success();
     return cypherResult;
   } finally {
     tx.close();
   }
 }
Example #2
0
 public static void createData(String filePath) throws IOException {
   long time = System.currentTimeMillis();
   final PrintWriter writer = new PrintWriter(new BufferedWriter(new FileWriter(filePath)));
   Random r = new Random(-1L);
   for (int nodes = 0; nodes < COUNT; nodes++) {
     writer.printf("%07d|%07d|%07d%n", nodes, r.nextInt(COUNT), r.nextInt(COUNT));
   }
   writer.close();
   System.out.println(
       "Creating data took " + (System.currentTimeMillis() - time) / 1000 + " seconds");
 }
Example #3
0
  public void importFromFile(String filePath) throws IOException {
    Map<String, Long> cache = new HashMap<String, Long>(COUNT);
    final File storeDir = new File(this.path);
    org.apache.commons.io.FileUtils.deleteDirectory(storeDir);
    BatchInserter batchInserter = new BatchInserterImpl(storeDir.getAbsolutePath());
    final BatchInserterIndexProvider indexProvider =
        new LuceneBatchInserterIndexProvider(batchInserter);
    final BatchInserterIndex index =
        indexProvider.nodeIndex("nodes", MapUtil.stringMap("type", "exact"));
    BufferedReader reader = new BufferedReader(new FileReader(filePath));
    String line = null;
    int nodes = 0;
    long time = System.currentTimeMillis();
    long batchTime = time;
    while ((line = reader.readLine()) != null) {
      final String[] nodeNames = line.split("\\|");
      final String name = nodeNames[0];
      final Map<String, Object> props = MapUtil.map("name", name);
      final long node = batchInserter.createNode(props);
      index.add(node, props);
      cache.put(name, node);
      nodes++;
      if ((nodes % REPORT_COUNT) == 0) {
        System.out.printf(
            "%d nodes created. Took %d %n", nodes, (System.currentTimeMillis() - batchTime));
        batchTime = System.currentTimeMillis();
      }
    }

    System.out.println("Creating nodes took " + (System.currentTimeMillis() - time) / 1000);
    index.flush();
    reader.close();
    reader = new BufferedReader(new FileReader(filePath));
    int rels = 0;
    time = System.currentTimeMillis();
    batchTime = time;
    String relationshipType = "KNOWS";
    while ((line = reader.readLine()) != null) {
      final String[] nodeNames = line.split("\\|");
      final String name = nodeNames[0];
      // final Long from = index.get("name", name).getSingle();
      Long from = cache.get(name);
      for (int j = 1; j < nodeNames.length; j++) {
        // final Long to = index.get("name", nodeNames[j]).getSingle();
        final Long to = cache.get(name);
        batchInserter.createRelationship(
            from, to, DynamicRelationshipType.withName(relationshipType), null);
      }
      rels++;
      if ((rels % REPORT_COUNT) == 0) {
        System.out.printf(
            "%d relationships created. Took %d %n", rels, (System.currentTimeMillis() - batchTime));
        batchTime = System.currentTimeMillis();
      }
    }
    System.out.println("Creating relationships took " + (System.currentTimeMillis() - time) / 1000);
    indexProvider.shutdown();
    batchInserter.shutdown();
  }
Example #4
0
 public static void createJsonData(String filePath) throws IOException {
   long time = System.currentTimeMillis();
   final PrintWriter writer = new PrintWriter(new BufferedWriter(new FileWriter(filePath)));
   Random r = new Random(-1L);
   int total = 100;
   for (int nodes = 0; nodes < total; nodes++) {
     // {"name" : "bar", relationships : [{"to" : "91", "type" : "LOVES", "data" : {"foo" :
     // "bar"}}]}
     // {"to" : "91", "type" : "LOVES", "data" : {"foo" : "bar"}}
     writer.printf("%07d|%07d|%07d%n", nodes, r.nextInt(total), r.nextInt(total));
   }
   writer.close();
   System.out.println(
       "Creating data took " + (System.currentTimeMillis() - time) / 1000 + " seconds");
 }
 @Test
 public void testEnableBatchTransactions() throws Exception {
   System.setProperty(Config.CONFIG_BATCH_TRANSACTION, "true");
   Transaction tx = restAPI.beginTx();
   tx.failure();
   tx.finish();
   assertTrue(tx instanceof BatchTransaction);
 }
  @Override
  public Iterable<Relationship> expand(Path path, BranchState branchState) {
    if (System.currentTimeMillis() < stopTime) {

      Long minimumConnectTime = 45L * 60L; // 45 minutes

      switch (path.length()) {
        case 0:
          return path.endNode()
              .getRelationships(Direction.OUTGOING, RelationshipTypes.HAS_DESTINATION);
        case 1:
          {
            Node lastNode = path.endNode();
            if (destinations.contains(((String) lastNode.getProperty("code")))
                || exclusions.contains((String) lastNode.getProperty("code"))) {
              return Collections.emptyList();
            } else {
              return path.endNode().getRelationships(Direction.OUTGOING, relationshipTypes);
            }
          }
        case 2:
          {
            return path.endNode().getRelationships(Direction.OUTGOING, relationshipTypes);
          }
        case 3:
          {
            return path.endNode()
                .getRelationships(Direction.OUTGOING, RelationshipTypes.HAS_DESTINATION);
          }
        case 4:
          {
            Node lastNode = path.endNode();
            if (destinations.contains(((String) lastNode.getProperty("code")))) {
              return path.endNode().getRelationships(Direction.OUTGOING, relationshipTypes);
            } else {
              return Collections.emptyList();
            }
          }
        case 5:
          {
            Node lastNode = path.endNode();
            Iterator<Node> nodes = path.nodes().iterator();
            nodes.next();
            nodes.next();
            Node lastFlight = nodes.next();

            if (((Long) lastFlight.getProperty("arrives") + minimumConnectTime)
                > (Long) lastNode.getProperty("departs")) {
              return Collections.emptyList();
            }

            if (lastNode.getProperty("code").equals(lastFlight.getProperty("code"))) {
              return Collections.emptyList();
            } else {
              return path.endNode().getRelationships(Direction.OUTGOING, relationshipTypes);
            }
          }
        default:
          return Collections.emptyList();
      }
    } else {
      return Collections.emptyList();
    }
  }
 @Before
 public void init() {
   System.setProperty(Config.CONFIG_BATCH_TRANSACTION, "true");
   this.restAPI = ((RestGraphDatabase) getRestGraphDb()).getRestAPI();
 }