Presto UDAF 输出数组(array)

最新推荐文章于 2024-01-20 10:48:10 发布

赞神天降

最新推荐文章于 2024-01-20 10:48:10 发布

阅读量1.5k

点赞数 1

分类专栏： presto 文章标签： Presto udaf array

本文链接：https://blog.csdn.net/LG_49/article/details/86243412

版权

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

摘要由CSDN通过智能技术生成

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();

    }
<