com.hazelcast.jet.impl.pipeline.transform.MapStatefulTransform Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2021, Hazelcast, Inc. All Rights Reserved.
*
* Licensed 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 com.hazelcast.jet.impl.pipeline.transform;
import com.hazelcast.function.FunctionEx;
import com.hazelcast.function.ToLongFunctionEx;
import com.hazelcast.jet.function.TriFunction;
import com.hazelcast.jet.impl.pipeline.PipelineImpl.Context;
import com.hazelcast.jet.impl.pipeline.Planner;
import com.hazelcast.jet.impl.pipeline.Planner.PlannerVertex;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.function.Supplier;
import static com.hazelcast.jet.core.Vertex.LOCAL_PARALLELISM_USE_DEFAULT;
import static com.hazelcast.jet.core.processor.Processors.mapStatefulP;
public class MapStatefulTransform extends StatefulKeyedTransformBase {
private static final long serialVersionUID = 1L;
private final TriFunction statefulMapFn;
@Nullable private TriFunction onEvictFn;
public MapStatefulTransform(
@Nonnull Transform upstream,
long ttl,
@Nonnull FunctionEx keyFn,
@Nonnull ToLongFunctionEx timestampFn,
@Nonnull Supplier createFn,
@Nonnull TriFunction statefulMapFn,
@Nullable TriFunction onEvictFn
) {
super("map-stateful-keyed", upstream, ttl, keyFn, timestampFn, createFn);
this.statefulMapFn = statefulMapFn;
this.onEvictFn = onEvictFn;
}
@Override
public void addToDag(Planner p, Context context) {
determineLocalParallelism(LOCAL_PARALLELISM_USE_DEFAULT, context, false);
PlannerVertex pv = p.addVertex(this, name(), determinedLocalParallelism(),
mapStatefulP(ttl, keyFn, timestampFn, createFn, statefulMapFn, onEvictFn));
p.addEdges(this, pv.v, edge -> edge.partitioned(keyFn).distributed());
}
}