@Override public int getNumColumns() { return resultSet.getNumColumns(); }
public static void toJson(ResultSet obj, java.util.Map<String, Object> json) { if (obj.getColumnNames() != null) { JsonArray array = new JsonArray(); obj.getColumnNames().forEach(item -> array.add(item)); json.put("columnNames", array); } if (obj.getNext() != null) { json.put("next", obj.getNext().toJson()); } json.put("numColumns", obj.getNumColumns()); json.put("numRows", obj.getNumRows()); if (obj.getOutput() != null) { json.put("output", obj.getOutput()); } if (obj.getResults() != null) { JsonArray array = new JsonArray(); obj.getResults().forEach(item -> array.add(item)); json.put("results", array); } if (obj.getRows() != null) { JsonArray array = new JsonArray(); obj.getRows().forEach(item -> array.add(item)); json.put("rows", array); } } }
@Test public void testResultSet() { assertEquals(numRows, rs.getNumRows()); assertEquals(columnNames.size(), rs.getNumColumns()); assertEquals(columnNames.size(), rs.getColumnNames().size()); assertEquals(columnNames, rs.getColumnNames()); assertEquals(results, rs.getResults()); List<JsonObject> rows = rs.getRows(); assertEquals(numRows, rs.getRows().size()); int index = 0; for (JsonObject row: rows) { JsonArray result = results.get(index); assertEquals(columnNames.size(), row.size()); assertEquals(row.size(), result.size()); for (int i = 0; i < columnNames.size(); i++) { String columnName = columnNames.get(i); String columnValue = result.getString(i); assertEquals(columnValue, row.getString(columnName)); } index++; } }