@Override public OUT nextRecord(OUT reuse) throws IOException { final Row item = resultSet.one(); for (int i = 0; i < reuse.getArity(); i++) { reuse.setField(item.getObject(i), i); } return reuse; } }
@Override public List<List<Values>> map(Session session, Statement statement, ITuple tuple) { List<List<Values>> list = new ArrayList<>(); List<Values> innerList = new LinkedList<>(); list.add(innerList); ResultSet resultSet = session.execute(statement); for (Row row : resultSet) { final Values values = new Values(); for (String field : outputDeclaredFields) { if (tuple.contains(field)) { values.add(tuple.getValueByField(field)); } else { values.add(row.getObject(field)); } } innerList.add(values); } return list; } }
@Override public void success(Integer index, ResultSet resultSet) { if (outputDeclaredFields != null) { List<Values> thisResult = new ArrayList<>(); for (Row row : resultSet) { final Values values = new Values(); for (String field : outputDeclaredFields) { ITuple tuple = tuples.get(index); if (tuple.contains(field)) { values.add(tuple.getValueByField(field)); } else { values.add(row.getObject(field)); } } thisResult.add(values); } results.set(index, thisResult); } }
protected List<Values> handleResult(ResultSet resultSet, ITuple tuple) { List<Values> list = new ArrayList<>(); for (Row row : resultSet) { final Values values = new Values(); for (String field : outputDeclaredFields) { if (tuple.contains(field)) { values.add(tuple.getValueByField(field)); } else { values.add(row.getObject(field)); } } list.add(values); } return list; }
private void assertPartialRow(Row row) { assertThat(row.getUUID(0)).isEqualTo(uuid); assertThat(row.getObject(0)).isEqualTo(uuid); assertThat(row.get(0, UUID.class)).isEqualTo(uuid); assertThat(row.getString(1)).isEqualTo("John Doe"); assertThat(row.getObject(1)).isEqualTo("John Doe"); assertThat(row.get(1, String.class)).isEqualTo("John Doe"); assertThat(row.getTupleValue(2)).isEqualTo(partialLocationValueRetrieved); assertThat(row.getObject(2)).isEqualTo(partialLocationValueRetrieved); assertThat(row.get(2, TupleValue.class)).isEqualTo(partialLocationValueRetrieved); }
private void assertRow(Row row) { assertThat(row.getUUID(0)).isEqualTo(uuid); assertThat(row.getObject(0)).isEqualTo(uuid); assertThat(row.get(0, UUID.class)).isEqualTo(uuid); assertThat(row.getString(1)).isEqualTo("John Doe"); assertThat(row.getObject(1)).isEqualTo("John Doe"); assertThat(row.get(1, String.class)).isEqualTo("John Doe"); assertThat(row.getTupleValue(2)).isEqualTo(locationValue); assertThat(row.getObject(2)).isEqualTo(locationValue); assertThat(row.get(2, TupleValue.class)).isEqualTo(locationValue); }
private void assertRow(Row row) { assertThat(row.getUUID(0)).isEqualTo(uuid); assertThat(row.getObject(0)).isEqualTo(uuid); assertThat(row.get(0, UUID.class)).isEqualTo(uuid); assertThat(row.getString(1)).isEqualTo("John Doe"); assertThat(row.getObject(1)).isEqualTo("John Doe"); assertThat(row.get(1, String.class)).isEqualTo("John Doe"); assertThat(row.getUDTValue(2)).isEqualTo(addressValue); assertThat(row.getObject(2)).isEqualTo(addressValue); assertThat(row.get(2, UDTValue.class)).isEqualTo(addressValue); }
private void assertRow(Row row) { assertThat(row.getInt(0)).isEqualTo(42); assertThat(row.getObject(0)).isEqualTo(42); // uses the default codec assertThat(row.get(0, Integer.class)).isEqualTo(42); assertThat(row.get(0, String.class)).isEqualTo("42"); assertThat(row.getList(1, Integer.class)).isEqualTo(newArrayList(42)); assertThat(row.getList(1, String.class)).isEqualTo(newArrayList("42")); assertThat(row.getObject(1)).isEqualTo(newArrayList(42)); // uses the default codec assertThat(row.get(1, TypeTokens.listOf(Integer.class))).isEqualTo(newArrayList(42)); assertThat(row.get(1, TypeTokens.listOf(String.class))).isEqualTo(newArrayList("42")); }
private void assertRow(Row row) { assertThat(row.getInt(0)).isEqualTo(n_int); assertThat(row.getLong(1)).isEqualTo(n_bigint); assertThat(row.getFloat(2)).isEqualTo(n_float); assertThat(row.getDouble(3)).isEqualTo(n_double); assertThat(row.getVarint(4)).isEqualTo(n_varint); assertThat(row.getDecimal(5)).isEqualTo(n_decimal); // with getObject assertThat(row.getObject(0)).isEqualTo(n_int); assertThat(row.getObject(1)).isEqualTo(n_bigint); assertThat(row.getObject(2)).isEqualTo(n_float); assertThat(row.getObject(3)).isEqualTo(n_double); assertThat(row.getObject(4)).isEqualTo(n_varint); assertThat(row.getObject(5)).isEqualTo(n_decimal); // with get + type assertThat(row.get(0, Integer.class)).isEqualTo(n_int); assertThat(row.get(1, Long.class)).isEqualTo(n_bigint); assertThat(row.get(2, Float.class)).isEqualTo(n_float); assertThat(row.get(3, Double.class)).isEqualTo(n_double); assertThat(row.get(4, BigInteger.class)).isEqualTo(n_varint); assertThat(row.get(5, BigDecimal.class)).isEqualTo(n_decimal); } }
private static CassandraBackendEntry row2Entry(HugeType type, Row row) { CassandraBackendEntry entry = new CassandraBackendEntry(type); List<Definition> cols = row.getColumnDefinitions().asList(); for (Definition col : cols) { String name = col.getName(); Object value = row.getObject(name); entry.column(CassandraTable.parseKey(name), value); } return entry; }
private void assertRow(Row row) { assertThat(row.getInt(0)).isEqualTo(n_int); assertThat(row.getList(1, Integer.class)).isEqualTo(l_int); assertThat(row.getList(2, Long.class)).isEqualTo(l_bigint); assertThat(row.getSet(3, Float.class)).isEqualTo(s_float); assertThat(row.getSet(4, Double.class)).isEqualTo(s_double); assertThat(row.getMap(5, Integer.class, BigInteger.class)).isEqualTo(m_varint); assertThat(row.getMap(6, Integer.class, BigDecimal.class)).isEqualTo(m_decimal); // with get + type assertThat(row.get(1, TypeTokens.listOf(Integer.class))).isEqualTo(l_int); assertThat(row.get(2, TypeTokens.listOf(Long.class))).isEqualTo(l_bigint); assertThat(row.get(3, TypeTokens.setOf(Float.class))).isEqualTo(s_float); assertThat(row.get(4, TypeTokens.setOf(Double.class))).isEqualTo(s_double); assertThat(row.get(5, TypeTokens.mapOf(Integer.class, BigInteger.class))).isEqualTo(m_varint); assertThat(row.get(6, TypeTokens.mapOf(Integer.class, BigDecimal.class))).isEqualTo(m_decimal); // with getObject assertThat(row.getObject(1)).isEqualTo(l_int); assertThat(row.getObject(2)).isEqualTo(l_bigint); assertThat(row.getObject(3)).isEqualTo(s_float); assertThat(row.getObject(4)).isEqualTo(s_double); assertThat(row.getObject(5)).isEqualTo(m_varint); assertThat(row.getObject(6)).isEqualTo(m_decimal); } }
assertThat(row.getDecimal(5)).isEqualTo(n_decimal); assertThat(row.getObject(0)).isEqualTo(n_int); assertThat(row.getObject(1)).isEqualTo(n_bigint); assertThat(row.getObject(2)).isEqualTo(n_float); assertThat(row.getObject(3)).isEqualTo(n_double); assertThat(row.getObject(4)).isEqualTo(n_varint); assertThat(row.getObject(5)).isEqualTo(n_decimal);
Row row = rows.one(); assertThat(row.getUUID(0)).isEqualTo(uuid); assertThat(row.getObject(0)).isEqualTo(uuid); assertThat(row.get(0, UUID.class)).isEqualTo(uuid); assertThat(row.getString(1)).isEqualTo("John Doe"); assertThat(row.getObject(1)).isEqualTo("John Doe"); assertThat(row.get(1, String.class)).isEqualTo("John Doe"); assertThat(row.getTupleValue(2)).isEqualTo(locationValue); assertThat(row.getObject(2)).isEqualTo(location); assertThat(row.get(2, TupleValue.class)).isEqualTo(locationValue); assertThat(row.get(2, Location.class)).isEqualTo(location);
Row row = rows.one(); assertThat(row.getUUID(0)).isEqualTo(uuid); assertThat(row.getObject(0)).isEqualTo(uuid); assertThat(row.get(0, UUID.class)).isEqualTo(uuid); assertThat(row.getString(1)).isEqualTo("John Doe"); assertThat(row.getObject(1)).isEqualTo("John Doe"); assertThat(row.get(1, String.class)).isEqualTo("John Doe"); assertThat(row.getTupleValue(2)).isEqualTo(locationType.newValue(37.387224f, null)); assertThat(row.getObject(2)).isEqualTo(partialLocation); assertThat(row.get(2, TupleValue.class)).isEqualTo(locationType.newValue(37.387224f, null)); assertThat(row.get(2, Location.class)).isEqualTo(partialLocation);
@Override public Object getObject(int i) { return row.getObject(i); }
Row row = rows.one(); assertThat(row.getUUID(0)).isEqualTo(uuid); assertThat(row.getObject(0)).isEqualTo(uuid); assertThat(row.get(0, UUID.class)).isEqualTo(uuid); assertThat(row.getString(1)).isEqualTo("John Doe"); assertThat(row.getObject(1)).isEqualTo("John Doe"); assertThat(row.get(1, String.class)).isEqualTo("John Doe"); assertThat(row.getUDTValue(2)).isEqualTo(addressValue); assertThat(row.getObject(2)).isEqualTo(address); assertThat(row.get(2, UDTValue.class)).isEqualTo(addressValue); assertThat(row.get(2, Address.class)).isEqualTo(address);
@Override public Tuple apply(Row input) { Object[] cols = new Object[input.getColumnDefinitions().size()]; for (int i = 0; i < input.getColumnDefinitions().size(); i++) { cols[i] = input.getObject(i); } return new Tuple(cols); } };
@Override public Map<String, Object> convert(Row row) { ColumnDefinitions cols = row.getColumnDefinitions(); Map<String, Object> map = new HashMap<>(cols.size()); cols.asList().stream().map(Definition::getName) .forEach(name -> map.put(name, row.isNull(name) ? null : row.getObject(name))); return map; } }
private static CassandraBackendEntry row2Entry(HugeType type, Row row) { CassandraBackendEntry entry = new CassandraBackendEntry(type); List<Definition> cols = row.getColumnDefinitions().asList(); for (Definition col : cols) { String name = col.getName(); Object value = row.getObject(name); entry.column(CassandraTable.parseKey(name), value); } return entry; }