/*
 * Copyright 2011 Revelytix, Inc.
 * 
 * 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.
 */
package sherpa.client;

import java.util.HashMap;
import java.util.Map;

import spark.api.Command;
import spark.api.Connection;
import spark.api.Result;
import spark.api.Solutions;
import spark.spi.BaseCommand;

public class SHPCommand extends BaseCommand implements Command {
  
  private QueryExecution query;
  private int batchSize;
  
  public SHPCommand(Connection connection, String command, QueryExecution query) {
    super(connection, command);
    this.query = query;
  }

  public void setBatchSize(int batchSize) {
    this.batchSize = batchSize;
  }
  
  @Override
  public void cancel() {
    this.query.cancel();
  }

  @Override
  public Result execute() {    
    // TODO Auto-generated method stub
    return null;
  }

  @Override
  public boolean executeAsk() {
    // TODO Auto-generated method stub
    return false;
  }

  @Override
  public Solutions executeQuery() {
    Map<String,String> params = new HashMap<String,String>();
    Map<String,String> props = new HashMap<String,String>();
    props.put(QueryExecution.BATCH_SIZE, "" + batchSize);
    props.put(QueryExecution.TIMEOUT, "" + getTimeout());
    query.query(getCommand(), params, props);
    return new SHPSolutions(this, query);
  }

}

@@ -917,19 +917,23 @@
 private 
-i
+I
 nt
+eger
  batchSi
@@ -930,24 +930,31 @@
 er batchSize
+ = null
 ;%0A  %0A  publi
@@ -1155,16 +1155,32 @@
 hSize = 
+Integer.valueOf(
 batchSiz
@@ -1180,16 +1180,17 @@
 atchSize
+)
 ;%0A  %7D%0A  
@@ -1638,24 +1638,47 @@
 ring%3E();%0A   
+ if (batchSize != null)
  props.put(Q
@@ -1702,21 +1702,16 @@
 CH_SIZE,
- %22%22 +
  batchSi
@@ -1708,24 +1708,35 @@
 E, batchSize
+.toString()
 );%0A    props

Use default batch size of 1000 if none explicitly set in Sherpa client