Signature mockPreparedSignature(String sql) { List<AvaticaParameter> params = new ArrayList<AvaticaParameter>(); int startIndex = 0; while (sql.indexOf("?", startIndex) >= 0) { AvaticaParameter param = new AvaticaParameter(false, 0, 0, 0, null, null, null); params.add(param); startIndex = sql.indexOf("?", startIndex) + 1; } ArrayList<ColumnMetaData> columns = new ArrayList<ColumnMetaData>(); Map<String, Object> internalParams = Collections.<String, Object> emptyMap(); return new Meta.Signature(columns, sql, params, internalParams, CursorFactory.ARRAY, Meta.StatementType.SELECT); }
@SuppressWarnings({ "rawtypes", "unchecked" }) private MetaResultSet createResultSet(List iterable, Class clazz, String... names) { final List<ColumnMetaData> columns = new ArrayList<ColumnMetaData>(); final List<Field> fields = new ArrayList<Field>(); final List<String> fieldNames = new ArrayList<String>(); for (String name : names) { final int index = fields.size(); final String fieldName = AvaticaUtils.toCamelCase(name); final Field field; try { field = clazz.getField(fieldName); } catch (NoSuchFieldException e) { throw new RuntimeException(e); } columns.add(columnMetaData(name, index, field.getType(), true)); fields.add(field); fieldNames.add(fieldName); } CursorFactory cursorFactory = CursorFactory.record(clazz, fields, fieldNames); Signature signature = new Signature(columns, "", null, Collections.<String, Object> emptyMap(), cursorFactory, StatementType.SELECT); StatementHandle sh = this.createStatement(connection().handle); Frame frame = new Frame(0, true, iterable); return MetaResultSet.create(connection().id, sh.id, true, signature, frame); }
private static Signature newSignature(String sql) { return new Signature( new DrillColumnMetaDataList(), sql, Collections.<AvaticaParameter> emptyList(), Collections.<String, Object>emptyMap(), null, // CursorFactory set to null, as SQL requests use DrillCursor Meta.StatementType.SELECT); }
protected MetaResultSet createResultSet( Map<String, Object> internalParameters, List<ColumnMetaData> columns, CursorFactory cursorFactory, Frame firstFrame) { try { final AvaticaStatement statement = connection.createStatement(); final Signature signature = new Signature(columns, "", Collections.<AvaticaParameter>emptyList(), internalParameters, cursorFactory, Meta.StatementType.SELECT); return MetaResultSet.create(connection.id, statement.getId(), true, signature, firstFrame); } catch (SQLException e) { throw new RuntimeException(e); } }
protected MetaResultSet createResultSet( Map<String, Object> internalParameters, List<ColumnMetaData> columns, CursorFactory cursorFactory, Frame firstFrame) { try { final AvaticaStatement statement = connection.createStatement(); final Signature signature = new Signature(columns, "", Collections.<AvaticaParameter>emptyList(), internalParameters, cursorFactory, Meta.StatementType.SELECT); return MetaResultSet.create(connection.id, statement.getId(), true, signature, firstFrame); } catch (SQLException e) { throw new RuntimeException(e); } }
protected MetaResultSet createResultSet( Map<String, Object> internalParameters, List<ColumnMetaData> columns, CursorFactory cursorFactory, Frame firstFrame) { try { final AvaticaStatement statement = connection.createStatement(); final Signature signature = new Signature(columns, "", Collections.<AvaticaParameter>emptyList(), internalParameters, cursorFactory, Meta.StatementType.SELECT); return MetaResultSet.create(connection.id, statement.getId(), true, signature, firstFrame); } catch (SQLException e) { throw new RuntimeException(e); } }
/** Creates a copy of this Signature with null lists and maps converted to * empty. */ public Signature sanitize() { if (columns == null || parameters == null || internalParameters == null || statementType == null) { return new Signature(sanitize(columns), sql, sanitize(parameters), sanitize(internalParameters), cursorFactory, Meta.StatementType.SELECT); } return this; }
protected static Signature signature(ResultSetMetaData metaData, ParameterMetaData parameterMetaData, String sql, Meta.StatementType statementType) throws SQLException { final CursorFactory cf = CursorFactory.LIST; // because JdbcResultSet#frame return new Signature(columns(metaData), sql, parameters(parameterMetaData), null, cf, statementType); }
/** Used by Jackson to create a Signature by de-serializing JSON. */ @JsonCreator public static Signature create( @JsonProperty("columns") List<ColumnMetaData> columns, @JsonProperty("sql") String sql, @JsonProperty("parameters") List<AvaticaParameter> parameters, @JsonProperty("cursorFactory") CursorFactory cursorFactory, @JsonProperty("statementType") Meta.StatementType statementType) { return new Signature(columns, sql, parameters, Collections.<String, Object>emptyMap(), cursorFactory, statementType); }
/** Creates a copy of this Signature with null lists and maps converted to * empty. */ public Signature sanitize() { if (columns == null || parameters == null || internalParameters == null || statementType == null) { return new Signature(sanitize(columns), sql, sanitize(parameters), sanitize(internalParameters), cursorFactory, Meta.StatementType.SELECT); } return this; }
/** Creates a copy of this Signature with null lists and maps converted to * empty. */ public Signature sanitize() { if (columns == null || parameters == null || internalParameters == null || statementType == null) { return new Signature(sanitize(columns), sql, sanitize(parameters), sanitize(internalParameters), cursorFactory, Meta.StatementType.SELECT); } return this; }
private static Signature newSignature(String sql) { return new Signature( new DrillColumnMetaDataList(), sql, Collections.<AvaticaParameter> emptyList(), Collections.<String, Object>emptyMap(), null, // CursorFactory set to null, as SQL requests use DrillCursor Meta.StatementType.SELECT); }
protected static Signature signature(ResultSetMetaData metaData, ParameterMetaData parameterMetaData, String sql, Meta.StatementType statementType) throws SQLException { final CursorFactory cf = CursorFactory.LIST; // because JdbcResultSet#frame return new Signature(columns(metaData), sql, parameters(parameterMetaData), null, cf, statementType); }
/** Returns a copy of this Signature, substituting given CursorFactory. */ public Signature setCursorFactory(CursorFactory cursorFactory) { return new Signature(columns, sql, parameters, internalParameters, cursorFactory, statementType); }
/** Used by Jackson to create a Signature by de-serializing JSON. */ @JsonCreator public static Signature create( @JsonProperty("columns") List<ColumnMetaData> columns, @JsonProperty("sql") String sql, @JsonProperty("parameters") List<AvaticaParameter> parameters, @JsonProperty("cursorFactory") CursorFactory cursorFactory, @JsonProperty("statementType") Meta.StatementType statementType) { return new Signature(columns, sql, parameters, Collections.<String, Object>emptyMap(), cursorFactory, statementType); }
/** Returns a copy of this Signature, substituting given CursorFactory. */ public Signature setCursorFactory(CursorFactory cursorFactory) { return new Signature(columns, sql, parameters, internalParameters, cursorFactory, statementType); }
/** Used by Jackson to create a Signature by de-serializing JSON. */ @JsonCreator public static Signature create( @JsonProperty("columns") List<ColumnMetaData> columns, @JsonProperty("sql") String sql, @JsonProperty("parameters") List<AvaticaParameter> parameters, @JsonProperty("cursorFactory") CursorFactory cursorFactory, @JsonProperty("statementType") Meta.StatementType statementType) { return new Signature(columns, sql, parameters, Collections.<String, Object>emptyMap(), cursorFactory, statementType); }
protected static Signature signature(ResultSetMetaData metaData, ParameterMetaData parameterMetaData, String sql, Meta.StatementType statementType) throws SQLException { final CursorFactory cf = CursorFactory.ARRAY; // because JdbcResultSet#frame return new Signature(columns(metaData), sql, parameters(parameterMetaData), null, cf, statementType); }
static Signature newSignature(String sql) { return new Signature( new DremioColumnMetaDataList(), sql, Collections.<AvaticaParameter> emptyList(), Collections.<String, Object>emptyMap(), null, // CursorFactory set to null, as SQL requests use DremioCursor StatementType.SELECT ); }
/** Returns a copy of this Signature, substituting given CursorFactory. */ public Signature setCursorFactory(CursorFactory cursorFactory) { return new Signature(columns, sql, parameters, internalParameters, cursorFactory, statementType); }