将用SCALA编写的Spark UDF转换为JAVA

Question

下面是我的Spark UD，有人可以帮我将其转换为Java吗？

val customUDF = udf((array: Seq[String]) => {
    val newts = array.filter(_.nonEmpty)
    if  (newts.size == 0) null
    else newts.head
})

Answer 1

类似：

UDF2 my_udf = new UDF2<WrappedArray<String>, String>() {
  public String call(WrappedArray<String> arr) throws Exception {
    String[] newts = arr.filter(_.nonEmpty)
    if (newts.length == 0) {
      return null
    } else { newts[0] }
  }
};

spark.udf().register("my_udf", my_udf, DataTypes.StringType);