spark SpecializedGettersReader 源码

spark SpecializedGettersReader 代码

文件路径：/sql/catalyst/src/main/java/org/apache/spark/sql/catalyst/expressions/SpecializedGettersReader.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.spark.sql.catalyst.expressions;

import org.apache.spark.sql.types.*;

public final class SpecializedGettersReader {
  private SpecializedGettersReader() {}

  public static Object read(
      SpecializedGetters obj,
      int ordinal,
      DataType dataType,
      boolean handleNull,
      boolean handleUserDefinedType) {
    if (handleNull && (obj.isNullAt(ordinal) || dataType instanceof NullType)) {
      return null;
    }
    if (dataType instanceof BooleanType) {
      return obj.getBoolean(ordinal);
    }
    if (dataType instanceof ByteType) {
      return obj.getByte(ordinal);
    }
    if (dataType instanceof ShortType) {
      return obj.getShort(ordinal);
    }
    if (dataType instanceof IntegerType) {
      return obj.getInt(ordinal);
    }
    if (dataType instanceof LongType) {
      return obj.getLong(ordinal);
    }
    if (dataType instanceof FloatType) {
      return obj.getFloat(ordinal);
    }
    if (dataType instanceof DoubleType) {
      return obj.getDouble(ordinal);
    }
    if (dataType instanceof StringType) {
      return obj.getUTF8String(ordinal);
    }
    if (dataType instanceof DecimalType) {
      DecimalType dt = (DecimalType) dataType;
      return obj.getDecimal(ordinal, dt.precision(), dt.scale());
    }
    if (dataType instanceof DateType) {
      return obj.getInt(ordinal);
    }
    if (dataType instanceof TimestampType) {
      return obj.getLong(ordinal);
    }
    if (dataType instanceof TimestampNTZType) {
      return obj.getLong(ordinal);
    }
    if (dataType instanceof CalendarIntervalType) {
      return obj.getInterval(ordinal);
    }
    if (dataType instanceof BinaryType) {
      return obj.getBinary(ordinal);
    }
    if (dataType instanceof StructType) {
      return obj.getStruct(ordinal, ((StructType) dataType).size());
    }
    if (dataType instanceof ArrayType) {
      return obj.getArray(ordinal);
    }
    if (dataType instanceof MapType) {
      return obj.getMap(ordinal);
    }
    if (handleUserDefinedType && dataType instanceof UserDefinedType) {
      return obj.get(ordinal, ((UserDefinedType)dataType).sqlType());
    }
    if (dataType instanceof DayTimeIntervalType) {
      return obj.getLong(ordinal);
    }
    if (dataType instanceof YearMonthIntervalType) {
      return obj.getInt(ordinal);
    }

    throw new UnsupportedOperationException("Unsupported data type " + dataType.simpleString());
  }
}

相关信息

spark 源码目录

相关文章

spark ExpressionDescription 源码

spark ExpressionImplUtils 源码

spark ExpressionInfo 源码

spark FixedLengthRowBasedKeyValueBatch 源码

spark RowBasedKeyValueBatch 源码

spark SpecializedGetters 源码

spark UnsafeArrayData 源码

spark UnsafeDataUtils 源码

spark UnsafeMapData 源码

spark UnsafeRow 源码