ResultSet rs = stmt.executeQuery("SELECT a, b, c FROM TABLE2"); ResultSetMetaData rsmd = rs.getMetaData(); String name = rsmd.getColumnName(1);
ResultSet rs = stmt.executeQuery("SELECT a, b, c FROM TABLE2"); ResultSetMetaData rsmd = rs.getMetaData(); int columnCount = rsmd.getColumnCount(); // The column count starts from 1 for (int i = 1; i <= columnCount; i++ ) { String name = rsmd.getColumnName(i); // Do stuff with name }
ResultSetMetaData metaData = resultSet.getMetaData(); int count = metaData.getColumnCount(); for (int i = 1; i <= count; i++) { if (metaData.getColumnName(i).equals(desiredColumnName)) { // Whatever you want to do here. } }
public List<Map> selectByParams(String sql, List params) throws SQLException { List<Map> list = new ArrayList<> (); int index = 1; pstmt = conn.prepareStatement(sql); if (null != params && !params.isEmpty()) { for (int i = 0; i < params.size(); i ++) { pstmt.setObject(index++, params.get(i)); } } rs = pstmt.executeQuery(); ResultSetMetaData metaData = rs.getMetaData(); int colsLen = metaData.getColumnCount(); while (rs.next()) { Map map = new HashMap(colsLen); for (int i = 0; i < colsLen; i ++) { String columnName = metaData.getColumnName(i + 1); Object columnValue = rs.getObject(columnName); if (null == columnValue) { columnValue = ""; } map.put(columnName, columnValue); } list.add(map); } return list; }
public void onAfter(Connection conn, ResultSet rs, Statement stmt) throws SQLException { if (rs != null && rs.next()) { String colName = rs.getMetaData().getColumnName(1); Object obj = entityField.getAdaptor().get(rs, colName); entityField.setValue(this.getOperatingObject(), obj); } }
String column = rsMeta.getColumnName(col + 1); while (pks.next()) { if (column.equalsIgnoreCase( pks.getString("COLUMN_NAME"))) { return (primaryKeys[col] = new Boolean(true)).booleanValue();
@Test public void testConvertToAvroStreamForShort() throws SQLException, IOException { final ResultSetMetaData metadata = mock(ResultSetMetaData.class); when(metadata.getColumnCount()).thenReturn(1); when(metadata.getColumnType(1)).thenReturn(Types.TINYINT); when(metadata.getColumnName(1)).thenReturn("t_int"); when(metadata.getTableName(1)).thenReturn("table"); final ResultSet rs = resultSetReturningMetadata(metadata); final short s = 25; when(rs.getObject(Mockito.anyInt())).thenReturn(s); final InputStream instream = convertResultSetToAvroInputStream(rs); final DatumReader<GenericRecord> datumReader = new GenericDatumReader<>(); try (final DataFileStream<GenericRecord> dataFileReader = new DataFileStream<>(instream, datumReader)) { GenericRecord record = null; while (dataFileReader.hasNext()) { record = dataFileReader.next(record); assertEquals(Short.toString(s), record.get("t_int").toString()); } } }
public List resultSetToArrayList(ResultSet rs) throws SQLException{ ResultSetMetaData md = rs.getMetaData(); int columns = md.getColumnCount(); ArrayList list = new ArrayList(50); while (rs.next()){ HashMap row = new HashMap(columns); for(int i=1; i<=columns; ++i){ row.put(md.getColumnName(i),rs.getObject(i)); } list.add(row); } return list; }
public static void createColumnIfNotExist(Connection conn) throws SQLException { ResultSetMetaData metaData = conn.createStatement().executeQuery("SELECT * FROM STAGES").getMetaData(); for (int i = 1; i <= metaData.getColumnCount(); i++) { String columnName = metaData.getColumnName(i); if (columnName.toLowerCase().equals("artifactsdeleted")) { return; } } conn.createStatement().executeUpdate("ALTER TABLE STAGES ADD COLUMN `ARTIFACTSDELETED` Boolean DEFAULT FALSE NOT NULL"); } }
@Test public void testPrepareStatementWithMockKylinClient() throws SQLException, IOException { String sql = "select 1 as val"; // mock client when(client.executeQuery(anyString(), Mockito.<List<Object>>any(), Mockito.<Map<String, String>>any())).thenReturn(getMockResult()); try (KylinConnection conn = getConnectionWithMockClient()) { PreparedStatement preparedStatement = conn.prepareStatement(sql); try (ResultSet resultSet = preparedStatement.executeQuery()) { verify(client).executeQuery(eq(sql), Mockito.<List<Object>>any(), Mockito.<Map<String, String>>any()); assertTrue(resultSet.next()); ResultSetMetaData metaData = resultSet.getMetaData(); assertEquals("VAL", metaData.getColumnName(1)); assertEquals(1, resultSet.getInt("VAL")); } } }
@Test public void testConvertToAvroStreamForNumbers() throws SQLException, IOException { final ResultSetMetaData metadata = mock(ResultSetMetaData.class); when(metadata.getColumnCount()).thenReturn(1); when(metadata.getColumnType(1)).thenReturn(testParams.sqlType); when(metadata.isSigned(1)).thenReturn(testParams.signed); when(metadata.getPrecision(1)).thenReturn(testParams.precision); when(metadata.getColumnName(1)).thenReturn("t_int"); when(metadata.getTableName(1)).thenReturn("table"); final ResultSet rs = resultSetReturningMetadata(metadata); final int ret = 0; when(rs.getObject(Mockito.anyInt())).thenReturn(ret); final InputStream instream = convertResultSetToAvroInputStream(rs); final DatumReader<GenericRecord> datumReader = new GenericDatumReader<>(); try (final DataFileStream<GenericRecord> dataFileReader = new DataFileStream<>(instream, datumReader)) { GenericRecord record = null; while (dataFileReader.hasNext()) { record = dataFileReader.next(record); assertEquals(Integer.toString(ret), record.get("t_int").toString()); } } } }
ResultSetMetaData rsMetaData = rs.getMetaData(); int numberOfColumns = rsMetaData.getColumnCount(); // get the column names; column indexes start from 1 for (int i = 1; i < numberOfColumns + 1; i++) { String columnName = rsMetaData.getColumnName(i); // Get the name of the column's table name if ("theColumn".equals(columnName)) { System.out.println("Bingo!"); } }
ResultSet resultSet = statement.executeQuery("SELECT * from foo"); ResultSetMetaData rsmd = resultSet.getMetaData(); int columnsNumber = rsmd.getColumnCount(); while (resultSet.next()) { for (int i = 1; i <= columnsNumber; i++) { if (i > 1) System.out.print(", "); String columnValue = resultSet.getString(i); System.out.print(columnValue + " " + rsmd.getColumnName(i)); } System.out.println(""); }
public static boolean hasColumn(ResultSet rs, String columnName) throws SQLException { ResultSetMetaData rsmd = rs.getMetaData(); int columns = rsmd.getColumnCount(); for (int x = 1; x <= columns; x++) { if (columnName.equals(rsmd.getColumnName(x))) { return true; } } return false; }
@Test public void testPrepareStatementWithMockHttp() throws IOException, SQLException { String sql = "select 1 as val"; try (KylinConnection connection = getConnectionWithMockHttp()) { // mock http HttpResponse response = TestUtil.mockHttpResponseWithFile(200, "OK", "query.json"); when(httpClient.execute(any(HttpUriRequest.class))).thenReturn(response); try (ResultSet resultSet = connection.prepareStatement(sql).executeQuery()) { assertTrue(resultSet.next()); ResultSetMetaData metaData = resultSet.getMetaData(); assertEquals("VAL", metaData.getColumnName(1)); assertEquals(1, resultSet.getInt("VAL")); } } }
@Test public void testConvertToAvroStreamForUnsignedIntegerWithPrecision1ReturnedAsLong_NIFI5612() throws SQLException, IOException { final String mockColumnName = "t_int"; final ResultSetMetaData metadata = mock(ResultSetMetaData.class); when(metadata.getColumnCount()).thenReturn(1); when(metadata.getColumnType(1)).thenReturn(Types.INTEGER); when(metadata.isSigned(1)).thenReturn(false); when(metadata.getPrecision(1)).thenReturn(1); when(metadata.getColumnName(1)).thenReturn(mockColumnName); when(metadata.getTableName(1)).thenReturn("table"); final ResultSet rs = resultSetReturningMetadata(metadata); final Long ret = 0L; when(rs.getObject(Mockito.anyInt())).thenReturn(ret); final InputStream instream = convertResultSetToAvroInputStream(rs); final DatumReader<GenericRecord> datumReader = new GenericDatumReader<>(); try (final DataFileStream<GenericRecord> dataFileReader = new DataFileStream<>(instream, datumReader)) { GenericRecord record = null; while (dataFileReader.hasNext()) { record = dataFileReader.next(record); assertEquals(Long.toString(ret), record.get(mockColumnName).toString()); } } }
@Override protected void resultSetOpenAfter(ResultSetProxy resultSet) { if (resultSetOpenAfterLogEnable && isResultSetLogEnabled()) { try { StringBuffer buf = new StringBuffer(); buf.append("{conn-"); buf.append(resultSet.getStatementProxy().getConnectionProxy().getId()); buf.append(", "); buf.append(stmtId(resultSet)); buf.append(", rs-"); buf.append(resultSet.getId()); buf.append("}"); String resultId = buf.toString(); resultSetLog(resultId + " open"); buf.append(" Header: ["); ResultSetMetaData meta = resultSet.getMetaData(); for (int i = 0, size = meta.getColumnCount(); i < size; ++i) { if (i != 0) { buf.append(", "); } buf.append(meta.getColumnName(i + 1)); } buf.append("]"); resultSetLog(buf.toString()); } catch (SQLException ex) { resultSetLogError("logging error", ex); } } }
public List<HashMap<String,Object>> convertResultSetToList(ResultSet rs) throws SQLException { ResultSetMetaData md = rs.getMetaData(); int columns = md.getColumnCount(); List<HashMap<String,Object>> list = new ArrayList<HashMap<String,Object>>(); while (rs.next()) { HashMap<String,Object> row = new HashMap<String, Object>(columns); for(int i=1; i<=columns; ++i) { row.put(md.getColumnName(i),rs.getObject(i)); } list.add(row); } return list; }
private Column[] getColumnsForResultSet(Table table, ResultSet rs) throws SQLException { ResultSetMetaData metaData = rs.getMetaData(); Column[] columns = new Column[metaData.getColumnCount()]; for(int i = 0; i < columns.length; i++) { columns[i] = table.columnWithName(metaData.getColumnName(i + 1)); } return columns; }
private HashMap<Long, HashMap<String, Object>> buildSchemaMap(Connection conn) throws SQLException { HashMap<Long, HashMap<String, Object>> schemas = new HashMap<>(); PreparedStatement p = conn.prepareStatement("SELECT * from `schemas`"); ResultSet rs = p.executeQuery(); ResultSetMetaData md = rs.getMetaData(); while ( rs.next() ) { HashMap<String, Object> row = new HashMap<>(); for ( int i = 1; i <= md.getColumnCount(); i++ ) row.put(md.getColumnName(i), rs.getObject(i)); schemas.put(rs.getLong("id"), row); } rs.close(); return schemas; }