/**

  * Copyright (c) 2012-2015, jcabi.com

  * 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) Neither the name of the jcabi.com nor

  * the names of its contributors may be used to endorse or promote

  * products derived from this software without specific prior written

  * permission.

  *

  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

  * "AS IS" AND ANY EXPRESS 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 COPYRIGHT HOLDER OR 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.

  */

 package com.jcabi.jdbc;

 import com.jcabi.aspects.Immutable;

 import com.jcabi.aspects.Loggable;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.sql.Statement;

 import java.util.Collection;

 import java.util.Date;

 import java.util.LinkedList;

 import lombok.EqualsAndHashCode;

 import lombok.ToString;

 /**

  * Outcome that returns first column.

  *

  * <p>Use it when you need the first column:

  *

  * <pre> Collection&lgt;Long&gt; salaries = new JdbcSession(source)

  *   .sql("SELECT salary FROM user")

  *   .select(new ColumnOutcome&lt;Long&gt;(Long.class));</pre>

  *

  * <p>Supported types are: {@link String}, {@link Long}, {@link Boolean},

  * {@link Byte}, {@link Date}, and {@link Utc}.

  *

  * @author Yegor Bugayenko (yegor@teamed.io)

  * @version $Id: 0d63c01ecb4e7b988bcdb2a4f452462d4b6e37f0 $

  * @since 0.13

  * @param <T> Type of items

  */

 @Immutable

 @ToString

 @EqualsAndHashCode(of = "type")

 public final class ColumnOutcome<T> implements Outcome<Collection<T>> {

     /**

      * The type name.

      */

     private final transient String type;

     /**

      * Public ctor.

      * @param tpe The type to convert to

      */

     public ColumnOutcome(final Class<T> tpe) {

         //@checkstyle BooleanExpressionComplexity (3 lines)

         if (tpe.equals(String.class) || tpe.equals(Long.class)

             || tpe.equals(Boolean.class) || tpe.equals(Byte.class)

             || tpe.equals(Date.class) || tpe.equals(Utc.class)

             || byte[].class.equals(tpe)

             ) {

             this.type = tpe.getName();

         } else {

             throw new IllegalArgumentException(

                 String.format("type %s is not supported", tpe.getName())

             );

         }

     }

     @Override

     @Loggable(Loggable.DEBUG)

     public Collection<T> handle(final ResultSet rset, final Statement stmt)

         throws SQLException {

         final Collection<T> result = new LinkedList<T>();

         while (rset.next()) {

             result.add(this.fetch(rset));

         }

         return result;

     }

     /**

      * Fetch the value from result set.

      * @param rset Result set

      * @return The result

      * @throws SQLException If some error inside

      */

     @SuppressWarnings("unchecked")

     private T fetch(final ResultSet rset) throws SQLException {

         final Object result;

         Class<T> tpe;

         try {

             tpe = (Class<T>) Class.forName(this.type);

             if (tpe.equals(String.class)) {

                 result = rset.getString(1);

             } else if (tpe.equals(Long.class)) {

                 result = rset.getLong(1);

             } else if (tpe.equals(Boolean.class)) {

                 result = rset.getBoolean(1);

             } else if (tpe.equals(Byte.class)) {

                 result = rset.getByte(1);

             } else if (tpe.equals(Date.class)) {

                 result = rset.getDate(1);

             } else if (tpe.equals(Utc.class)) {

                 result = new Utc(Utc.getTimestamp(rset, 1));

             } else if (byte[].class.equals(tpe)) {

                 result = rset.getBytes(1);

             } else {

                 throw new IllegalStateException(

                     String.format("type %s is not allowed", tpe.getName())

                 );

             }

         } catch (final ClassNotFoundException ex) {

             throw new IllegalArgumentException(

                 String.format("Unknown type: %s", this.type), ex

             );

         }

         return tpe.cast(result);

     }

 }