@Override public SelectorRecord map(ResultSet resultSet, StatementContext context) throws SQLException { return new SelectorRecord( resultSet.getLong("resource_group_id"), resultSet.getLong("priority"), Optional.ofNullable(resultSet.getString("user_regex")).map(Pattern::compile), Optional.ofNullable(resultSet.getString("source_regex")).map(Pattern::compile), Optional.ofNullable(resultSet.getString("query_type")), Optional.ofNullable(resultSet.getString("client_tags")).map(LIST_STRING_CODEC::fromJson), Optional.ofNullable(resultSet.getString("selector_resource_estimate")).map(SELECTOR_RESOURCE_ESTIMATE_JSON_CODEC::fromJson)); } }
@Override public KeyRecord map(int i, ResultSet resultSet, StatementContext statementContext) throws SQLException { return new KeyRecord(resultSet.getLong("id"), resultSet.getString("number"), resultSet.getLong("device_id"), resultSet.getLong("key_id"), resultSet.getString("public_key"), resultSet.getInt("last_resort") == 1); } }
private Long findSchemaForPositionSHA(Connection c, String sha) throws SQLException { PreparedStatement p = c.prepareStatement("SELECT * from `schemas` where position_sha = ?"); p.setString(1, sha); ResultSet rs = p.executeQuery(); if ( rs.next() ) { Long id = rs.getLong("id"); LOGGER.debug("findSchemaForPositionSHA: found schema_id: " + id + " for sha: " + sha); return id; } else { return null; } }
static Migration convertPipelineSelectionsToFilters() { return (cxn) -> { if (!required(cxn)) return; try (Statement s = cxn.createStatement()) { final ResultSet rs = s.executeQuery("SELECT id, selections, isblacklist FROM pipelineselections WHERE version = 0"); while (rs.next()) { perform(cxn, rs.getLong(ID), rs.getString(SELECTIONS), rs.getBoolean(BLACKLIST)); } } }; }
@Override public List<ProjectLogEvent> handle(final ResultSet rs) throws SQLException { if (!rs.next()) { return Collections.emptyList(); } final ArrayList<ProjectLogEvent> events = new ArrayList<>(); do { final int projectId = rs.getInt(1); final int eventType = rs.getInt(2); final long eventTime = rs.getLong(3); final String username = rs.getString(4); final String message = rs.getString(5); final ProjectLogEvent event = new ProjectLogEvent(projectId, ProjectLogEvent.EventType.fromInteger(eventType), eventTime, username, message); events.add(event); } while (rs.next()); return events; } }
private static Long executeScalar(PreparedStatement ps, Long defaultValue) throws SQLException { ResultSet rs = null; try { rs = ps.executeQuery(); if (rs.next()) { return rs.getLong(1); } return defaultValue; } finally { closeQuietly(rs); } }
@Test public void testQueryForLongWithParamBeanWithCollection() throws Exception { given(resultSet.getMetaData()).willReturn(resultSetMetaData); given(resultSet.next()).willReturn(true, false); given(resultSet.getLong(1)).willReturn(87L); BeanPropertySqlParameterSource params = new BeanPropertySqlParameterSource(new ParameterCollectionBean(3, 5)); long l = template.queryForObject("SELECT AGE FROM CUSTMR WHERE ID IN (:ids)", params, Long.class).longValue(); assertEquals("Return of a long", 87, l); verify(connection).prepareStatement("SELECT AGE FROM CUSTMR WHERE ID IN (?, ?)"); verify(preparedStatement).setObject(1, 3); verify(preparedStatement).setObject(2, 5); }
private Long calculateRowCount(Connection connection, String db, String table, String whereClause) throws SQLException { LOGGER.info("counting rows"); String sql = String.format("select count(*) from `%s`.%s", db, table); if ( whereClause != null ) { sql += String.format(" where %s", whereClause); } PreparedStatement preparedStatement = connection.prepareStatement(sql); ResultSet resultSet = preparedStatement.executeQuery(); resultSet.next(); return resultSet.getLong(1); }
static Migration ensureFilterStateIsNotNull() { return (cxn) -> { if (!required(cxn)) return; try (Statement s = cxn.createStatement()) { final ResultSet rs = s.executeQuery("SELECT id, filters FROM pipelineselections WHERE version = 1"); while (rs.next()) { perform(cxn, rs.getLong(ID), rs.getString(FILTERS)); } } }; }
@Override public ColumnDef convert(ResultSet rs) throws SQLException { String nullableText = rs.getString(7); boolean nullable = "YES".equalsIgnoreCase(nullableText); return new ColumnDef( rs.getString(1), rs.getString(2), rs.getString(3), rs.getString(4), rs.getString(5), rs.getLong(6), nullable); } }
private long getNextLobId() throws SQLException { String sql = "SELECT MAX(LOB) FROM " + LOB_MAP; PreparedStatement prep = prepare(sql); ResultSet rs = prep.executeQuery(); rs.next(); long x = rs.getLong(1) + 1; reuse(sql, prep); sql = "SELECT MAX(ID) FROM " + LOBS; prep = prepare(sql); rs = prep.executeQuery(); rs.next(); x = Math.max(x, rs.getLong(1) + 1); reuse(sql, prep); return x; }
@Test public void testQueryForLongWithParamBean() throws Exception { given(resultSet.getMetaData()).willReturn(resultSetMetaData); given(resultSet.next()).willReturn(true, false); given(resultSet.getLong(1)).willReturn(87L); BeanPropertySqlParameterSource params = new BeanPropertySqlParameterSource(new ParameterBean(3)); long l = template.queryForObject("SELECT AGE FROM CUSTMR WHERE ID = :id", params, Long.class).longValue(); assertEquals("Return of a long", 87, l); verify(connection).prepareStatement("SELECT AGE FROM CUSTMR WHERE ID = ?"); verify(preparedStatement).setObject(1, 3, Types.INTEGER); }
private long getInsertedRowsCount(Connection connection, long rowId) throws SQLException, MissingBootstrapRowException { String sql = "select inserted_rows from `bootstrap` where id = ?"; PreparedStatement preparedStatement = connection.prepareStatement(sql); preparedStatement.setLong(1, rowId); ResultSet resultSet = preparedStatement.executeQuery(); if ( resultSet.next() ) { return resultSet.getLong(1); } else { throw new MissingBootstrapRowException(rowId); } }
private Position positionFromResultSet(ResultSet rs) throws SQLException { if ( !rs.next() ) return null; String gtid = gtidMode ? rs.getString("gtid_set") : null; BinlogPosition pos = new BinlogPosition( gtid, null, rs.getLong("binlog_position"), rs.getString("binlog_file") ); return new Position(pos, rs.getLong("last_heartbeat_read")); }
throws SQLException long id = resultSet.getLong("resource_group_id"); ResourceGroupNameTemplate nameTemplate = new ResourceGroupNameTemplate(resultSet.getString("name")); String softMemoryLimit = resultSet.getString("soft_memory_limit"); int maxQueued = resultSet.getInt("max_queued"); Optional<Integer> softConcurrencyLimit = Optional.of(resultSet.getInt("soft_concurrency_limit")); if (resultSet.wasNull()) { softConcurrencyLimit = Optional.empty(); int hardConcurrencyLimit = resultSet.getInt("hard_concurrency_limit"); Optional<String> schedulingPolicy = Optional.ofNullable(resultSet.getString("scheduling_policy")); Optional<Integer> schedulingWeight = Optional.of(resultSet.getInt("scheduling_weight")); if (resultSet.wasNull()) { Optional<String> softCpuLimit = Optional.ofNullable(resultSet.getString("soft_cpu_limit")); Optional<String> hardCpuLimit = Optional.ofNullable(resultSet.getString("hard_cpu_limit")); Optional<Long> parentId = Optional.of(resultSet.getLong("parent")); if (resultSet.wasNull()) { parentId = Optional.empty();
@Test public void testDuplicateColumnLabels() throws Exception { try (ResultSet rs = statement.executeQuery("SELECT 123 x, 456 x")) { ResultSetMetaData metadata = rs.getMetaData(); assertEquals(metadata.getColumnCount(), 2); assertEquals(metadata.getColumnName(1), "x"); assertEquals(metadata.getColumnName(2), "x"); assertTrue(rs.next()); assertEquals(rs.getLong(1), 123L); assertEquals(rs.getLong(2), 456L); assertEquals(rs.getLong("x"), 123L); } }