com.gemstone.gemfire.internal.tools.gfsh.app.cache.index.task.QuerySizeTask 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.internal.tools.gfsh.app.cache.index.task;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.Iterator;
import java.util.List;

import com.gemstone.gemfire.DataSerializer;
import com.gemstone.gemfire.cache.Cache;
import com.gemstone.gemfire.cache.CacheFactory;
import com.gemstone.gemfire.cache.Region;
import com.gemstone.gemfire.cache.execute.FunctionContext;
import com.gemstone.gemfire.internal.cache.PartitionedRegion;
import com.gemstone.gemfire.internal.tools.gfsh.aggregator.AggregateFunction;
import com.gemstone.gemfire.internal.tools.gfsh.aggregator.AggregateResults;
import com.gemstone.gemfire.internal.tools.gfsh.app.aggregator.AggregatorException;
import com.gemstone.gemfire.internal.tools.gfsh.app.aggregator.AggregatorPeer;
import com.gemstone.gemfire.internal.tools.gfsh.app.cache.index.Indexer;
import com.gemstone.gemfire.internal.tools.gfsh.app.cache.index.IndexerManager;
import com.gemstone.gemfire.internal.tools.gfsh.app.misc.util.DataSerializerEx;
import com.gemstone.gemfire.internal.tools.gfsh.command.CommandResults;
import com.gemstone.gemfire.internal.tools.gfsh.command.CommandTask;

public class QuerySizeTask implements CommandTask, AggregateFunction
{
	private static final long serialVersionUID = 1L;
	
  private transient volatile boolean aggregationExecuted = false;
  private final Object aggregationExecutedLock = new Object();
	
	private String regionPath;
	private Object queryKey;
	
	public QuerySizeTask() {}

	public QuerySizeTask(String regionPath, Object queryKey)
	{
		this.regionPath = regionPath;
		this.queryKey = queryKey;
	}
	
	public CommandResults runTask(Object userData)
	{
		Cache cache = CacheFactory.getAnyInstance();
		cache.getLogger().fine("QuerySizeTask.runTask(): regionPath = " + regionPath + ", queryKey = " + queryKey);
		
		CommandResults results = new CommandResults();
		
		Region region = cache.getRegion(regionPath);
		if (region == null) {
			results.setCode(QueryTask.ERROR_REGION_UNDEFINED);
			results.setCodeMessage("The specified region " + regionPath + " is undefined.");
			return results;
		}
		
		if (region instanceof PartitionedRegion) {	

			// Partitioned Region
			AggregatorPeer aggregator = new AggregatorPeer((PartitionedRegion)region);
			try {
				Object obj = aggregator.aggregate(this);
				results.setCode(QueryTask.SUCCESS_PR);
				results.setDataObject(obj);
			} catch (AggregatorException ex) {
				results.setCode(QueryTask.ERROR_AGGREGATOR);
				results.setCodeMessage("Unabled to create aggregator: " + ex.getMessage());
				ex.printStackTrace();
			}
			
		} else {
			
			// Replicated Region
			results.setCode(QueryTask.SUCCESS_RR);
			results.setDataObject(size());
		}
		
		return results;
	}

	public AggregateResults run(FunctionContext context)
	{
		AggregateResults results = null;
		synchronized (aggregationExecutedLock) {
			if (aggregationExecuted == false) {
				results = new AggregateResults();
				results.setDataObject(size());
				aggregationExecuted = true;
			}
		}
		return results;
	}

	public Object aggregate(List list)
	{
		int totalSize = 0;
		Iterator iterator = list.iterator();
		while (iterator.hasNext()) {
			AggregateResults results = iterator.next();
			if (results != null) {
				Integer size = (Integer)results.getDataObject();
				if (size != null) {
					totalSize += size;
				}
			}
		}
		return totalSize;
	}

	public Object aggregateDistributedSystems(Object[] results)
	{
		// Not supported
		return null;
	}
	
	private int size()
	{
		Indexer indexer = IndexerManager.getIndexerManager().getIndexer(regionPath);
		return indexer.size(queryKey);
	}

	public void fromData(DataInput in) throws IOException,
			ClassNotFoundException
	{
		regionPath = DataSerializerEx.readUTF(in);
		queryKey = DataSerializer.readObject(in);
	}

	public void toData(DataOutput out) throws IOException
	{
		DataSerializerEx.writeUTF(regionPath, out);
		DataSerializer.writeObject(queryKey, out);
	}
}