anonymous_namespace{ArrowResultSet.cpp} Namespace Reference

Functions
SQLTypeInfo	type_from_arrow_field (const arrow::Field &field)

Function Documentation

SQLTypeInfo anonymous_namespace{ArrowResultSet.cpp}::type_from_arrow_field

(

const arrow::Field &

field

)

Definition at line 28 of file ArrowResultSet.cpp.

References CHECK, field(), kBIGINT, kDATE, kDOUBLE, kENCODING_DATE_IN_DAYS, kENCODING_DICT, kFLOAT, kINT, kSMALLINT, kTEXT, kTIME, kTIMESTAMP, kTINYINT, run_benchmark_import::type, SQLTypeInfo::type, and UNREACHABLE.

Referenced by ArrowResultSet::ArrowResultSet().

                                                           {
  switch (field.type()->id()) {
    case arrow::Type::INT8:
      return SQLTypeInfo(kTINYINT, !field.nullable());
    case arrow::Type::INT16:
      return SQLTypeInfo(kSMALLINT, !field.nullable());
    case arrow::Type::INT32:
      return SQLTypeInfo(kINT, !field.nullable());
    case arrow::Type::INT64:
      return SQLTypeInfo(kBIGINT, !field.nullable());
    case arrow::Type::FLOAT:
      return SQLTypeInfo(kFLOAT, !field.nullable());
    case arrow::Type::DOUBLE:
      return SQLTypeInfo(kDOUBLE, !field.nullable());
    case arrow::Type::DICTIONARY:
      return SQLTypeInfo(kTEXT, !field.nullable(), kENCODING_DICT);
    case arrow::Type::TIMESTAMP: {
      // TODO(Wamsi): go right fold expr in c++17
      auto get_precision = [&field](auto type) { return field.type()->Equals(type); };
      if (get_precision(arrow::timestamp(arrow::TimeUnit::SECOND))) {
        return SQLTypeInfo(kTIMESTAMP, !field.nullable());
      } else if (get_precision(arrow::timestamp(arrow::TimeUnit::MILLI))) {
        return SQLTypeInfo(kTIMESTAMP, 3, 0, !field.nullable());
      } else if (get_precision(arrow::timestamp(arrow::TimeUnit::MICRO))) {
        return SQLTypeInfo(kTIMESTAMP, 6, 0, !field.nullable());
      } else if (get_precision(arrow::timestamp(arrow::TimeUnit::NANO))) {
        return SQLTypeInfo(kTIMESTAMP, 9, 0, !field.nullable());
      } else {
        UNREACHABLE();
      }
    }
    case arrow::Type::DATE32:
      return SQLTypeInfo(kDATE, !field.nullable(), kENCODING_DATE_IN_DAYS);
    case arrow::Type::DATE64:
      return SQLTypeInfo(kDATE, !field.nullable());
    case arrow::Type::TIME32:
      return SQLTypeInfo(kTIME, !field.nullable());
    default:
      CHECK(false);
  }
  CHECK(false);
  return SQLTypeInfo();
}

Here is the call graph for this function:

Here is the caller graph for this function: