cvs commit: jakarta-taglibs/jdbc/src/org/apache/taglibs/jdbc/statement RowCountTag.java

[morgand]

morgand     01/03/06 13:21:07

  Added:       jdbc/src/org/apache/taglibs/jdbc/statement RowCountTag.java
  Log:
  rowCount for adding up rows in a result set
  
  Revision  Changes    Path
  1.1                  
jakarta-taglibs/jdbc/src/org/apache/taglibs/jdbc/statement/RowCountTag.java
  
  Index: RowCountTag.java
  ===================================================================
  /*
   *
   * ====================================================================
   *
   * The Apache Software License, Version 1.1
   *
   * Copyright (c) 1999 The Apache Software Foundation.  All rights
   * reserved.
   *
   * Redistribution and use in source and binary forms, with or without
   * modification, are permitted provided that the following conditions
   * are met:
   *
   * 1. Redistributions of source code must retain the above copyright
   *    notice, this list of conditions and the following disclaimer.
   *
   * 2. Redistributions in binary form must reproduce the above copyright
   *    notice, this list of conditions and the following disclaimer in
   *    the documentation and/or other materials provided with the
   *    distribution.
   *
   * 3. The end-user documentation included with the redistribution, if
   *    any, must include the following acknowlegement:
   *       "This product includes software developed by the
   *        Apache Software Foundation (http://www.apache.org/)."
   *    Alternately, this acknowlegement may appear in the software itself,
   *    if and wherever such third-party acknowlegements normally appear.
   *
   * 4. The names "The Jakarta Project", "Tomcat", and "Apache Software
   *    Foundation" must not be used to endorse or promote products derived
   *    from this software without prior written permission. For written
   *    permission, please contact [EMAIL PROTECTED]
   *
   * 5. Products derived from this software may not be called "Apache"
   *    nor may "Apache" appear in their names without prior written
   *    permission of the Apache Group.
   *
   * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
   * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
   * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
   * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
   * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
   * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
   * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
   * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
   * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
   * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   * SUCH DAMAGE.
   * ====================================================================
   *
   * This software consists of voluntary contributions made by many
   * individuals on behalf of the Apache Software Foundation.  For more
   * information on the Apache Software Foundation, please see
   * <http://www.apache.org/>.
   *
   */
  package org.apache.taglibs.jdbc.statement;
  
  import java.io.IOException;
  
  import javax.servlet.jsp.JspTagException;
  import javax.servlet.jsp.tagext.TagSupport;
  
  /**
   * <p>Prints out the number of rows retrieved from the database.
   * It can be used inside a ResultSet tag to provide a running
   * count of rows retreived, or after the ResultSet tag to
   * display the total number.  Using the tag before the ResultSet
   * or outside of a Statement or PreparedStatement will
   * produce an error.</p>
   * 
   * <p>JSP Tag Lib Descriptor
   * <pre>
   * &lt;name>rowCount&lt;/name>
   * &lt;tagclass>org.apache.taglibs.jdbc.statement.RowCountTag&lt;/tagclass>
   * &lt;bodycontent>empty&lt;/bodycontent>
   * &lt;info>Prints out the number of rows retrieved from the database.
   * It can be used inside a ResultSet tag to provide a running
   * count of rows retreived, or after the ResultSet tag to
   * display the total number.  Using the tag before the ResultSet
   * or outside of a Statement or PreparedStatement will
   * produce an error.&lt;/info>
   * </pre>
   * 
   * @author Morgan Delagrange
   * @see ResultSet
   */
  public class RowCountTag extends TagSupport {
  
    /**
     * Get the enclosing StatementTag tag
     * 
     * @return StatementTag
     * @exception ClassNotFoundException
     */
    private StatementTag getStatementTag() throws ClassNotFoundException {
      StatementTag statementTag = 
      (StatementTag) findAncestorWithClass(this, 
Class.forName("org.apache.taglibs.jdbc.statement.StatementTag"));
      return statementTag;
    }
  
    public int doStartTag() throws JspTagException {
  
      try {
        StatementTag stmtTag = getStatementTag();
  
        int rowCount = stmtTag.getTotalRowCount();
  
        if (rowCount == -1) {
          throw new JspTagException("rowCount tag must be used inside or after a 
ResultSet tag.");
        }
  
        pageContext.getOut().write(String.valueOf(rowCount));
      } catch (ClassNotFoundException e) {
        throw new JspTagException(e.toString());
      } catch (IOException e) {
        throw new JspTagException(e.toString());
      }
  
      return SKIP_BODY;
    }
  
  }