Como obtenho o tamanho de um java.sql.ResultSet?

284

Isso não deveria ser uma operação bastante direta? No entanto, vejo que não há um método size()nem length().

Jake
fonte
11
Eu adoraria saber o motivo dessa omissão.
Slamice
O meu entendimento da questão foi que você quer encontrar o tamanho do conjunto de resultados em bytes, não o número de tuplas ...
DejanLekic
É muito irritante não ter a dimensão correta antes dos dados do processo, mas se você precisar armazená-los em uma matriz, considere usar uma estrutura de dados como List e depois convertê-los em uma matriz com o método toArray ().
AndreaTaroni86

Respostas:

270

Faça uma SELECT COUNT(*) FROM ...consulta em seu lugar.

OU

int size =0;
if (rs != null) 
{
  rs.last();    // moves cursor to the last row
  size = rs.getRow(); // get row id 
}

Em qualquer um dos casos, você não precisará fazer um loop sobre os dados inteiros.

finnw
fonte
8
last () e getRow () não são métodos estáticos na classe ResultSet.
JeeBee
70
Por uma questão de brevidade, sempre faço referência a métodos dessa maneira ao escrever sobre eles para outras pessoas, independentemente de serem estáticos ou não. Na verdade, está implícita a criação de uma instância do objeto e a chamada do método.
Laz
51
Eu escrevo SomeClass.staticMethod () e SomeClass # instanceMethod () para menos confusão.
Jake
9
Como se busca o valor retornado ao executar um select count?
Naftuli Kay
16
ResultSet#last()não funciona em todos os tipos de ResultSetobjetos, você precisa ter certeza de que usa um que seja ResultSet.TYPE_SCROLL_INSENSITIVEouResultSet.TYPE_SCROLL_SENSITIVE
Marius Ion
91
ResultSet rs = ps.executeQuery();
int rowcount = 0;
if (rs.last()) {
  rowcount = rs.getRow();
  rs.beforeFirst(); // not rs.first() because the rs.next() below will move on, missing the first element
}
while (rs.next()) {
  // do your standard per row stuff
}
JeeBee
fonte
5
Dentro do bloco de código if (rs.last ()), o método correto não seria rs.beforeFirst () em vez de rs.first ()? Dessa forma, você não está pulando o primeiro registro em seu conjunto de resultados para processamento no loop while.
karlgrz
não se esqueça de definir o cursor de volta para beforeFirst fora do bloco if?
precisa saber é o seguinte
Como dizem os documentos do ResultSet , getRow()funciona para o TYPE_FORWARD_ONLYResultSets e beforeFirst()gera erros para eles. Esta resposta não está com defeito então?
entanto
5
Isso funciona apenas quando a instrução é criada com a opção sem rolagem:ps=conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);
BullyWiiPlaza 21/02
19

Bem, se você tem um ResultSettipo, ResultSet.TYPE_FORWARD_ONLYdeseja mantê-lo dessa maneira (e não mudar para um ResultSet.TYPE_SCROLL_INSENSITIVEou ResultSet.TYPE_SCROLL_INSENSITIVEpara poder usá-lo .last()).

Sugiro um hack muito agradável e eficiente, no qual você adiciona uma primeira linha falsa / falsa no topo, contendo o número de linhas.

Exemplo

Digamos que sua consulta seja a seguinte

select MYBOOL,MYINT,MYCHAR,MYSMALLINT,MYVARCHAR
from MYTABLE
where ...blahblah...

e sua saída se parece

true    65537 "Hey" -32768 "The quick brown fox"
false  123456 "Sup"    300 "The lazy dog"
false -123123 "Yo"       0 "Go ahead and jump"
false       3 "EVH"    456 "Might as well jump"
...
[1000 total rows]

Basta refatorar seu código para algo como isto:

Statement s=myConnection.createStatement(ResultSet.TYPE_FORWARD_ONLY,
                                         ResultSet.CONCUR_READ_ONLY);
String from_where="FROM myTable WHERE ...blahblah... ";
//h4x
ResultSet rs=s.executeQuery("select count(*)as RECORDCOUNT,"
                           +       "cast(null as boolean)as MYBOOL,"
                           +       "cast(null as int)as MYINT,"
                           +       "cast(null as char(1))as MYCHAR,"
                           +       "cast(null as smallint)as MYSMALLINT,"
                           +       "cast(null as varchar(1))as MYVARCHAR "
                           +from_where
                           +"UNION ALL "//the "ALL" part prevents internal re-sorting to prevent duplicates (and we do not want that)
                           +"select cast(null as int)as RECORDCOUNT,"
                           +       "MYBOOL,MYINT,MYCHAR,MYSMALLINT,MYVARCHAR "
                           +from_where);

