Application: gvSIG desktop » gvSIG data provider for PostgreSQL

package org.gvsig.postgresql.dal;

import java.util.Arrays;

import java.util.Collections;

import java.util.HashSet;

import java.util.Set;

import java.util.function.Function;

import org.gvsig.fmap.dal.DataParameters;

import static org.gvsig.fmap.dal.DataStoreProviderFactory.NO;

import static org.gvsig.fmap.dal.DataStoreProviderFactory.UNKNOWN;

import static org.gvsig.fmap.dal.DataStoreProviderFactory.YES;

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

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

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

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

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

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

import org.gvsig.fmap.dal.store.jdbc2.impl.JDBCStoreProviderFactory;

public class PostgreSQLStoreProviderFactory extends JDBCStoreProviderFactory {

    private static final Set<String> ALLOWED_FUNCTIONS = new HashSet<>(

            Arrays.asList(new String[]{

        "COUNT",

        "SUM",

        "MIN",

        "MAX"

    }

            )

    );

    private static final Set<String> DENIED_FUNCTIONS = Collections.EMPTY_SET;

    public PostgreSQLStoreProviderFactory() {

        super(

                PostgreSQLLibrary.NAME,

                "PostgreSQL store"

        );

    }

    @Override

    public JDBCStoreProvider createProvider(

            DataParameters parameters,

            DataStoreProviderServices providerServices

    ) throws InitializeException {

        JDBCHelper helper = new PostgreSQLHelper((JDBCConnectionParameters) parameters);

        JDBCStoreProvider provider = helper.createProvider(

                (JDBCStoreParameters) parameters,

                providerServices

        );

        return provider;

    }

    @Override

    public JDBCStoreParameters createParameters() {

        JDBCStoreParameters params = new PostgreSQLStoreParameters();

        return params;

    }

    @Override

    public Function<String, Integer> allowFunction() {

        return (Function<String, Integer>) (String name) -> {

            if (ALLOWED_FUNCTIONS.contains(name)) {

                return YES;

            }

            if (DENIED_FUNCTIONS.contains(name)) {

                return NO;

            }

            return UNKNOWN;

        };

    }

    @Override

    public int allowGroupBy() {

        return YES;

    }

    @Override

    public int hasVectorialSupport() {

        return YES;

    }

    @Override

    public int hasRasterSupport() {

        return NO;

    }

    @Override

    public int hasTabularSupport() {

        return YES;

    }

    @Override

    public int hasSQLSupport() {

        return YES;

    }

}