/** * Create an empty statement result. */ private StatementResult createEmptyStatementResult() { return new StatementResult(); }
public Object getFieldValue(_Fields field) { switch (field) { case RESULT_SET: return getResultSet(); case UPDATE_COUNT: return getUpdateCount(); case BATCH_UPDATE_COUNTS: return getBatchUpdateCounts(); case PROCEDURE_OUT_PARAMS: return getProcedureOutParams(); case GENERATED_KEYS: return getGeneratedKeys(); case NEW_DEFAULT_SCHEMA: return getNewDefaultSchema(); case WARNINGS: return getWarnings(); case PREPARED_RESULT: return getPreparedResult(); } throw new IllegalStateException(); }
case RESULT_SET: if (value == null) { unsetResultSet(); } else { setResultSet((RowSet)value); unsetUpdateCount(); } else { setUpdateCount((Integer)value); unsetBatchUpdateCounts(); } else { setBatchUpdateCounts((List<Integer>)value); unsetProcedureOutParams(); } else { setProcedureOutParams((Map<Integer,ColumnValue>)value); unsetGeneratedKeys(); } else { setGeneratedKeys((RowSet)value); unsetNewDefaultSchema(); } else { setNewDefaultSchema((String)value); unsetWarnings(); } else { setWarnings((SnappyExceptionData)value);
sr.setBatchUpdateCounts(Collections.singletonList(sr.updateCount)); sr = new StatementResult(); sr.setBatchUpdateCounts(ur.getBatchUpdateCounts()); sr.setGeneratedKeys(ur.getGeneratedKeys()); sr.setNewDefaultSchema(ur.getNewDefaultSchema()); sr.setWarnings(ur.getWarnings()); sr.setPreparedResult(prepResult); return sr; } finally {
protected final boolean execute(final String sql, boolean getAutoInc, int[] autoIncColumns, String[] autoIncColumnNames) throws SQLException { checkClosed(); reset(); setAutoIncAttributes(getAutoInc, autoIncColumns, autoIncColumnNames); try { StatementResult sr = this.service.execute(sql, null, getAttributes()); clearPendingTransactionAttrs(); this.warnings = sr.getWarnings(); if (getAutoInc) { this.currentGeneratedKeys = sr.getGeneratedKeys(); } initializeProcedureOutParams(sr); final RowSet rs = sr.getResultSet(); if (rs != null) { setCurrentRowSet(rs); return true; } else { this.currentUpdateCount = sr.getUpdateCount(); return false; } } catch (SnappyException se) { throw ThriftExceptionUtil.newSQLException(se); } }
source.connId, prh.getSQL(), paramsBatch, null, attrs, source.token); prh.updatePrepareResult(sr.getPreparedResult()); attrs.setPossibleDuplicate(false); if (sr.getNewDefaultSchema() != null) { this.currentDefaultSchema = sr.getNewDefaultSchema(); source.connId, source.token, ns, false, null); return new UpdateResult().setUpdateCount(sr.getUpdateCount()) .setGeneratedKeys(sr.getGeneratedKeys()) .setWarnings(sr.getWarnings()); } catch (Throwable t) { attrs.setPossibleDuplicate(false);
source.connId, prh.getSQL(), paramsBatch, null, attrs, source.token); prh.updatePrepareResult(sr.getPreparedResult()); attrs.setPossibleDuplicate(false); if (sr.getNewDefaultSchema() != null) { this.currentDefaultSchema = sr.getNewDefaultSchema(); .setBatchUpdateCounts(sr.getBatchUpdateCounts()) .setGeneratedKeys(sr.getGeneratedKeys()) .setWarnings(sr.getWarnings()); } catch (Throwable t) { attrs.setPossibleDuplicate(false);
source.connId, prh.getSQL(), paramsBatch, null, attrs, source.token); prh.updatePrepareResult(sr.getPreparedResult()); attrs.setPossibleDuplicate(false); if (sr.getNewDefaultSchema() != null) { this.currentDefaultSchema = sr.getNewDefaultSchema(); source.connId, source.token, ns, false, null); RowSet rs = sr.getResultSet(); setSourceConnection(rs); return rs;
attrs.setPossibleDuplicate(false); if (sr.getNewDefaultSchema() != null) { this.currentDefaultSchema = sr.getNewDefaultSchema(); source.token); setSourceConnection(sr); prh.updatePrepareResult(sr.getPreparedResult()); attrs.setPossibleDuplicate(false); if (sr.getNewDefaultSchema() != null) { this.currentDefaultSchema = sr.getNewDefaultSchema();
attrs.setPossibleDuplicate(false); if (sr.getNewDefaultSchema() != null) { this.currentDefaultSchema = sr.getNewDefaultSchema();
@Override protected void initializeProcedureOutParams( StatementResult sr) throws SQLException { final TreeMap<Integer, OutputParameter> outParams = this.outParams; if (outParams != null) { Map<Integer, ColumnValue> outValues; if ((outValues = sr.getProcedureOutParams()) != null && outValues.size() > 0) { setCurrentSource(snappydataConstants.BULK_CLOSE_STATEMENT, statementId, null); this.outParamsPositionInRow = new TIntIntHashMap(outParams.size()); // starting index at 1 since get on TIntIntHashMap will give 0 if absent int outIndex = 1; // copy as a Row to outParamValues which also tracks ClientFinalizer // and makes the getters for all params in parent class uniform this.outParamValues = new Row(); this.outParamValues.initialize(outParams.size()); for (Integer parameterIndex : outParams.keySet()) { ColumnValue outValue = outValues.get(parameterIndex); this.outParamValues.setColumnValue(outIndex - 1, outValue); this.outParamsPositionInRow.put(parameterIndex, outIndex); outIndex++; } // create LOBs in the row, if any this.outParamValues.initializeLobs(this.service); } } }
/** * {@inheritDoc} */ @Override public boolean execute() throws SQLException { checkClosed(); reset(); try { StatementResult sr = this.service.executePrepared( // don't throw exception in getLobSource rather return null and // service will failover to new node and do re-prepare as required getLobSource(false, "executePrepared"), statementId, this.paramsList, getOutputParameters(), this); this.warnings = sr.getWarnings(); if (this.attrs.isRequireAutoIncCols()) { this.currentGeneratedKeys = sr.getGeneratedKeys(); } initializeProcedureOutParams(sr); final RowSet rs = sr.getResultSet(); if (rs != null) { setCurrentRowSet(rs); return true; } else { this.currentUpdateCount = sr.getUpdateCount(); return false; } } catch (SnappyException se) { throw informListeners(ThriftExceptionUtil.newSQLException(se)); } }
attrs.setPossibleDuplicate(false); if (sr.getNewDefaultSchema() != null) { this.currentDefaultSchema = sr.getNewDefaultSchema();
public StatementResult deepCopy() { return new StatementResult(this); }