Sua saída da consulta agora será algo como

1000 null     null null    null null
null true    65537 "Hey" -32768 "The quick brown fox"
null false  123456 "Sup"    300 "The lazy dog"
null false -123123 "Yo"       0 "Go ahead and jump"
null false       3 "EVH"    456 "Might as well jump"
...
[1001 total rows]

Então você só precisa

if(rs.next())
    System.out.println("Recordcount: "+rs.getInt("RECORDCOUNT"));//hack: first record contains the record count
while(rs.next())
    //do your stuff
Unai Vivi
fonte
Interessante, mas como você geraria dinamicamente / genericamente as primeiras instruções de seleção: cast (nulo como booleano) como MYBOOL, ect? Para isso, serão necessários metadados dos campos e tipos de dados da instrução "select", como booleano, char, int, ect ...) que podem exigir uma viagem extra ao banco de dados que negará todos os benefícios.
user1697575
Isso é útil quando você tem acesso a todos os detalhes de campo e velocidade é a sua principal preocupação (e, portanto, precisa ficar com um jejum ResultSet.TYPE_FORWARD_ONLY)
Unai Vivi
11
int i = 0;
while(rs.next()) {
    i++;
}
Bhaskar
fonte
Não entendo qual é a desvantagem de usar esse método para calcular o tamanho do ResultSet. Isso é ótimo ... sem uso de um parâmetro SQL extra. Por favor, comente sobre este método.
Madeyedexter
5
Desempenho é a palavra-chave aqui. Imagine o seu conjunto de resultados é de 100 milhões de registros, em seguida, você vai ver a questão
Pierre
7
Quero saber o tamanho do conjunto de resultados ANTES de processar os resultados, pois preciso criar uma matriz do mesmo tamanho antes. E, como observado em outras respostas, a varredura de todas as linhas duas vezes nem sempre funciona.
Ivo
11

Eu recebi uma exceção ao usar rs.last()

if(rs.last()){
    rowCount = rs.getRow(); 
    rs.beforeFirst();
}

:

java.sql.SQLException: Invalid operation for forward only resultset

é por padrão que é ResultSet.TYPE_FORWARD_ONLY, o que significa que você só pode usarrs.next()

a solução é:

stmt=conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE,
    ResultSet.CONCUR_READ_ONLY); 
Dan
fonte
12
Mudar de ResultSet.TYPE_FORWARD_ONLYpara ResultSet.TYPE_SCROLL_INSENSITIVEgeralmente incorre em uma enorme penalidade de desempenho.
Unai Vivi
3
Eu testei na minha mesa (10 colunas, 187 392 linhas). Meu teste consultou e carregou todos os elementos na string. Para TYPE_FORWARD_ONLY, levou aproximadamente 1 segundo. Para TYPE_SCROLL_INSENSITIVE, levou aproximadamente 7 segundos. Quando eu usei SELECT COUNT(*) FROM default_tblantes do SELECT COUNT(*) FROM default_tblque levou menos de 1,5 segundo. Eu testei no banco de dados Derby Embedded 10.11.1.1
Vit Bernatik
4

A maneira de obter o tamanho do ResultSet, Não há necessidade de usar ArrayList etc

int size =0;  
if (rs != null)   
{  
rs.beforeFirst();  
 rs.last();  
size = rs.getRow();
}

Agora você obterá tamanho. Se desejar imprimir o ResultSet, antes de imprimir, use também a seguinte linha de código,

rs.beforeFirst();  
Anptk
fonte
4

[Consideração de velocidade]

Muitas pessoas aqui sugerem, ResultSet.last()mas para isso você precisaria abrir a conexão, ResultSet.TYPE_SCROLL_INSENSITIVEpois o banco de dados incorporado do Derby é até 10 vezes MAIS LENTO que o ResultSet.TYPE_FORWARD_ONLY.

De acordo com meus microtestes para bancos de dados Derby e H2 incorporados, é significativamente mais rápido ligar SELECT COUNT(*)antes do seu SELECT.

Aqui está mais detalhadamente meu código e meus benchmarks

Vit Bernatik
fonte
3

