@Override public OResultSet syncPull(OCommandContext ctx, int nRecords) throws OTimeoutException { OInternalResultSet result = new OInternalResultSet(); if (!executed) { result.add(copy(nextItem)); executed = true; } return result; }
private void doExecute(int n) { if (!executed) { executeUntilReturn(); executed = true; finalResult = new OInternalResultSet(); OResultSet partial = lastStep.syncPull(ctx, n); while (partial.hasNext()) { while (partial.hasNext()) { ((OInternalResultSet) finalResult).add(partial.next()); } partial = lastStep.syncPull(ctx, n); } if (lastStep instanceof ScriptLineStep) { ((OInternalResultSet) finalResult).setPlan(((ScriptLineStep) lastStep).plan); } } }
@Override public Map.Entry<Object, OIdentifiable> nextEntry() { if (!copy.hasNext()) return null; final OResult value = copy.next(); return new Map.Entry<Object, OIdentifiable>() { @Override public Object getKey() { return value.getProperty("key"); } @Override public OIdentifiable getValue() { return value.getProperty("rid"); } @Override public OIdentifiable setValue(OIdentifiable value) { throw new UnsupportedOperationException("setValue"); } }; } };
public Object convert(Object value) { if (value instanceof ORidBag) { List result = new ArrayList(); ((ORidBag) value).forEach(x -> result.add(x)); return result; } if (value instanceof OEdgeToVertexIterable) { value = ((OEdgeToVertexIterable) value).iterator(); } if (value instanceof OEdgeToVertexIterator) { List<ORID> result = new ArrayList<>(); while (((OEdgeToVertexIterator) value).hasNext()) { result.add(((OEdgeToVertexIterator) value).next().getIdentity()); } return result; } if (value instanceof OInternalResultSet) { ((OInternalResultSet) value).reset(); value = ((OInternalResultSet) value).stream().collect(Collectors.toList()); } return value; }
@Override public OResultSet fetchNext(int n) { return new OInternalResultSet(); }
private void buildNextRecord() { long begin = profilingEnabled ? System.nanoTime() : 0; try { if (currentTuple == null) { nextRecord = null; return; } nextRecord = new OResultInternal(); for (int i = 0; i < this.currentTuple.size(); i++) { OResult res = this.currentTuple.get(i); for (String s : res.getPropertyNames()) { nextRecord.setProperty(s, res.getProperty(s)); } if (!completedPrefetch.get(i)) { preFetches.get(i).add(res); if (!resultSets.get(i).hasNext()) { completedPrefetch.set(i, true); } } } } finally { if (profilingEnabled) { cost += (System.nanoTime() - begin); } } }
return ((OInternalResultSet) iObject).size();
@Override public OResultSet fetchNext(int n) { if (next >= result.size()) { return new OInternalResultSet();//empty } OIteratorResultSet nextBlock = new OIteratorResultSet(result.subList(next, Math.min(next + n, result.size())).iterator()); next += n; return nextBlock; }
@Override public OResultSet executeDDL(OCommandContext ctx) { OInternalResultSet result = new OInternalResultSet(); if (customPropertyName == null) { result.add(executeSimpleAlter(settingName, settingValue, ctx)); } else { result.add(executeCustomAlter(customPropertyName, customPropertyValue, ctx)); } return result; }
OInternalResultSet resultSet = new OInternalResultSet(); resultSet.setPlan(new OGremlinExecutionPlan((TraversalExplanation) eval)); OResultInternal item = new OResultInternal(); item.setProperty("executionPlan", ((TraversalExplanation) eval).prettyPrint()); resultSet.add(item); return resultSet; } else { OInternalResultSet resultSet = new OInternalResultSet(); OResultInternal item = new OResultInternal(); item.setProperty("value", this.transformer.toResult(eval)); resultSet.add(item); return resultSet;
@Override public OResultSet syncPull(OCommandContext ctx, int nRecords) throws OTimeoutException { if (executed) { return new OInternalResultSet(); } executed = true; return statement.executeSimple(ctx); }
@Override public Map.Entry<Object, OIdentifiable> nextEntry() { if (!copy.hasNext()) return null; final OResult next = copy.next(); return new Map.Entry<Object, OIdentifiable>() { @Override public Object getKey() { return next.getProperty("key"); } @Override public OIdentifiable getValue() { return next.getProperty("rid"); } @Override public OIdentifiable setValue(OIdentifiable value) { throw new UnsupportedOperationException("cannot set value of index entry"); } }; } };
@Override public OResultSet executeSimple(OCommandContext ctx) { ctx.getDatabase().rollback(); OInternalResultSet result = new OInternalResultSet(); OResultInternal item = new OResultInternal(); item.setProperty("operation", "rollback"); result.add(item); return result; }
@Override public OResultSet executeSimple(OCommandContext ctx) { Object result; if (expression != null) { result = expression.execute((OResult) null, ctx); } else { Map<Object, Object> params = ctx.getInputParameters(); if (statement.originalStatement == null) { statement.setOriginalStatement(statement.toString()); } result = statement.execute(ctx.getDatabase(), params, ctx); } if (result instanceof OResultSet) { OInternalResultSet rs = new OInternalResultSet(); ((OResultSet) result).stream().forEach(x -> rs.add(x)); rs.setPlan(((OResultSet) result).getExecutionPlan().orElse(null)); ((OResultSet) result).close(); result = rs; } if (ctx != null) { if (ctx.getParent() != null) { ctx.getParent().setVariable(name.getStringValue(), result); } else { ctx.setVariable(name.getStringValue(), result); } } return new OInternalResultSet(); }
@Override public OResultSet fetchNext(int n) { if (next >= result.size()) { return new OInternalResultSet();//empty } OIteratorResultSet nextBlock = new OIteratorResultSet(result.subList(next, Math.min(next + n, result.size())).iterator()); next += n; return nextBlock; }
@Override public Map.Entry<Object, OIdentifiable> nextEntry() { if (!copy.hasNext()) return null; final OResult value = copy.next(); return new Map.Entry<Object, OIdentifiable>() { @Override public Object getKey() { return value.getProperty("key"); } @Override public OIdentifiable getValue() { return value.getProperty("rid"); } @Override public OIdentifiable setValue(OIdentifiable value) { throw new UnsupportedOperationException("setValue"); } }; } };
@Override public OResultSet executeSimple(OCommandContext ctx) { ctx.getDatabase().commit(); OInternalResultSet result = new OInternalResultSet(); OResultInternal item = new OResultInternal(); item.setProperty("operation", "commit"); result.add(item); return result; }
private void init(OCommandContext ctx) { if (subPlans == null || subPlans.isEmpty()) { return; } if (inited) { return; } for (OInternalExecutionPlan plan : subPlans) { resultSets.add(new OLocalResultSet(plan)); this.preFetches.add(new OInternalResultSet()); } fetchFirstRecord(); inited = true; }
@Override public OResultSet executeDDL(OCommandContext ctx) { Object execResult = execute(ctx); OInternalResultSet rs = new OInternalResultSet(); if (execResult != null) { OResultInternal result = new OResultInternal(); result.setProperty("operation", "create index"); result.setProperty("name", name.getValue()); rs.add(result); } return rs; }