MaxCompute：UDF樣本：複雜資料類型

Java UDF 程式碼範例

package com.aliyun; // package名稱，可以根據您的情況定義

import com.aliyun.odps.data.Struct;
import com.aliyun.odps.udf.UDF;
import com.aliyun.odps.udf.annotation.Resolve;

import java.text.SimpleDateFormat;
import java.util.*;

@Resolve("struct<input_name:string, input_timestamp:bigint>->map<string,string>")
public class ComplexDataTypeExample extends UDF{
    private static final String PATTERN = "yyyy-MM-dd HH:mm:ss";

    /**
     * 將時間戳記列錶轉換為時間字串列表
     * @param timestamps 時間戳記列表
     * @return 時間字串列表
     */
    public List<String> evaluate(List<Long> timestamps) {
        if (timestamps == null) {
            return null;
        }
        List<String> result = new ArrayList<>();
        SimpleDateFormat formatter = new SimpleDateFormat(PATTERN);
        for (Long timestamp : timestamps) {
            Date date = new Date(timestamp < 9999999999L ? timestamp * 1000 : timestamp);
            String dateString = formatter.format(date);
            result.add(dateString);
        }
        return result;
    }

    /**
     * 將時間戳記map轉換為時間字串map
     * @param timestamps 時間戳記map，其中value為時間戳記
     * @return 時間字串map
     */
    public Map<String, String> evaluate(Map<String, Long> timestamps) {
        if (timestamps == null) {
            return null;
        }
        Map<String, String> result = new HashMap<>(timestamps.size());
        SimpleDateFormat formatter = new SimpleDateFormat(PATTERN);
        for (String key : timestamps.keySet()) {
            Long timestamp = timestamps.get(key);
            Date date = new Date(timestamp < 9999999999L ? timestamp * 1000 : timestamp);
            String dateString = formatter.format(date);
            result.put(key, dateString);
        }
        return result;
    }

    /**
     * 將時間戳記轉換為時間字串
     * @param input 時間戳記Struct
     * @return 時間字串Struct
     */
    public Map<String, String> evaluate(Struct input) {
        if (input == null) {
            return null;
        }
        SimpleDateFormat formatter = new SimpleDateFormat(PATTERN);
        String nameValue = (String) input.getFieldValue("input_name");
        Long timestampValue = (Long) input.getFieldValue("input_timestamp");
        Date date = new Date(timestampValue < 9999999999L ? timestampValue * 1000 : timestampValue);
        String dateString = formatter.format(date);
        Map<String, String> result = new HashMap<>(8);
        result.put("output_name", nameValue);
        result.put("output_time", dateString);
        return result;
    }
}

範例程式碼中，定義了3個重載的evaluate方法。其中：

• 第一個用ARRAY作為參數，ARRAY對應java.util.List。

• 第二個用MAP作為參數，MAP對應java.util.Map。

• 第三個用STRUCT作為參數，STRUCT對應com.aliyun.odps.data.Struct。

  說明

  com.aliyun.odps.data.Struct無法通過反射分析擷取到field name和field type，需要輔助使用@Resolve annotation，即如果您需要在UDF中使用STRUCT，要求在UDF class上也標註上@Resolve註解，該註解只會影響參數或傳回值中包含com.aliyun.odps.data.Struct的重載。

使用Java語言編寫UDF代碼必須繼承UDF類，本例中evaluate方法定義了三個string類型的入參和string類型的傳回值，輸入參數和傳回值的資料類型將作為SQL語句中UDF的函數簽名Signature，其他代碼規範和要求請參考：UDF開發規範與通用流程（Java）。

Python3 UDF 程式碼範例

下面樣本中輸入資料類型為map<string，bigint>，該範例程式碼在本文中將註冊為自訂函數：UDF_COMPLEX_DATA_MAP。

from odps.udf import annotate
import datetime
@annotate('map<string,bigint>->map<string,datetime>')
class MapExample:
    def evaluate(self, intput_dict):
        output_dict = dict()
        for key in intput_dict:
            value = intput_dict[key]
            t = datetime.datetime.fromtimestamp(value)
            output_dict[key] = t
        return output_dict

下面樣本中輸入資料類型為array<bigint>，該範例程式碼在本文中將註冊為自訂函數：UDF_COMPLEX_DATA_ARRAY。

from odps.udf import annotate
import datetime
@annotate('array<bigint>->array<datetime>')
class ArrayExample:
    def evaluate(self, intput_list):
        output_list = list()
        for item in intput_list:
            t = datetime.datetime.fromtimestamp(item)
            output_list.append(t)
        return output_list

下面樣本中輸入資料類型為struct<input_name:string,input_timestamp:bigint>，該範例程式碼在本文中將註冊為自訂函數：UDF_COMPLEX_DATA_STRUCT。

from odps.udf import annotate
import datetime, collections
@annotate('struct<input_name:string,input_timestamp:bigint>->struct<output_name:string,output_time:datetime>')
class StructExample:
    def evaluate(self, intput_namedtuple):
        OutputNamedTuple = collections.namedtuple('output_namedtuple', ['output_name', 'output_time'])
        name_val = intput_namedtuple.input_name
        time_val = datetime.datetime.fromtimestamp(intput_namedtuple.input_timestamp)
        output_namedtuple = OutputNamedTuple(name_val, time_val)
        return output_namedtuple

MaxCompute預設使用Python 2，可以在Session層級使用命令set odps.sql.python.version=cp37開啟Python 3。更多python3 UDF規範請參考：UDF開發規範與通用流程（Python3）。