Flink 自定义GroupConcat函数

函数实现逻辑

import org.apache.flink.table.api.DataTypes;
import org.apache.flink.table.api.dataview.ListView;
import org.apache.flink.table.runtime.functions.aggregate.BuiltInAggregateFunction;
import org.apache.flink.table.types.DataType;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;

public class GroupConcatFullUDF extends BuiltInAggregateFunction<String, GroupConcatFullUDF.GroupConcatAccumulator> {
   
   

    private static final String DEFAULT_SEPARATOR = ",";

    @Override
    public DataType getOutputDataType() {
   
   
        return DataTypes.STRING();
    }

    @Override
    public List<DataType> getArgumentDataTypes() {
   
   
        List<DataType> types = new ArrayList<>(3);
        types.add(DataTypes.STRING());
        types.add(DataTypes.STRING());
        types.add(DataTypes.BOOLEAN());
        return types;
    }

    @Override
    public DataType getAccumulatorDataType() {
   
   
        return DataTypes.STRUCTURED(
                GroupConcatUDF.GroupConcatAccumulator.class,
                DataTypes.FIELD(
                        "list",
                        ListView.newListViewDataType(
                                DataTypes.STRING())),
                DataTypes.FIELD(
                        "retractList",
                        ListView.newListViewDataType(
                                DataTypes.STRING())));
    }

    public void resetAccumulator(GroupConcatUDF.GroupConcatAccumulator acc) {
   
   
        acc.list.clear();
        acc.retractList.clear();
    }

    @Override
    public GroupConcatAccumulator createAccumulator() {
   
   
        
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值