É uma maneira simples de fazer contagem de linhas.

ResultSet rs = job.getSearchedResult(stmt);
int rsCount = 0;

//but notice that you'll only get correct ResultSet size after end of the while loop
while(rs.next())
{
    //do your other per row stuff 
    rsCount = rsCount + 1;
}//end while
CounterSpell
fonte
4
Sim, isso funciona. Mas acho que o OP luta para saber o número de linhas antes de realmente processá-las. Razões da vida real eu teria que lutar contra este problema até agora: 1.) paginação de linhas recorde 2.) mostrando as linhas processadas em tarefas de longa execução para fins de monitoramento do progresso ...
ppeterka
Pré-alocar o tamanho da estrutura de dados é outro motivo. Já vi muitas bibliotecas retornarem 10 listas de elementos quando há apenas um valor, porque os desenvolvedores tiveram o mesmo problema com o ResultSet.
Joseph Lust
2
String sql = "select count(*) from message";
ps =  cn.prepareStatement(sql);

rs = ps.executeQuery();
int rowCount = 0;
while(rs.next()) {
    rowCount = Integer.parseInt(rs.getString("count(*)"));
    System.out.println(Integer.parseInt(rs.getString("count(*)")));
}
System.out.println("Count : " + rowCount);
Peter.Chu
fonte
1

Eu verifiquei o valor do tempo de execução da interface ResultSet e descobri que era praticamente um ResultSetImpl o tempo todo. ResultSetImpl tem um método chamado getUpdateCount()que retorna o valor que você está procurando.

Este exemplo de código deve ser suficiente:
ResultSet resultSet = executeQuery(sqlQuery);
double rowCount = ((ResultSetImpl)resultSet).getUpdateCount()

Sei que o downcasting geralmente é um procedimento inseguro, mas esse método ainda não me falhou.

clausavram
fonte
2
Não está funcionando com Tomcat / MySQL:java.lang.ClassCastException: org.apache.tomcat.dbcp.dbcp.DelegatingResultSet cannot be cast to com.mysql.jdbc.ResultSetImpl
Panu Haaramo
1

Hoje, usei essa lógica porque não sei obter a contagem do RS.

int chkSize = 0;
if (rs.next()) {
    do {  ..... blah blah
        enter code here for each rs.
        chkSize++;
    } while (rs.next());
} else {
    enter code here for rs size = 0 
}
// good luck to u.
Parksangdonews
fonte
0
theStatement=theConnection.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

ResultSet theResult=theStatement.executeQuery(query); 

//Get the size of the data returned
theResult.last();     
int size = theResult.getRow() * theResult.getMetaData().getColumnCount();       
theResult.beforeFirst();
Ben
fonte
0

Eu estava tendo o mesmo problema. Usando ResultSet.first()desta maneira logo após a execução resolvê-lo:

if(rs.first()){
    // Do your job
} else {
    // No rows take some actions
}

Documentação ( link ):

boolean first()
    throws SQLException

Move o cursor para a primeira linha deste ResultSetobjeto.

Devoluções:

truese o cursor estiver em uma linha válida; falsese não houver linhas no conjunto de resultados

Lances:

SQLException- se ocorrer um erro de acesso ao banco de dados; esse método é chamado em um conjunto de resultados fechado ou o tipo de conjunto de resultados éTYPE_FORWARD_ONLY

SQLFeatureNotSupportedException - se o driver JDBC não suportar este método

Desde a:

1.2.

Israel Hernández
fonte
0

A abordagem mais fácil, execute a consulta Count Count (*), faça resultSet.next () para apontar para a primeira linha e, em seguida, faça resultSet.getString (1) para obter a contagem. Código:

ResultSet rs = statement.executeQuery("Select Count(*) from your_db");
if(rs.next()) {
   int count = rs.getString(1).toInt()
}
user7120462
fonte
-1

Atribua um nome à coluna ..

String query = "SELECT COUNT(*) as count FROM

Faça referência a essa coluna do objeto ResultSet para um int e faça sua lógica a partir daí.

PreparedStatement statement = connection.prepareStatement(query);
statement.setString(1, item.getProductId());
ResultSet resultSet = statement.executeQuery();
while (resultSet.next()) {
    int count = resultSet.getInt("count");
    if (count >= 1) {
        System.out.println("Product ID already exists.");
    } else {
        System.out.println("New Product ID.");
    }
}
ReMaX
fonte