UDF Java Examples

Sample Code Block

A simple distributed example of a UDF which copies data from one set of tables to another would look like the following:

/* Sourced from:  gpudb-udf-api-java/proc-example/src/main/java/com/kinetica/ProcExample.java */

package com.kinetica;

public class ProcExample {
    public static void main(String[] args) {
        ProcData procData = ProcData.get();

        // Loop through input and output tables (assume the same number)
        for (int i = 0; i < procData.getInputData().getTableCount(); i++) {
            ProcData.InputTable inputTable = procData.getInputData().getTable(i);
            ProcData.OutputTable outputTable = procData.getOutputData().getTable(i);
            outputTable.setSize(inputTable.getSize());

            // Loop through columns in the input and output tables (assume the same number and types)
            for (int j = 0; j < inputTable.getColumnCount(); j++) {
                ProcData.InputColumn inputColumn = inputTable.getColumn(j);
                ProcData.OutputColumn outputColumn = outputTable.getColumn(j);

                // For each record, copy the data from the input column to the output column
                for (long k = 0; k < inputTable.getSize(); k++) {
                    switch (inputColumn.getType()) {
                        case BYTES: outputColumn.appendVarBytes(inputColumn.getVarBytes(k)); break;
                        case CHAR1: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR2: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR4: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR8: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR16: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR32: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR64: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR128: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case CHAR256: outputColumn.appendChar(inputColumn.getChar(k)); break;
                        case DATE: outputColumn.appendCalendar(inputColumn.getCalendar(k)); break;
                        case DATETIME: outputColumn.appendCalendar(inputColumn.getCalendar(k)); break;
                        case DECIMAL: outputColumn.appendBigDecimal(inputColumn.getBigDecimal(k)); break;
                        case DOUBLE: outputColumn.appendDouble(inputColumn.getDouble(k)); break;
                        case FLOAT: outputColumn.appendFloat(inputColumn.getFloat(k)); break;
                        case INT: outputColumn.appendInt(inputColumn.getInt(k)); break;
                        case INT8: outputColumn.appendByte(inputColumn.getByte(k)); break;
                        case INT16: outputColumn.appendShort(inputColumn.getShort(k)); break;
                        case IPV4: outputColumn.appendInet4Address(inputColumn.getInet4Address(k)); break;
                        case LONG: outputColumn.appendLong(inputColumn.getLong(k)); break;
                        case STRING: outputColumn.appendVarString(inputColumn.getVarString(k)); break;
                        case TIME: outputColumn.appendCalendar(inputColumn.getCalendar(k)); break;
                        case TIMESTAMP: outputColumn.appendLong(inputColumn.getLong(k)); break;
                    }
                }
            }
        }

        // Copy any parameters from the input parameter map into the output results map (not necessary for table copying, just for illustrative purposes)
        procData.getResults().putAll(procData.getParams());
        procData.getBinResults().putAll(procData.getBinParams());

        // Inform Kinetica that the proc has finished successfully
        procData.complete();
    }
}