Search in sources :

Example 6 with StreamInfo

use of backtype.storm.generated.StreamInfo in project jstorm by alibaba.

the class LinearDRPCTopologyBuilder method createTopology.

private StormTopology createTopology(DRPCSpout spout) {
    final String SPOUT_ID = "spout";
    final String PREPARE_ID = "prepare-request";
    TopologyBuilder builder = new TopologyBuilder();
    builder.setSpout(SPOUT_ID, spout);
    builder.setBolt(PREPARE_ID, new PrepareRequest()).noneGrouping(SPOUT_ID);
    int i = 0;
    for (; i < _components.size(); i++) {
        Component component = _components.get(i);
        Map<String, SourceArgs> source = new HashMap<String, SourceArgs>();
        if (i == 1) {
            source.put(boltId(i - 1), SourceArgs.single());
        } else if (i >= 2) {
            source.put(boltId(i - 1), SourceArgs.all());
        }
        IdStreamSpec idSpec = null;
        if (i == _components.size() - 1 && component.bolt instanceof FinishedCallback) {
            idSpec = IdStreamSpec.makeDetectSpec(PREPARE_ID, PrepareRequest.ID_STREAM);
        }
        BoltDeclarer declarer = builder.setBolt(boltId(i), new CoordinatedBolt(component.bolt, source, idSpec), component.parallelism);
        for (Map conf : component.componentConfs) {
            declarer.addConfigurations(conf);
        }
        if (idSpec != null) {
            declarer.fieldsGrouping(idSpec.getGlobalStreamId().get_componentId(), PrepareRequest.ID_STREAM, new Fields("request"));
        }
        if (i == 0 && component.declarations.isEmpty()) {
            declarer.noneGrouping(PREPARE_ID, PrepareRequest.ARGS_STREAM);
        } else {
            String prevId;
            if (i == 0) {
                prevId = PREPARE_ID;
            } else {
                prevId = boltId(i - 1);
            }
            for (InputDeclaration declaration : component.declarations) {
                declaration.declare(prevId, declarer);
            }
        }
        if (i > 0) {
            declarer.directGrouping(boltId(i - 1), Constants.COORDINATED_STREAM_ID);
        }
    }
    IRichBolt lastBolt = _components.get(_components.size() - 1).bolt;
    OutputFieldsGetter getter = new OutputFieldsGetter();
    lastBolt.declareOutputFields(getter);
    Map<String, StreamInfo> streams = getter.getFieldsDeclaration();
    if (streams.size() != 1) {
        throw new RuntimeException("Must declare exactly one stream from last bolt in LinearDRPCTopology");
    }
    String outputStream = streams.keySet().iterator().next();
    List<String> fields = streams.get(outputStream).get_output_fields();
    if (fields.size() != 2) {
        throw new RuntimeException("Output stream of last component in LinearDRPCTopology must contain exactly two fields. The first should be the request id, and the second should be the result.");
    }
    builder.setBolt("JoinResult", new JoinResult(PREPARE_ID)).fieldsGrouping(boltId(i - 1), outputStream, new Fields(fields.get(0))).fieldsGrouping(PREPARE_ID, PrepareRequest.RETURN_STREAM, new Fields("request"));
    i++;
    builder.setBolt("ReturnResults", new ReturnResults()).noneGrouping("JoinResult");
    return builder.createTopology();
}
Also used : IRichBolt(backtype.storm.topology.IRichBolt) TopologyBuilder(backtype.storm.topology.TopologyBuilder) HashMap(java.util.HashMap) OutputFieldsGetter(backtype.storm.topology.OutputFieldsGetter) IdStreamSpec(backtype.storm.coordination.CoordinatedBolt.IdStreamSpec) SourceArgs(backtype.storm.coordination.CoordinatedBolt.SourceArgs) Fields(backtype.storm.tuple.Fields) BoltDeclarer(backtype.storm.topology.BoltDeclarer) StreamInfo(backtype.storm.generated.StreamInfo) FinishedCallback(backtype.storm.coordination.CoordinatedBolt.FinishedCallback) HashMap(java.util.HashMap) Map(java.util.Map) CoordinatedBolt(backtype.storm.coordination.CoordinatedBolt)

Example 7 with StreamInfo

use of backtype.storm.generated.StreamInfo in project jstorm by alibaba.

the class ClojureBolt method declareOutputFields.

