package backtype.storm.testing;
import backtype.storm.Config;
import backtype.storm.topology.OutputFieldsDeclarer;
import java.util.Map;
import backtype.storm.spout.SpoutOutputCollector;
import backtype.storm.task.TopologyContext;
import backtype.storm.topology.base.BaseRichSpout;
import backtype.storm.tuple.Fields;
import backtype.storm.utils.Utils;
import java.util.HashMap;
public class TestPlannerSpout extends BaseRichSpout {
boolean _isDistributed;
Fields _outFields;
public TestPlannerSpout(Fields outFields, boolean isDistributed) {
_isDistributed = isDistributed;
_outFields = outFields;
}
public TestPlannerSpout(boolean isDistributed) {
this(new Fields("field1", "field2"), isDistributed);
}
public TestPlannerSpout(Fields outFields) {
this(outFields, true);
}
public Fields getOutputFields() {
return _outFields;
}
public void open(Map conf, TopologyContext context, SpoutOutputCollector collector) {
}
public void close() {
}
public void nextTuple() {
Utils.sleep(100);
}
public void ack(Object msgId){
}
public void fail(Object msgId){
}
public void declareOutputFields(OutputFieldsDeclarer declarer) {
declarer.declare(getOutputFields());
}
@Override
public Map<String, Object> getComponentConfiguration() {
Map<String, Object> ret = new HashMap<String, Object>();
if(!_isDistributed) {
ret.put(Config.TOPOLOGY_MAX_TASK_PARALLELISM, 1);
}
return ret;
}
}