Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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.flink.streaming.runtime.operators.windowing;
import org.apache.flink.annotation.VisibleForTesting;
import org.apache.flink.api.common.ExecutionConfig;
import org.apache.flink.api.common.functions.AggregateFunction;
import org.apache.flink.api.common.functions.Function;
import org.apache.flink.api.common.functions.ReduceFunction;
import org.apache.flink.api.common.functions.RichFunction;
import org.apache.flink.api.common.state.AggregatingStateDescriptor;
import org.apache.flink.api.common.state.AppendingState;
import org.apache.flink.api.common.state.ListStateDescriptor;
import org.apache.flink.api.common.state.ReducingStateDescriptor;
import org.apache.flink.api.common.state.StateDescriptor;
import org.apache.flink.api.common.typeinfo.TypeInformation;
import org.apache.flink.api.common.typeutils.TypeSerializer;
import org.apache.flink.api.java.functions.KeySelector;
import org.apache.flink.streaming.api.functions.windowing.AggregateApplyWindowFunction;
import org.apache.flink.streaming.api.functions.windowing.ProcessWindowFunction;
import org.apache.flink.streaming.api.functions.windowing.ReduceApplyProcessWindowFunction;
import org.apache.flink.streaming.api.functions.windowing.ReduceApplyWindowFunction;
import org.apache.flink.streaming.api.functions.windowing.WindowFunction;
import org.apache.flink.streaming.api.windowing.assigners.BaseAlignedWindowAssigner;
import org.apache.flink.streaming.api.windowing.assigners.MergingWindowAssigner;
import org.apache.flink.streaming.api.windowing.assigners.WindowAssigner;
import org.apache.flink.streaming.api.windowing.evictors.Evictor;
import org.apache.flink.streaming.api.windowing.time.Time;
import org.apache.flink.streaming.api.windowing.triggers.Trigger;
import org.apache.flink.streaming.api.windowing.windows.Window;
import org.apache.flink.streaming.runtime.operators.windowing.functions.InternalAggregateProcessWindowFunction;
import org.apache.flink.streaming.runtime.operators.windowing.functions.InternalIterableProcessWindowFunction;
import org.apache.flink.streaming.runtime.operators.windowing.functions.InternalIterableWindowFunction;
import org.apache.flink.streaming.runtime.operators.windowing.functions.InternalSingleValueProcessWindowFunction;
import org.apache.flink.streaming.runtime.operators.windowing.functions.InternalSingleValueWindowFunction;
import org.apache.flink.streaming.runtime.operators.windowing.functions.InternalWindowFunction;
import org.apache.flink.streaming.runtime.streamrecord.StreamElementSerializer;
import org.apache.flink.streaming.runtime.streamrecord.StreamRecord;
import org.apache.flink.util.OutputTag;
import org.apache.flink.util.Preconditions;
import javax.annotation.Nullable;
import java.lang.reflect.Type;
/**
* A builder for creating {@link WindowOperator WindowOperators}.
*
* @param The type of key returned by the {@code KeySelector}.
* @param The type of the incoming elements.
* @param The type of {@code Window} that the {@code WindowAssigner} assigns.
*/
public class WindowOperatorBuilder {
private static final String WINDOW_STATE_NAME = "window-contents";
private final ExecutionConfig config;
private final WindowAssigner windowAssigner;
private final TypeInformation inputType;
private final KeySelector keySelector;
private final TypeInformation keyType;
private Trigger trigger;
@Nullable private Evictor evictor;
private long allowedLateness = 0L;
@Nullable private OutputTag lateDataOutputTag;
public WindowOperatorBuilder(
WindowAssigner windowAssigner,
Trigger trigger,
ExecutionConfig config,
TypeInformation inputType,
KeySelector keySelector,
TypeInformation keyType) {
this.windowAssigner = windowAssigner;
this.config = config;
this.inputType = inputType;
this.keySelector = keySelector;
this.keyType = keyType;
this.trigger = trigger;
}
public void trigger(Trigger trigger) {
Preconditions.checkNotNull(trigger, "Window triggers cannot be null");
if (windowAssigner instanceof MergingWindowAssigner && !trigger.canMerge()) {
throw new UnsupportedOperationException(
"A merging window assigner cannot be used with a trigger that does not support merging.");
}
if (windowAssigner instanceof BaseAlignedWindowAssigner) {
throw new UnsupportedOperationException(
"Cannot use a "
+ windowAssigner.getClass().getSimpleName()
+ " with a custom trigger.");
}
this.trigger = trigger;
}
public void allowedLateness(Time lateness) {
Preconditions.checkNotNull(lateness, "Allowed lateness cannot be null");
final long millis = lateness.toMilliseconds();
Preconditions.checkArgument(millis >= 0, "The allowed lateness cannot be negative.");
this.allowedLateness = millis;
}
public void sideOutputLateData(OutputTag outputTag) {
Preconditions.checkNotNull(outputTag, "Side output tag must not be null.");
this.lateDataOutputTag = outputTag;
}
public void evictor(Evictor evictor) {
Preconditions.checkNotNull(evictor, "Evictor cannot be null");
if (windowAssigner instanceof BaseAlignedWindowAssigner) {
throw new UnsupportedOperationException(
"Cannot use a "
+ windowAssigner.getClass().getSimpleName()
+ " with an Evictor.");
}
this.evictor = evictor;
}
public WindowOperator reduce(
ReduceFunction reduceFunction, WindowFunction function) {
Preconditions.checkNotNull(reduceFunction, "ReduceFunction cannot be null");
Preconditions.checkNotNull(function, "WindowFunction cannot be null");
if (reduceFunction instanceof RichFunction) {
throw new UnsupportedOperationException(
"ReduceFunction of apply can not be a RichFunction.");
}
if (evictor != null) {
return buildEvictingWindowOperator(
new InternalIterableWindowFunction<>(
new ReduceApplyWindowFunction<>(reduceFunction, function)));
} else {
ReducingStateDescriptor stateDesc =
new ReducingStateDescriptor<>(
WINDOW_STATE_NAME, reduceFunction, inputType.createSerializer(config));
return buildWindowOperator(
stateDesc, new InternalSingleValueWindowFunction<>(function));
}
}
public WindowOperator reduce(
ReduceFunction reduceFunction, ProcessWindowFunction function) {
Preconditions.checkNotNull(reduceFunction, "ReduceFunction cannot be null");
Preconditions.checkNotNull(function, "ProcessWindowFunction cannot be null");
if (reduceFunction instanceof RichFunction) {
throw new UnsupportedOperationException(
"ReduceFunction of apply can not be a RichFunction.");
}
if (evictor != null) {
return buildEvictingWindowOperator(
new InternalIterableProcessWindowFunction<>(
new ReduceApplyProcessWindowFunction<>(reduceFunction, function)));
} else {
ReducingStateDescriptor stateDesc =
new ReducingStateDescriptor<>(
WINDOW_STATE_NAME, reduceFunction, inputType.createSerializer(config));
return buildWindowOperator(
stateDesc, new InternalSingleValueProcessWindowFunction<>(function));
}
}
public WindowOperator aggregate(
AggregateFunction aggregateFunction,
WindowFunction windowFunction,
TypeInformation accumulatorType) {
Preconditions.checkNotNull(aggregateFunction, "AggregateFunction cannot be null");
Preconditions.checkNotNull(windowFunction, "WindowFunction cannot be null");
if (aggregateFunction instanceof RichFunction) {
throw new UnsupportedOperationException(
"This aggregate function cannot be a RichFunction.");
}
if (evictor != null) {
return buildEvictingWindowOperator(
new InternalIterableWindowFunction<>(
new AggregateApplyWindowFunction<>(aggregateFunction, windowFunction)));
} else {
AggregatingStateDescriptor stateDesc =
new AggregatingStateDescriptor<>(
WINDOW_STATE_NAME,
aggregateFunction,
accumulatorType.createSerializer(config));
return buildWindowOperator(
stateDesc, new InternalSingleValueWindowFunction<>(windowFunction));
}
}
public WindowOperator aggregate(
AggregateFunction aggregateFunction,
ProcessWindowFunction windowFunction,
TypeInformation accumulatorType) {
Preconditions.checkNotNull(aggregateFunction, "AggregateFunction cannot be null");
Preconditions.checkNotNull(windowFunction, "ProcessWindowFunction cannot be null");
if (aggregateFunction instanceof RichFunction) {
throw new UnsupportedOperationException(
"This aggregate function cannot be a RichFunction.");
}
if (evictor != null) {
return buildEvictingWindowOperator(
new InternalAggregateProcessWindowFunction<>(
aggregateFunction, windowFunction));
} else {
AggregatingStateDescriptor stateDesc =
new AggregatingStateDescriptor<>(
WINDOW_STATE_NAME,
aggregateFunction,
accumulatorType.createSerializer(config));
return buildWindowOperator(
stateDesc, new InternalSingleValueProcessWindowFunction<>(windowFunction));
}
}
public WindowOperator apply(WindowFunction function) {
Preconditions.checkNotNull(function, "WindowFunction cannot be null");
return apply(new InternalIterableWindowFunction<>(function));
}
public WindowOperator process(ProcessWindowFunction function) {
Preconditions.checkNotNull(function, "ProcessWindowFunction cannot be null");
return apply(new InternalIterableProcessWindowFunction<>(function));
}
private WindowOperator apply(
InternalWindowFunction, R, K, W> function) {
if (evictor != null) {
return buildEvictingWindowOperator(function);
} else {
ListStateDescriptor stateDesc =
new ListStateDescriptor<>(
WINDOW_STATE_NAME, inputType.createSerializer(config));
return buildWindowOperator(stateDesc, function);
}
}
private WindowOperator buildWindowOperator(
StateDescriptor, ?> stateDesc,
InternalWindowFunction function) {
return new WindowOperator<>(
windowAssigner,
windowAssigner.getWindowSerializer(config),
keySelector,
keyType.createSerializer(config),
stateDesc,
function,
trigger,
allowedLateness,
lateDataOutputTag);
}
private WindowOperator, R, W> buildEvictingWindowOperator(
InternalWindowFunction, R, K, W> function) {
@SuppressWarnings({"unchecked", "rawtypes"})
TypeSerializer> streamRecordSerializer =
(TypeSerializer>)
new StreamElementSerializer(inputType.createSerializer(config));
ListStateDescriptor> stateDesc =
new ListStateDescriptor<>(WINDOW_STATE_NAME, streamRecordSerializer);
return new EvictingWindowOperator<>(
windowAssigner,
windowAssigner.getWindowSerializer(config),
keySelector,
keyType.createSerializer(config),
stateDesc,
function,
trigger,
evictor,
allowedLateness,
lateDataOutputTag);
}
private static String generateFunctionName(Function function) {
Class functionClass = function.getClass();
if (functionClass.isAnonymousClass()) {
// getSimpleName returns an empty String for anonymous classes
Type[] interfaces = functionClass.getInterfaces();
if (interfaces.length == 0) {
// extends an existing class (like RichMapFunction)
Class functionSuperClass = functionClass.getSuperclass();
return functionSuperClass.getSimpleName()
+ functionClass
.getName()
.substring(functionClass.getEnclosingClass().getName().length());
} else {
// implements a Function interface
Class functionInterface = functionClass.getInterfaces()[0];
return functionInterface.getSimpleName()
+ functionClass
.getName()
.substring(functionClass.getEnclosingClass().getName().length());
}
} else {
return functionClass.getSimpleName();
}
}
public String generateOperatorName(Function function1, @Nullable Function function2) {
return "Window("
+ windowAssigner
+ ", "
+ trigger.getClass().getSimpleName()
+ ", "
+ (evictor == null ? "" : (evictor.getClass().getSimpleName() + ", "))
+ generateFunctionName(function1)
+ (function2 == null ? "" : (", " + generateFunctionName(function2)))
+ ")";
}
@VisibleForTesting
public long getAllowedLateness() {
return allowedLateness;
}
}