@Override
public void declareOutputFields(OutputFieldsDeclarer declarer) {
    for (String stream : _fields.keySet()) {
        StreamInfo info = _fields.get(stream);
        declarer.declareStream(stream, info.is_direct(), new Fields(info.get_output_fields()));
    }
}
Also used : Fields(backtype.storm.tuple.Fields) StreamInfo(backtype.storm.generated.StreamInfo)

Example 8 with StreamInfo

use of backtype.storm.generated.StreamInfo in project jstorm by alibaba.

the class TransactionBolt method declareOutputFields.

@Override
public void declareOutputFields(OutputFieldsDeclarer declarer) {
    TransactionOutputFieldsDeclarer transactionDeclarer = new TransactionOutputFieldsDeclarer();
    boltExecutor.declareOutputFields(transactionDeclarer);
    Map<String, StreamInfo> streams = transactionDeclarer.getFieldsDeclaration();
    for (Entry<String, StreamInfo> entry : streams.entrySet()) {
        String streamName = entry.getKey();
        StreamInfo streamInfo = entry.getValue();
        declarer.declareStream(streamName, streamInfo.is_direct(), new Fields(streamInfo.get_output_fields()));
    }
    if (streams.size() > 0) {
        declarer.declareStream(TransactionCommon.BARRIER_STREAM_ID, new Fields(TransactionCommon.BATCH_GROUP_ID_FIELD, TransactionCommon.BARRIER_SNAPSHOT_FIELD));
    } else {
        isEndBolt = true;
    }
}
Also used : Fields(backtype.storm.tuple.Fields) TransactionOutputFieldsDeclarer(com.alibaba.jstorm.transactional.TransactionOutputFieldsDeclarer) StreamInfo(backtype.storm.generated.StreamInfo)

Example 9 with StreamInfo

use of backtype.storm.generated.StreamInfo in project jstorm by alibaba.

the class TridentUtils method getSingleOutputStreamFields.

public static Fields getSingleOutputStreamFields(IComponent component) {
    OutputFieldsGetter getter = new OutputFieldsGetter();
    component.declareOutputFields(getter);
    Map<String, StreamInfo> declaration = getter.getFieldsDeclaration();
    if (declaration.size() != 1) {
        throw new RuntimeException("Trident only supports components that emit a single stream");
    }
    StreamInfo si = declaration.values().iterator().next();
    if (si.is_direct()) {
        throw new RuntimeException("Trident does not support direct streams");
    }
    return new Fields(si.get_output_fields());
}
Also used : Fields(backtype.storm.tuple.Fields) OutputFieldsGetter(backtype.storm.topology.OutputFieldsGetter) StreamInfo(backtype.storm.generated.StreamInfo)

Example 10 with StreamInfo

use of backtype.storm.generated.StreamInfo in project storm by nathanmarz.

the class TridentUtils method getSingleOutputStreamFields.

public static Fields getSingleOutputStreamFields(IComponent component) {
    OutputFieldsGetter getter = new OutputFieldsGetter();
    component.declareOutputFields(getter);
    Map<String, StreamInfo> declaration = getter.getFieldsDeclaration();
    if (declaration.size() != 1) {
        throw new RuntimeException("Trident only supports components that emit a single stream");
    }
    StreamInfo si = declaration.values().iterator().next();
    if (si.is_direct()) {
        throw new RuntimeException("Trident does not support direct streams");
    }
    return new Fields(si.get_output_fields());
}
Also used : Fields(backtype.storm.tuple.Fields) OutputFieldsGetter(backtype.storm.topology.OutputFieldsGetter) StreamInfo(backtype.storm.generated.StreamInfo)

Aggregations

StreamInfo (backtype.storm.generated.StreamInfo)12 Fields (backtype.storm.tuple.Fields)11 OutputFieldsGetter (backtype.storm.topology.OutputFieldsGetter)4 CoordinatedBolt (backtype.storm.coordination.CoordinatedBolt)2 FinishedCallback (backtype.storm.coordination.CoordinatedBolt.FinishedCallback)2 IdStreamSpec (backtype.storm.coordination.CoordinatedBolt.IdStreamSpec)2 SourceArgs (backtype.storm.coordination.CoordinatedBolt.SourceArgs)2 BoltDeclarer (backtype.storm.topology.BoltDeclarer)2 IRichBolt (backtype.storm.topology.IRichBolt)2 TopologyBuilder (backtype.storm.topology.TopologyBuilder)2 TransactionOutputFieldsDeclarer (com.alibaba.jstorm.transactional.TransactionOutputFieldsDeclarer)2 HashMap (java.util.HashMap)2 Map (java.util.Map)2 ArrayList (java.util.ArrayList)1