ApplyOperation

이 섹션의 코드는 ApplyOperation.java라는 파일에 저장됩니다.

개념

이것은 다음과 같이 수정된 "ApplyOp" 예제의 변형입니다:

데이터 처리 로직이 연결 생성 로직에서 분리되었습니다. 독립형 클래스는 서로 다른 연결 로직을 사용하는 여러 AE 예제 애플리케이션에서 재사용됩니다.
이 AE는 콜백 패러다임 대신 파일-IO 패러다임을 사용하여 입력 행을 검색합니다. 콜백 패러다임의 예는 Java 언어 스칼라 함수의 MyHandler 참조하세요.
이 코드는 뺄셈 연산자를 지원합니다.

이 함수 AE는 입력 행당 정확히 하나의 출력 행을 쓰고 출력 행에는 정확히 하나의 열이 포함되므로 스칼라 또는 테이블 함수에서 AE를 호출할 수 있다는 점에 유의하세요.

코드

// ApplyOperation.java
// Function AE
// Performs arithmetic operations (+, -, or *)
// across columns in a row
// Ignores non-numeric columns
// First argument column must be a string type operation
// Handles a variable number of input columns and data types
// Returns a double result
// Can be called as a table function or a scalar function
package org.netezza.education;

import java.io.*;
import java.math.BigDecimal;
import java.text.DateFormat;
import java.sql.Date;
import java.sql.Time;
import java.sql.Timestamp;
import org.netezza.ae.*;

public class ApplyOperation {
    private static final int OP_ADD = 1;
    private static final int OP_MULT = 2;
    private static final int OP_SUBTRACT = 3;

    public static void runAe(Nzae ae) {
        final NzaeMetadata meta = ae.getMetadata();

        for (;;) {
            int op = 0;
            double result = 0;
            NzaeRecord input = ae.next();
            if (input == null) {
                break;
            }
            NzaeRecord output = ae.createOutputRecord();

            for (int i = 0; i < input.size(); i++) {
                if (input.getField(i) == null) {
                    continue;
            }

                int dataType = meta.getInputNzType(i);

                if (i == 0) {
                    if (!(dataType == NzaeDataTypes.NZUDSUDX_FIXED
                            || dataType ==
                                NzaeDataTypes.NZUDSUDX_VARIABLE)) {
                        ae.userError(
                            "first column must be a string");
                    }
                    String opStr = input.getFieldAsString(0);
                    if (opStr.equals("*")) {
                        result = 1;
                        op = OP_MULT;
                    } else if (opStr.equals("+")) {
                        result = 0;
                        op = OP_ADD;
                    } else if (opStr.equals("-")) {
                       result = 0;
                       op = OP_SUBTRACT;
                    } else {
                       ae.userError("invalid operator = " + op);
                    }
                    continue;
                }

                switch (dataType) {
                case NzaeDataTypes.NZUDSUDX_INT8:
                case NzaeDataTypes.NZUDSUDX_INT16:
                case NzaeDataTypes.NZUDSUDX_INT32:
                case NzaeDataTypes.NZUDSUDX_INT64:
                case NzaeDataTypes.NZUDSUDX_FLOAT:
                case NzaeDataTypes.NZUDSUDX_DOUBLE:
               case NzaeDataTypes.NZUDSUDX_NUMERIC32:
               case NzaeDataTypes.NZUDSUDX_NUMERIC64:
               case NzaeDataTypes.NZUDSUDX_NUMERIC128:
                   switch (op) {
                   case OP_ADD:
                       result +=
                           input.getFieldAsNumber(i).doubleValue();
                       break;
                   case OP_SUBTRACT:
                       result -=
                           input.getFieldAsNumber(i).doubleValue();
                       break;
                   case OP_MULT:
                       result *=
                           input.getFieldAsNumber(i).doubleValue();
                       break;
                   default:
                       break;
                   }
                   break;

               default:
                   break;
               }
           }   // end of column for loop

           output.setField(0, result);
           ae.outputResult(output);
       }

       ae.done();
    }
}