Application: gvSIG desktop

package org.gvsig.fmap.dal.store.h2.operations;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorerParameters;

import org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters;

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

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

import static org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer.SHOW_TABLES;

import static org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer.SHOW_TABLES_AND_VIEWS;

import static org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer.SHOW_VIEWS;

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

import org.gvsig.fmap.dal.store.jdbc2.spi.operations.ListTablesOperation;

public class H2SpatialListTablesOperation extends ListTablesOperation {

    public H2SpatialListTablesOperation(JDBCHelper helper, int mode, JDBCServerExplorerParameters serverParameters, boolean informationTables, int tablesOrViews) {

        super(helper, mode, serverParameters, informationTables, tablesOrViews);

    }

    @Override

    public List<JDBCStoreParameters> listTables(

            JDBCConnection conn,

            int mode,

            JDBCServerExplorerParameters serverParameters,

            boolean informationTables,

            int tablesOrViews

    ) {

        ResultSet rs = null;

        List<JDBCStoreParameters> tables = new ArrayList<>();

        try {

            StringBuilder builder = new StringBuilder();

            builder.append("SELECT TABLE_CATALOG, TABLE_SCHEMA, TABLE_NAME FROM INFORMATION_SCHEMA.TABLES where table_type in ");

            switch(tablesOrViews){

                default:

                case SHOW_TABLES_AND_VIEWS:

                    builder.append("('TABLE','VIEW','EXTERNAL')");

                    break;

                case SHOW_TABLES:

                    builder.append("('TABLE')");

                    break;

                case SHOW_VIEWS:

                    builder.append("('VIEW','EXTERNAL')");

                    break;

            }

            if( !informationTables ) {

                builder.append(" AND TABLE_SCHEMA<>'INFORMATION_SCHEMA'");

            }

            String sql = builder.toString();

            rs = conn.createStatement().executeQuery(sql);

            while (rs.next()) {

                JDBCStoreParameters params = this.helper.createOpenStoreParameters(serverParameters);

                params.setCatalog(rs.getString("TABLE_CATALOG"));

                params.setSchema(rs.getString("TABLE_SCHEMA"));

                params.setTable(rs.getString("TABLE_NAME"));

                tables.add(params);

            }

            return tables;

        } catch (SQLException ex) {

            throw new RuntimeException("Can't fetch tables information",ex);

        } finally {

            JDBCUtils.closeQuietly(rs);

        }

    }

}