com.gemstone.gemfire.cache.query.internal.CompiledRegion Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of gemfire-core Show documentation
SnappyData store based off Pivotal GemFireXD
The newest version!
/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You
 * may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License. See accompanying
 * LICENSE file.
 */
package com.gemstone.gemfire.cache.query.internal;

import java.util.*;
import com.gemstone.gemfire.cache.*;
import com.gemstone.gemfire.cache.query.*;
import com.gemstone.gemfire.cache.Region;
import com.gemstone.gemfire.internal.cache.BucketRegion;
import com.gemstone.gemfire.internal.cache.LocalDataSet;
import com.gemstone.gemfire.internal.cache.PartitionedRegion;
import com.gemstone.gemfire.internal.i18n.LocalizedStrings;


/**
 * Class Description
 *
 * @author      ericz
 */



public class CompiledRegion extends AbstractCompiledValue {
  private String regionPath;
  
  
  public CompiledRegion(String regionPath) {
    this.regionPath = regionPath;
  }
  
  public int getType() {
    return RegionPath;
  }
  
  public String getRegionPath() {
    return this.regionPath;
  }
  
  public Object evaluate(ExecutionContext context)
  throws RegionNotFoundException {
    Region rgn;
    Cache cache = context.getCache();
    // do PR bucketRegion substitution here for expressions that evaluate to a Region.
    PartitionedRegion pr = context.getPartitionedRegion();

    
    if (pr != null && pr.getFullPath().equals(this.regionPath)) {
      rgn = context.getBucketRegion();
    }else if(pr != null) {
    //Asif : This is a   very crude hack to allow equijoin queries on PartitionedRegion locally 
      //Done for Medco. This should be handled properly ASAP.      
      //We have possibly got a situation of equijoin. it may be across PRs. so use the context's bucket region
      // to get ID and then retrieve the this region's bucket region
      BucketRegion br = context.getBucketRegion();
      int bucketID = br.getId();
      //Is current region a partitioned region
      rgn = cache.getRegion(this.regionPath);
      if(rgn.getAttributes().getDataPolicy().withPartitioning()) {
        // convert it into bucket region.
        PartitionedRegion prLocal = (PartitionedRegion)rgn;
        rgn = prLocal.getDataStore().getLocalBucketById(bucketID);
      }
      
    }
    else {
      rgn = cache.getRegion(this.regionPath);
    }
    
    if (rgn == null) {
      // if we couldn't find the region because the cache is closed, throw
      // a CacheClosedException
      if (cache.isClosed()) {
        throw new CacheClosedException();
      }
      throw new RegionNotFoundException(LocalizedStrings.CompiledRegion_REGION_NOT_FOUND_0.toLocalizedString(this.regionPath));
    }
    
    if (context.isCqQueryContext()) {
      return new QRegion(rgn, true, context);
    } else {
      return new QRegion(rgn, false, context);  
    }
  }
  
  @Override
  public void generateCanonicalizedExpression(StringBuffer clauseBuffer,ExecutionContext context ) throws NameResolutionException{
    clauseBuffer.insert(0, regionPath);
    //rahul : changed for running queries on partitioned region.
    //clauseBuffer.insert(0, ((QRegion)this.evaluate(context)).getFullPath());
  }
  
  @Override
  public void getRegionsInQuery(Set regionsInQuery, Object[] parameters) {
    regionsInQuery.add(this.regionPath);
  }
  
  
}