| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 org.apache.streampipes.processors.filters.jvm.processor.enrich; |
| |
| import org.apache.streampipes.commons.exceptions.SpRuntimeException; |
| import org.apache.streampipes.model.DataProcessorType; |
| import org.apache.streampipes.model.graph.DataProcessorDescription; |
| import org.apache.streampipes.model.runtime.Event; |
| import org.apache.streampipes.model.runtime.EventFactory; |
| import org.apache.streampipes.model.schema.EventSchema; |
| import org.apache.streampipes.sdk.builder.ProcessingElementBuilder; |
| import org.apache.streampipes.sdk.builder.StreamRequirementsBuilder; |
| import org.apache.streampipes.sdk.helpers.*; |
| import org.apache.streampipes.sdk.utils.Assets; |
| import org.apache.streampipes.wrapper.context.EventProcessorRuntimeContext; |
| import org.apache.streampipes.wrapper.routing.SpOutputCollector; |
| import org.apache.streampipes.wrapper.standalone.ProcessorParams; |
| import org.apache.streampipes.wrapper.standalone.StreamPipesDataProcessor; |
| |
| import java.util.List; |
| |
| public class MergeByEnrichProcessor extends StreamPipesDataProcessor { |
| |
| private static final String SELECT_STREAM = "select-stream"; |
| |
| private List<String> outputKeySelectors; |
| private String selectedStream; |
| private EventSchema outputSchema; |
| private Event eventBuffer; |
| |
| @Override |
| public DataProcessorDescription declareModel() { |
| return ProcessingElementBuilder.create("org.apache.streampipes.processors.filters.jvm.enrich") |
| .category(DataProcessorType.TRANSFORM) |
| .withAssets(Assets.DOCUMENTATION, Assets.ICON) |
| .withLocales(Locales.EN) |
| .requiredStream(StreamRequirementsBuilder |
| .create() |
| .requiredProperty(EpRequirements.anyProperty()) |
| .build()) |
| .requiredStream(StreamRequirementsBuilder |
| .create() |
| .requiredProperty(EpRequirements.anyProperty()) |
| .build()) |
| .requiredSingleValueSelection(Labels.withId(SELECT_STREAM), |
| Options.from("Stream 1", "Stream 2")) |
| .outputStrategy(OutputStrategies.custom(true)) |
| .build(); |
| } |
| |
| @Override |
| public void onInvocation(ProcessorParams processorParams, SpOutputCollector spOutputCollector, EventProcessorRuntimeContext eventProcessorRuntimeContext) throws SpRuntimeException { |
| this.outputKeySelectors = processorParams.extractor().outputKeySelectors(); |
| |
| this.selectedStream = processorParams.extractor().selectedSingleValue(SELECT_STREAM, String.class); |
| |
| this.outputSchema = processorParams.getGraph().getOutputStream().getEventSchema(); |
| |
| if (this.selectedStream.equals("Stream 1")) { |
| this.selectedStream = "s0"; |
| } else { |
| this.selectedStream = "s1"; |
| } |
| |
| this.eventBuffer = null; |
| } |
| |
| @Override |
| public void onEvent(Event event, SpOutputCollector spOutputCollector) throws SpRuntimeException { |
| String streamId = event.getSourceInfo().getSelectorPrefix(); |
| |
| // Enrich the selected stream and store last event of other stream |
| if (this.selectedStream.equals(streamId)) { |
| if (this.eventBuffer != null) { |
| Event result = mergeEvents(event, this.eventBuffer); |
| spOutputCollector.collect(result); |
| } |
| } else { |
| this.eventBuffer = event; |
| } |
| } |
| |
| @Override |
| public void onDetach() throws SpRuntimeException { |
| |
| } |
| |
| private Event mergeEvents(Event e1, Event e2) { |
| return EventFactory.fromEvents(e1, e2, outputSchema).getSubset(outputKeySelectors); |
| } |
| } |