Flink:有没有其他方法可以代替RichAggregateFunction来计算平均值和状态变量？

import org.apache.flink.api.common.state.ValueState;
import org.apache.flink.api.common.state.ValueStateDescriptor;
import org.apache.flink.streaming.api.functions.ProcessFunction;
import import org.apache.flink.util.Collector;

public class AverageProcessFunction extends ProcessFunction<Integer, Double> {
    private ValueState<Integer> sumState;
    private ValueState<Integer> countState;

    @Override
    public void open(Configuration parameters) throws Exception {
        ValueStateDescriptor<Integer> sumDescriptor = new ValueStateDescriptor<>("sum", Integer.class);
        sumState = getRuntimeContext().getState(sumDescriptor);

        ValueStateDescriptor<Integer> countDescriptor = new ValueStateDescriptor<>("count", Integer.class);
        countState = getRuntimeContext().getState(countDescriptor);
    }

    @Override
    public void processElement(Integer value, Context ctx, Collector<Double> out) throws Exception {
        Integer sum = sumState.value();
        Integer count = countState.value();

        if (sum == null) {
            sum = 0;
        }
        if (count == null) {
            count = 0;
        }

        sum += value;
        count++;

        sumState.update(sum);
        countState.update(count);

        double average = (double) sum / count;
        out.collect(average);
    }
}