Application: gvSIG desktop

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

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.JDBCStoreProviderFactoryBase;

public class H2SpatialStoreProviderFactory extends JDBCStoreProviderFactoryBase {

  private static final String NAME = H2SpatialLibrary.NAME;

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

           Arrays.asList(new String[] {

             "COUNT",

             "SUM",

             "MIN",

             "MAX",

             "AVG"

           }

       )

    );

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

  public H2SpatialStoreProviderFactory() {

    super(

            NAME,

            "H2Spatial store"

    );

  }

  @Override

  public JDBCStoreProvider createProvider(

          DataParameters parameters,

          DataStoreProviderServices providerServices

  ) throws InitializeException {

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

    JDBCStoreProvider provider = helper.createProvider(

            (JDBCStoreParameters) parameters,

            providerServices

    );

    return provider;

  }

  @Override

  public JDBCStoreParameters createParameters() {

    JDBCStoreParameters params = new H2SpatialStoreParameters();

    return params;

  }

  @Override

  public int hasVectorialSupport() {

    return YES;

  }

  @Override

  public int hasRasterSupport() {

    return NO;

  }

  @Override

  public int hasTabularSupport() {

    return YES;

  }

  @Override

  public int hasSQLSupport() {

    return YES;

  }

  @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 allowEditableFeatureType() {

    return YES;

  }

}