Presto UDAF 输出数组(array)

最新推荐文章于 2023-02-02 09:31:14 发布

原创

最新推荐文章于 2023-02-02 09:31:14 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Presto #udaf #array

本文介绍了Presto中UDAF（用户定义聚合函数）处理数组(array)的方式，对比了版本0.189之前和之后的使用方法，虽然逻辑相同，但存在使用上的差异。

1.版本0.189之前的使用:

@OutputFunction("array(" + StandardTypes.BIGINT + ")")

    public static void output(SliceState state, BlockBuilder out) {

        // 获取状态

        Slice slice = state.getSlice();

        // 数据为空, 返回一个空数组

        if (null == slice) {

            BlockBuilder blockBuilder = BigintType.BIGINT.createBlockBuilder(new BlockBuilderStatus(), 0);

            out.writeObject(blockBuilder.build());

            out.closeEntry();

            return;

        }

        // 构造结果: [A:100, B:50, C:10, ......]

        BlockBuilder blockBuilder = BigintType.BIGINT.createBlockBuilder(new BlockBuilderStatus(), slice.length() / 4);

        for (int index = 0; index < slice.length(); index += 4) {

            BigintType.BIGINT.writeLong(blockBuilder, slice.getInt(index));

        }

        // 返回结果

        out.writeObject(blockBuilder.build());

        out.closeEntry();

    }