UDT示例

UDT运行时自带一个JSON的依赖（2.2.4），可以直接使用JSON。

set odps.sql.type.system.odps2=true;
set odps.sql.session.java.imports=java.util.*,java,com.google.gson.*; --同时导入多个Package时用逗号隔开。
@a := select new Gson() gson;   --构建Gson对象。
select 
gson.toJson(new ArrayList<Integer>(Arrays.asList(1, 2, 3))), --将任意对象转成JSON字符串。
cast(gson.fromJson('["a","b","c"]', List.class) as List<String>) --反序列化JSON字符串，注意Gson的接口，直接反序列化后是List<Object>类型，这里强制转换成List<String>，方便后续使用。
from @a;

内置类型ARRAY与java.util.List、MAP和java.util.Map存在映射关系。

• Java中实现java.util.List或java.util.Map接口类的对象，都可参与MaxCompute SQL的复杂类型操作。
• MaxCompute中ARRAY或MAP的数据，能够直接调用List或者MAP的接口。

set odps.sql.type.system.odps2=true;
set odps.sql.session.java.imports=java.util.*;
select
    size(new ArrayList<Integer>()),        --对ArrayList数据调用内置函数Size。
    array(1,2,3).size(),                   --对内置类型ARRAY调用List的Size方法。
    sort_array(new ArrayList<Integer>()),  --对ArrayList的数据进行排序。
    al[1],                                 --虽然Java的List不支持下标操作，但ARRAY支持。
    Objects.toString(a),        --之前不支持将ARRAY类型Cast成STRING，现在有绕过方法了。
    array(1,2,3).subList(1, 2)             --计算subList。
from (select new ArrayList<Integer>(array(1,2,3)) as al, array(1,2,3) as a) t;

示例如下，计算BigInteger的中位数（由于数据是java.math.BigInteger类型的，所以不能直接用内建函数MEDIAN）。

set odps.sql.session.java.imports=java.math.*;
@test_data := select * from values (1),(2),(3),(5) as t(value);
@a := select collect_list(new BigInteger(value)) values from @test_data;  --先把数据聚合成List。
@b := select sort_array(values) as values, values.size() cnt from @a;  --计算中位数的逻辑，先将数据排序。
@c := select if(cnt % 2 == 1, new BigDecimal(values[cnt div 2]), new BigDecimal(values[cnt div 2 - 1].add(values[cnt div 2])).divide(new BigDecimal(2))) med from @b;
--最终结果。
select med.toString() from @c;

表值函数允许输入多行多列数据，输出多行多列数据。可以按照如下操作实现：

• 输入多行多列数据，详情请参见聚合操作。
• 输出多行多列数据，可以用UDT方法输出一个Collection类型的数据（List或者MAP），然后调用Explode函数，将Collections展开成多行。
• UDT可以包含多个数据域，通过调用不同的Getter方法获取各个域的内容即可展开成多列。

展开一个JSON字符串的内容，示例如下。

@a := select '[{"a":"1","b":"2"},{"a":"1","b":"2"}]' str; --示例数据。
@b := select new com.google.gson.Gson().fromJson(str, java.util.List.class) l from @a; --反序列化JSON。
@c := select cast(e as java.util.Map<Object,Object>) m from @b lateral view explode(l) t as e;  --用explode展开成多行。
@d := select m.get('a') as a, m.get('b') as b from @c; --展开成多列。
select a.toString() a, b.toString() b from @d; --最终结果输出（注意变量d的输出中a、b两列是Object类型）。

通过UDT实现函数重载，可以解决上述问题。UDT支持泛型、类继承和变长参数，为您提供灵活的函数定义方式，示例如下。

public class UDTClass {
    // 这个函数支持一个数值类型（可以是TINYINT、SMALLINT、INT、BIGINT、FLOAT、DOUBLE以及任何以Number为基类的UDT），返回DOUBLE。
    public static Double doubleValue(Number input) {
        return input.doubleValue();
    }
    // 这个方法支持一个数值类型参数和一个任意类型的参数，返回值类型与第二个参数的类型相同。
    public static <T extends Number, R> R nullOrValue(T a, R b) {
        return a.doubleValue() > 0 ? b : null;
    }
    // 这个方法支持一个任意元素类型的ARRAY或List，返回BIGINT。
    public static Long length(java.util.List<? extends Object> input) {
        return input.size();
    }
    // 注意在不做强制转换的情况下参数只支持UDT的java.util.Map<Object, Object>对象。如果需要传入任何MAP对象，例如map<bigint,bigint>可以考虑:
    // 1. 定义函数时使用java.util.Map<? extends Object, ? extends Object>。
    // 2. 调用时做强制转换，例如UDTClass.mapSize(cast(mapObj as java.util.Map<Object, Object>))。
    public static Long mapSize(java.util.Map<Object, Object> input) {
        return input.size();
    }
}