要编写通用的array_agg UDF，您可以使用Apache Drill的自定义函数功能。以下是一个示例代码，展示了如何编写一个名为array_agg的通用UDF，该函数将数组中的所有值聚合到一个数组中。

首先，创建一个Java类，命名为ArrayAggFunction。该类需要继承org.apache.drill.exec.expr.fn.AbstractAggregator类，并实现其抽象方法。

import org.apache.drill.common.exceptions.UserException;
import org.apache.drill.exec.expr.fn.AbstractAggregator;
import org.apache.drill.exec.expr.holders.ObjectHolder;

public class ArrayAggFunction extends AbstractAggregator {

  private Object[] resultArray;

  @Override
  public void setup() {
    // 初始化聚合结果数组
    resultArray = new Object[10];
  }

  @Override
  public void add(ObjectHolder holder) {
    // 判断是否需要扩展数组大小
    if (resultArray.length <= count) {
      Object[] newArray = new Object[resultArray.length * 2];
      System.arraycopy(resultArray, 0, newArray, 0, resultArray.length);
      resultArray = newArray;
    }
    
    // 将值添加到数组中
    resultArray[count++] = holder.obj;
  }

  @Override
  public void output(ObjectHolder out) {
    // 将结果数组赋值给输出对象
    out.obj = resultArray;
    out.isSet = true;
  }

  @Override
  public void reset() {
    // 重置计数和结果数组
    count = 0;
    resultArray = new Object[10];
  }

  @Override
  public void cleanup() {
    // 清理资源
  }
}

接下来，您需要在Drill中注册该函数。创建一个名为drill-udf-example.conf的配置文件，并添加以下内容：

drill.exec.functions.udfs: {
  "array_agg": {
    "function-class": "com.example.ArrayAggFunction"
  }
}

将该配置文件放置在Drill的conf目录下。

最后，使用以下SQL语句测试array_agg UDF：

SELECT array_agg(col) AS agg_array
FROM table_name

此查询将返回一个包含表中所有值的数组。

请注意，以上示例代码仅用于说明目的。实际情况中，您可能需要根据自己的需求进行修改和调整。另外，还可以在UDF中添加其他逻辑来处理不同的数据类型和聚合需求。