Search in sources :

Example 11 with ProjectionFactory

use of org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory in project storm by apache.

the class ProjectedProcessor method prepare.

@Override
public void prepare(Map conf, TopologyContext context, TridentContext tridentContext) {
    if (tridentContext.getParentTupleFactories().size() != 1) {
        throw new RuntimeException("Projection processor can only have one parent");
    }
    _context = tridentContext;
    _factory = new ProjectionFactory(tridentContext.getParentTupleFactories().get(0), _projectFields);
}
Also used : ProjectionFactory(org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory)

Example 12 with ProjectionFactory

use of org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory in project storm by apache.

the class StateQueryProcessor method prepare.

@Override
public void prepare(Map<String, Object> conf, TopologyContext context, TridentContext tridentContext) {
    List<Factory> parents = tridentContext.getParentTupleFactories();
    if (parents.size() != 1) {
        throw new RuntimeException("State query operation can only have one parent");
    }
    this.context = tridentContext;
    state = (State) context.getTaskData(stateId);
    projection = new ProjectionFactory(parents.get(0), inputFields);
    collector = new AppendCollector(tridentContext);
    function.prepare(conf, new TridentOperationContext(context, projection));
}
Also used : ProjectionFactory(org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory) Factory(org.apache.storm.trident.tuple.TridentTuple.Factory) ProjectionFactory(org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory) TridentOperationContext(org.apache.storm.trident.operation.TridentOperationContext)

Example 13 with ProjectionFactory

use of org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory in project storm by apache.

the class GroupedMultiReducerExecutor method execute.

@Override
public void execute(Map<TridentTuple, Object> state, int streamIndex, TridentTuple full, TridentCollector collector) {
    ProjectionFactory groupFactory = groupFactories.get(streamIndex);
    ProjectionFactory inputFactory = inputFactories.get(streamIndex);
    TridentTuple group = groupFactory.create(full);
    TridentTuple input = inputFactory.create(full);
    Object curr;
    if (!state.containsKey(group)) {
        curr = reducer.init(collector, group);
        state.put(group, curr);
    } else {
        curr = state.get(group);
    }
    reducer.execute(curr, streamIndex, group, input, collector);
}
Also used : ProjectionFactory(org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory) TridentTuple(org.apache.storm.trident.tuple.TridentTuple)

Aggregations

ProjectionFactory (org.apache.storm.trident.tuple.TridentTupleView.ProjectionFactory)13 Factory (org.apache.storm.trident.tuple.TridentTuple.Factory)10 TridentOperationContext (org.apache.storm.trident.operation.TridentOperationContext)8 List (java.util.List)2 TridentMultiReducerContext (org.apache.storm.trident.operation.TridentMultiReducerContext)2 TridentTuple (org.apache.storm.trident.tuple.TridentTuple)1