Cancelamento de instruções

A maneira recomendada para cancelar uma instrução é usar a interface do aplicativo no qual a consulta está sendo executada (por exemplo, a planilha na interface da Web do Snowflake) ou a API de cancelamento fornecida pelo driver ODBC ou JDBC do Snowflake. No entanto, em alguns casos, é necessário cancelar uma consulta usando SQL.

O Snowflake fornece as seguintes funções para oferecer suporte a o uso de SQL para cancelar instruções em execução/ativas:

Exemplo

O seguinte código Java de exemplo usa SYSTEM$CANCEL_ALL_QUERIES e outras funções do Snowflake para cancelar uma instrução em execução na sessão atual após 5 segundos:

  1. O código de exemplo primeiro emite um comando SQL para CURRENT_SESSION para obter o identificador da sessão.

  2. Em seguida, cria uma tarefa a ser executada 5 segundos depois. Esta tarefa usa o identificador da sessão como parâmetro para SYSTEM$CANCEL_ALL_QUERIES.

  3. Em seguida, uma instrução de longo prazo é executada usando a função de tabela GENERATOR para gerar linhas por 120 segundos.

public void testCancelQuery() throws IOException, SQLException
{
  Statement         statement         = null;
  ResultSet         resultSet         = null;
  ResultSetMetaData resultSetMetaData = null;
  final Connection  connection        = getConnection(true);
  try
  {
    // Get the current session identifier
    Statement getSessionIdStmt = connection.createStatement();
    resultSet                  = getSessionIdStmt.executeQuery("SELECT current_session()");
    resultSetMetaData          = resultSet.getMetaData();
    assertTrue(resultSet.next());
    final int sessionId = resultSet.getInt(1);

    // Use Timer to cancel all queries of session in 5 seconds
    Timer timer = new Timer();
    timer.schedule( new TimerTask()
    {
      @Override
      public void run()
      {
        try
        {
          // Cancel all queries on session
          PreparedStatement cancelAll;
          cancelAll = connection.prepareStatement(
                                    "call system$cancel_all_queries(?)");

          // bind the session identifier as first argument
          cancelAll.setInt(1, sessionId);
          cancelAll.executeQuery();
        }
        catch (SQLException ex)
        {
          logger.log(Level.SEVERE, "Cancel failed with exception {}", ex);
        }
      }
    }, 5000);

    // Use the internal row generator to execute a query for 120 seconds
    statement = connection.createStatement();
    resultSet = statement.executeQuery(
                   "SELECT count(*) FROM TABLE(generator(timeLimit => 120))");
    resultSetMetaData = resultSet.getMetaData();
    statement.close();
  }
  catch (SQLException ex)
  {
    // assert the sqlstate is what we expect (QUERY CANCELLED)
    assertEquals("sqlstate mismatch",
                 SqlState.QUERY_CANCELED, ex.getSQLState());
  }
  catch (Throwable ex)
  {
    logger.log(Level.SEVERE, "Test failed with exception: ", ex);
  }
  finally
  {
    if (resultSet != null)
      resultSet.close();
    if (statement != null)
      statement.close();
    // close connection
    if (connection != null)
      connection.close();
  }
}