use of org.apache.geode.cache.control.RebalanceResults in project geode by apache.
the class RebalanceFunction method execute.
@Override
public void execute(FunctionContext context) {
RebalanceOperation op = null;
String[] str = new String[0];
Cache cache = CacheFactory.getAnyInstance();
ResourceManager manager = cache.getResourceManager();
Object[] args = (Object[]) context.getArguments();
String simulate = ((String) args[0]);
Set<String> includeRegionNames = (Set<String>) args[1];
Set<String> excludeRegionNames = (Set<String>) args[2];
RebalanceFactory rbFactory = manager.createRebalanceFactory();
rbFactory.excludeRegions(excludeRegionNames);
rbFactory.includeRegions(includeRegionNames);
RebalanceResults results = null;
if (simulate.equals("true")) {
op = rbFactory.simulate();
} else {
op = rbFactory.start();
}
try {
results = op.getResults();
logger.info("Starting RebalanceFunction got results = {}", results);
StringBuilder str1 = new StringBuilder();
str1.append(results.getTotalBucketCreateBytes() + "," + results.getTotalBucketCreateTime() + "," + results.getTotalBucketCreatesCompleted() + "," + results.getTotalBucketTransferBytes() + "," + results.getTotalBucketTransferTime() + "," + results.getTotalBucketTransfersCompleted() + "," + results.getTotalPrimaryTransferTime() + "," + results.getTotalPrimaryTransfersCompleted() + "," + results.getTotalTime() + ",");
Set<PartitionRebalanceInfo> regns1 = results.getPartitionRebalanceDetails();
Iterator it = regns1.iterator();
while (it.hasNext()) {
PartitionRebalanceInfo rgn = (PartitionRebalanceInfo) it.next();
str1.append(rgn.getRegionPath() + ",");
}
logger.info("Starting RebalanceFunction str1={}", str1);
context.getResultSender().lastResult(str1.toString());
} catch (CancellationException e) {
logger.info("Starting RebalanceFunction CancellationException: ", e.getMessage(), e);
context.getResultSender().lastResult("CancellationException1 " + e.getMessage());
} catch (InterruptedException e) {
logger.info("Starting RebalanceFunction InterruptedException: {}", e.getMessage(), e);
context.getResultSender().lastResult("InterruptedException2 " + e.getMessage());
}
}
use of org.apache.geode.cache.control.RebalanceResults in project geode by apache.
the class TestFunction method executeWithLastResult.
private void executeWithLastResult(FunctionContext context) {
RegionFunctionContext rfContext = (RegionFunctionContext) context;
final PartitionedRegion pr = (PartitionedRegion) rfContext.getDataSet();
ResourceManager resMan = pr.getCache().getResourceManager();
RebalanceFactory factory = resMan.createRebalanceFactory();
RebalanceOperation rebalanceOp = factory.start();
try {
RebalanceResults rebalanceResults = rebalanceOp.getResults();
} catch (CancellationException e) {
e.printStackTrace();
} catch (InterruptedException e) {
e.printStackTrace();
}
context.getResultSender().lastResult((Serializable) context.getArguments());
}
use of org.apache.geode.cache.control.RebalanceResults in project geode by apache.
the class LuceneQueriesAccessorBase method rebalanceRegion.
protected void rebalanceRegion(VM vm) {
// Do a rebalance
vm.invoke(() -> {
RebalanceOperation op = getCache().getResourceManager().createRebalanceFactory().start();
RebalanceResults results = op.getResults();
});
}
Aggregations