Application: gvSIG desktop

package org.gvsig.fmap.dal.store.jdbc2.spi.operations;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;

import org.gvsig.fmap.dal.store.jdbc2.JDBCUtils;

import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;

import static org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase.PROP_TABLE;

public class CountOperation extends AbstractConnectionOperation {

    private final TableReference table;

    private final String baseFilter;

    private final String filter;

    private final FeatureType featureType;

    public CountOperation(

            JDBCHelper helper

        ) {

        this(helper, null, null, null, null);

    }

    public CountOperation(

            JDBCHelper helper,

            FeatureType featureType,

            TableReference table,

            String baseFilter,

            String filter

        ) {

        super(helper);

        this.featureType = featureType;

        this.table = table;

        this.baseFilter = baseFilter;

        this.filter = filter;

    }

    @Override

    public final Object perform(Connection conn) throws DataException {

        return this.count(conn);

    }

    public String getSQL() {

        JDBCSQLBuilderBase sqlbuilder = this.createSQLBuilder();

        ExpressionBuilder expbuilder = sqlbuilder.expression();

        sqlbuilder.select().column().value(sqlbuilder.count().all());

        sqlbuilder.select().from().table()

                .database(this.table.getDatabase())

                .schema(this.table.getSchema())

                .name(this.table.getTable());

        sqlbuilder.select().from().subquery(this.table.getSubquery());

        if (!StringUtils.isEmpty(baseFilter)) {

            sqlbuilder.select().where().set( expbuilder.custom(baseFilter) );

        }

        if (!StringUtils.isEmpty(filter)) {

            // El and() hace un set() si no hay un filtro previo

            sqlbuilder.select().where().and(expbuilder.toValue(filter));

        }

        this.helper.replaceForeingValueFunction(sqlbuilder, featureType);

        sqlbuilder.select().remove_all_columns();

        sqlbuilder.select().column().value(sqlbuilder.count().all());

        sqlbuilder.setProperties(

                ExpressionBuilder.Variable.class, 

                PROP_TABLE, table

        );

        String sql = sqlbuilder.select().toString();

        return sql;

    }

    public long count(Connection conn) throws DataException {

        String sql = this.getSQL();

        Statement st = null;

        ResultSet rs = null;

        try {

            st = conn.createStatement();

            rs = JDBCUtils.executeQuery(st, sql);

            if (!rs.next()) {

                return 0;

            }

            return rs.getLong(1);

        } catch (SQLException ex) {

            throw new JDBCSQLException(ex);

        } finally {

            JDBCUtils.closeQuietly(st);

            JDBCUtils.closeQuietly(rs);

        }

    }

}