package com.gemstone.gemfire.internal.cache;

import java.util.Map;

import com.gemstone.gemfire.cache.AttributesFactory;
import com.gemstone.gemfire.cache.DataPolicy;
import com.gemstone.gemfire.cache.client.ClientRegionShortcut;
import com.gemstone.gemfire.cache.client.Pool;
import com.gemstone.gemfire.cache.client.PoolFactory;
import com.gemstone.gemfire.cache.query.QueryService;
import com.gemstone.gemfire.distributed.DistributedSystem;

/**
 * Created by ashahid on 3/30/17.
 */
public class GemFireSparkConnectorCacheImpl extends GemFireCacheImpl {


  private Map<String, String> gfeGridMappings = null;

  public GemFireSparkConnectorCacheImpl(PoolFactory pf, Map<String, String> gfeGridMappings,
      DistributedSystem system, CacheConfig cacheConfig) {
    super(false, pf, system, cacheConfig);
    this.gfeGridMappings = gfeGridMappings;
  }

  public static GemFireCacheImpl create(PoolFactory pf, Map<String, String> gfeGridMappings,
      DistributedSystem system, CacheConfig cacheConfig) {

    return new GemFireSparkConnectorCacheImpl(pf, gfeGridMappings, system, cacheConfig).init();
  }


  @Override
  protected GemFireCacheImpl init() {
    PoolFactory temp = this.clientpf;
    super.init();
    //check if default pool has already been created
    boolean defaultPoolIdentified = this.getDefaultPool() != null;
    //re assign the clientPf which would otherwise be nullified by the super call
    this.clientpf = temp;

    if (temp != null && !defaultPoolIdentified) {
      this.determineDefaultPool();
      defaultPoolIdentified = true;
    } else if (temp != null && defaultPoolIdentified ) {
      throw new IllegalStateException("Default grid could not spawn default pool");
    }
    AttributesFactory af = new AttributesFactory();
    af.setDataPolicy(DataPolicy.EMPTY);
    UserSpecifiedRegionAttributes ra = (UserSpecifiedRegionAttributes)af.create();
    ra.requiresPoolName = !defaultPoolIdentified;
    this.setRegionAttributes(ClientRegionShortcut.PROXY.toString(), ra);
    this.clientpf = null;
    return this;
  }

  @Override
  protected void checkValidityForPool() {
  }

  public QueryService getRemoteGemFireQueryService() {
    Pool p = getDefaultPool();
    if (p == null) {
      throw new IllegalStateException("Client cache does not have a default pool. " +
          "Use getQueryService(String poolName) instead.");
    } else {
      return p.getQueryService();
    }
  }

}

package com.gemstone.gemfire.internal.cache;

import java.util.Map;

import com.gemstone.gemfire.cache.AttributesFactory;
import com.gemstone.gemfire.cache.DataPolicy;
import com.gemstone.gemfire.cache.client.ClientRegionShortcut;
import com.gemstone.gemfire.cache.client.Pool;
import com.gemstone.gemfire.cache.client.PoolFactory;
import com.gemstone.gemfire.cache.query.QueryService;
import com.gemstone.gemfire.distributed.DistributedSystem;

/**
 * Created by ashahid on 3/30/17.
 */
public class GemFireSparkConnectorCacheImpl extends GemFireCacheImpl {


  private Map<String, String> gfeGridMappings = null;

  public GemFireSparkConnectorCacheImpl(PoolFactory pf, Map<String, String> gfeGridMappings,
      DistributedSystem system, CacheConfig cacheConfig) {
    super(false, pf, system, cacheConfig);
    this.gfeGridMappings = gfeGridMappings;
  }

  public static GemFireCacheImpl create(PoolFactory pf, Map<String, String> gfeGridMappings,
      DistributedSystem system, CacheConfig cacheConfig) {

    return new GemFireSparkConnectorCacheImpl(pf, gfeGridMappings, system, cacheConfig).init();
  }

  @Override
  public boolean hasPool() {
    return true;
  }

  @Override
  protected GemFireCacheImpl init() {
    PoolFactory temp = this.clientpf;
    super.init();
    //check if default pool has already been created
    boolean defaultPoolIdentified = this.getDefaultPool() != null;
    //re assign the clientPf which would otherwise be nullified by the super call
    this.clientpf = temp;

    if (temp != null && !defaultPoolIdentified) {
      this.determineDefaultPool();
      defaultPoolIdentified = true;
    } else if (temp != null && defaultPoolIdentified ) {
      throw new IllegalStateException("Default grid could not spawn default pool");
    }
    AttributesFactory af = new AttributesFactory();
    af.setDataPolicy(DataPolicy.EMPTY);
    UserSpecifiedRegionAttributes ra = (UserSpecifiedRegionAttributes)af.create();
    ra.requiresPoolName = !defaultPoolIdentified;
    this.setRegionAttributes(ClientRegionShortcut.PROXY.toString(), ra);
    this.clientpf = null;
    return this;
  }

  @Override
  protected void checkValidityForPool() {
  }

  public QueryService getRemoteGemFireQueryService() {
    Pool p = getDefaultPool();
    if (p == null) {
      throw new IllegalStateException("Client cache does not have a default pool. " +
          "Use getQueryService(String poolName) instead.");
    } else {
      return p.getQueryService();
    }
  }

}

change to create ClientTypeRegistry for spark gemfire connector type cache (#235)