io.ray.streaming.operator.impl.FlatMapOperator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of streaming-api Show documentation

ray streaming api

There is a newer version: 1.10.0

package io.ray.streaming.operator.impl;

import io.ray.streaming.api.collector.CollectionCollector;
import io.ray.streaming.api.collector.Collector;
import io.ray.streaming.api.context.RuntimeContext;
import io.ray.streaming.api.function.impl.FlatMapFunction;
import io.ray.streaming.message.Record;
import io.ray.streaming.operator.OneInputOperator;
import io.ray.streaming.operator.StreamOperator;
import java.util.List;

public class FlatMapOperator extends StreamOperator> implements
    OneInputOperator {

  private CollectionCollector collectionCollector;

  public FlatMapOperator(FlatMapFunction flatMapFunction) {
    super(flatMapFunction);
  }

  @Override
  public void open(List collectorList, RuntimeContext runtimeContext) {
    super.open(collectorList, runtimeContext);
    this.collectionCollector = new CollectionCollector(collectorList);
  }

  @Override
  public void processElement(Record record) throws Exception {
    this.function.flatMap(record.getValue(), (Collector) collectionCollector);
  }
}