ThriftTypesConvert.h File Reference

#include "gen-cpp/heavy_types.h"
#include "Logger/Logger.h"
#include <string>

Include dependency graph for ThriftTypesConvert.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros
#define	THRIFT_ENCODING_CASE(encoding)
#define	UNTHRIFT_ENCODING_CASE(encoding)

Functions
TDatumType::type	type_to_thrift (const SQLTypeInfo &type_info)
SQLTypes	thrift_to_type (const TDatumType::type &type)
TEncodingType::type	encoding_to_thrift (const SQLTypeInfo &type_info)
EncodingType	thrift_to_encoding (const TEncodingType::type tEncodingType)
std::string	thrift_to_name (const TTypeInfo &ti)
std::string	thrift_to_encoding_name (const TTypeInfo &ti)
SQLTypeInfo	type_info_from_thrift (const TTypeInfo &thrift_ti, const bool strip_geo_encoding=false)

Macro Definition Documentation

#define THRIFT_ENCODING_CASE

(

encoding

)

Value:

case kENCODING_##encoding:           \
    return TEncodingType::encoding;

Definition at line 137 of file ThriftTypesConvert.h.

Referenced by encoding_to_thrift().

#define UNTHRIFT_ENCODING_CASE

(

encoding

)

Value:

case TEncodingType::encoding:          \
    return kENCODING_##encoding;

Definition at line 141 of file ThriftTypesConvert.h.

Referenced by thrift_to_encoding().

Function Documentation

TEncodingType::type encoding_to_thrift ( const SQLTypeInfo &  type_info )

inline

Definition at line 145 of file ThriftTypesConvert.h.

References CHECK, SQLTypeInfo::get_compression(), NONE, and THRIFT_ENCODING_CASE.

Referenced by DBHandler::detect_column_types(), DBHandler::populateThriftColumnType(), ThriftSerializers::target_meta_info_to_thrift(), and ThriftSerializers::type_info_to_thrift().

                                                                          {
  switch (type_info.get_compression()) {
    THRIFT_ENCODING_CASE(NONE)
    THRIFT_ENCODING_CASE(FIXED)
    THRIFT_ENCODING_CASE(RL)
    THRIFT_ENCODING_CASE(DIFF)
    THRIFT_ENCODING_CASE(DICT)
    THRIFT_ENCODING_CASE(SPARSE)
    THRIFT_ENCODING_CASE(GEOINT)
    THRIFT_ENCODING_CASE(DATE_IN_DAYS)
    default:
      CHECK(false);
  }
  abort();
}

Here is the call graph for this function:

Here is the caller graph for this function:

EncodingType thrift_to_encoding ( const TEncodingType::type  tEncodingType )

inline

Definition at line 163 of file ThriftTypesConvert.h.

References CHECK, NONE, and UNTHRIFT_ENCODING_CASE.

Referenced by DBHandler::create_table(), thrift_to_encoding_name(), thrift_to_name(), and type_info_from_thrift().

                                                                              {
  switch (tEncodingType) {
    UNTHRIFT_ENCODING_CASE(NONE)
    UNTHRIFT_ENCODING_CASE(FIXED)
    UNTHRIFT_ENCODING_CASE(RL)
    UNTHRIFT_ENCODING_CASE(DIFF)
    UNTHRIFT_ENCODING_CASE(DICT)
    UNTHRIFT_ENCODING_CASE(SPARSE)
    UNTHRIFT_ENCODING_CASE(GEOINT)
    UNTHRIFT_ENCODING_CASE(DATE_IN_DAYS)
    default:
      CHECK(false);
  }
  abort();
}

Here is the caller graph for this function:

std::string thrift_to_encoding_name ( const TTypeInfo &  ti )

inline

Definition at line 204 of file ThriftTypesConvert.h.

References kARRAY, kNULLT, thrift_to_encoding(), thrift_to_type(), and run_benchmark_import::type.

Referenced by DBHandler::create_table().

                                                              {
  const auto type = thrift_to_type(ti.type);
  const auto encoding = thrift_to_encoding(ti.encoding);
  auto internal_ti = SQLTypeInfo(ti.is_array ? kARRAY : type,
                                 0,
                                 0,
                                 !ti.nullable,
                                 encoding,
                                 0,
                                 ti.is_array ? type : kNULLT);
  return internal_ti.get_compression_name();
}

Here is the call graph for this function:

Here is the caller graph for this function:

std::string thrift_to_name ( const TTypeInfo &  ti )

inline

Definition at line 179 of file ThriftTypesConvert.h.

References IS_GEO, kARRAY, kDECIMAL, kNULLT, kNUMERIC, kTIMESTAMP, thrift_to_encoding(), thrift_to_type(), and run_benchmark_import::type.

Referenced by DBHandler::create_table().

                                                     {
  const auto type = thrift_to_type(ti.type);
  const auto encoding = thrift_to_encoding(ti.encoding);
  auto internal_ti = SQLTypeInfo(ti.is_array ? kARRAY : type,
                                 0,
                                 0,
                                 !ti.nullable,
                                 encoding,
                                 ti.comp_param,
                                 ti.is_array ? type : kNULLT);
  if (type == kDECIMAL || type == kNUMERIC) {
    internal_ti.set_precision(ti.precision);
    internal_ti.set_scale(ti.scale);
  } else if (type == kTIMESTAMP) {
    internal_ti.set_precision(ti.precision);
  }
  if (IS_GEO(type)) {
    internal_ti.set_subtype(static_cast<SQLTypes>(ti.precision));
    internal_ti.set_input_srid(ti.scale);
    internal_ti.set_output_srid(ti.scale);
  }
  internal_ti.set_size(ti.size);
  return internal_ti.get_type_name();
}

Here is the call graph for this function:

Here is the caller graph for this function:

SQLTypes thrift_to_type ( const TDatumType::type &  type )

inline

Definition at line 85 of file ThriftTypesConvert.h.

References kBIGINT, kBOOLEAN, kDATE, kDECIMAL, kDOUBLE, kFLOAT, kGEOGRAPHY, kGEOMETRY, kINT, kINTERVAL_DAY_TIME, kINTERVAL_YEAR_MONTH, kLINESTRING, kMULTILINESTRING, kMULTIPOINT, kMULTIPOLYGON, kPOINT, kPOLYGON, kSMALLINT, kTEXT, kTIME, kTIMESTAMP, and kTINYINT.

Referenced by thrift_to_encoding_name(), thrift_to_name(), and type_info_from_thrift().

                                                           {
  switch (type) {
    case TDatumType::BOOL:
      return kBOOLEAN;
    case TDatumType::TINYINT:
      return kTINYINT;
    case TDatumType::SMALLINT:
      return kSMALLINT;
    case TDatumType::INT:
      return kINT;
    case TDatumType::BIGINT:
      return kBIGINT;
    case TDatumType::FLOAT:
      return kFLOAT;
    case TDatumType::DECIMAL:
      return kDECIMAL;
    case TDatumType::DOUBLE:
      return kDOUBLE;
    case TDatumType::STR:
      return kTEXT;
    case TDatumType::TIME:
      return kTIME;
    case TDatumType::TIMESTAMP:
      return kTIMESTAMP;
    case TDatumType::DATE:
      return kDATE;
    case TDatumType::INTERVAL_DAY_TIME:
      return kINTERVAL_DAY_TIME;
    case TDatumType::INTERVAL_YEAR_MONTH:
      return kINTERVAL_YEAR_MONTH;
    case TDatumType::POINT:
      return kPOINT;
    case TDatumType::MULTIPOINT:
      return kMULTIPOINT;
    case TDatumType::LINESTRING:
      return kLINESTRING;
    case TDatumType::MULTILINESTRING:
      return kMULTILINESTRING;
    case TDatumType::POLYGON:
      return kPOLYGON;
    case TDatumType::MULTIPOLYGON:
      return kMULTIPOLYGON;
    case TDatumType::GEOMETRY:
      return kGEOMETRY;
    case TDatumType::GEOGRAPHY:
      return kGEOGRAPHY;
    default:
      break;
  }
  abort();
}

Here is the caller graph for this function:

SQLTypeInfo type_info_from_thrift ( const TTypeInfo &  thrift_ti, const bool  strip_geo_encoding = false  )

inline

Definition at line 217 of file ThriftTypesConvert.h.

References CHECK_LT, SQLTypeInfo::is_dict_encoded_string(), IS_GEO, SQLTypeInfo::is_subtype_dict_encoded_string(), kARRAY, kENCODING_NONE, kNULLT, kSQLTYPE_LAST, SQLTypeInfo::set_size(), SQLTypeInfo::setStringDictKey(), thrift_to_encoding(), and thrift_to_type().

Referenced by ThriftSerializers::target_info_from_thrift(), and ThriftSerializers::target_meta_infos_from_thrift().

                                                                                {
  SQLTypeInfo type_info;
  const auto ti = thrift_to_type(thrift_ti.type);
  if (IS_GEO(ti)) {
    const auto base_type = static_cast<SQLTypes>(thrift_ti.precision);
    CHECK_LT(base_type, kSQLTYPE_LAST);
    type_info = SQLTypeInfo(
        ti,
        thrift_ti.scale,
        thrift_ti.scale,
        !thrift_ti.nullable,
        strip_geo_encoding ? kENCODING_NONE : thrift_to_encoding(thrift_ti.encoding),
        thrift_ti.comp_param,
        base_type);
  } else if (thrift_ti.is_array) {
    type_info = SQLTypeInfo(kARRAY,
                            thrift_ti.precision,
                            thrift_ti.scale,
                            !thrift_ti.nullable,
                            thrift_to_encoding(thrift_ti.encoding),
                            thrift_ti.comp_param,
                            ti);
    type_info.set_size(thrift_ti.size);
  } else {
    type_info = SQLTypeInfo(ti,
                            thrift_ti.precision,
                            thrift_ti.scale,
                            !thrift_ti.nullable,
                            thrift_to_encoding(thrift_ti.encoding),
                            thrift_ti.comp_param,
                            kNULLT);
  }
  if (type_info.is_dict_encoded_string() || type_info.is_subtype_dict_encoded_string()) {
    const auto& dict_key = thrift_ti.dict_key;
    type_info.setStringDictKey({dict_key.db_id, dict_key.dict_id});
  }
  return type_info;
}

Here is the call graph for this function:

Here is the caller graph for this function:

TDatumType::type type_to_thrift ( const SQLTypeInfo &  type_info )

inline

Definition at line 26 of file ThriftTypesConvert.h.

References SQLTypeInfo::get_subtype(), SQLTypeInfo::get_type(), kARRAY, kBIGINT, kBOOLEAN, kCHAR, kCOLUMN, kCOLUMN_LIST, kDATE, kDECIMAL, kDOUBLE, kFLOAT, kGEOGRAPHY, kGEOMETRY, kINT, kINTERVAL_DAY_TIME, kINTERVAL_YEAR_MONTH, kLINESTRING, kMULTILINESTRING, kMULTIPOINT, kMULTIPOLYGON, kNUMERIC, kPOINT, kPOLYGON, kSMALLINT, kTEXT, kTIME, kTIMESTAMP, kTINYINT, kVARCHAR, and run_benchmark_import::type.

Referenced by DBHandler::detect_column_types(), DBHandler::populateThriftColumnType(), ThriftSerializers::target_meta_info_to_thrift(), and ThriftSerializers::type_info_to_thrift().

                                                                   {
  SQLTypes type = type_info.get_type();
  if (type == kARRAY || type == kCOLUMN || type == kCOLUMN_LIST) {
    type = type_info.get_subtype();
  }
  switch (type) {
    case kBOOLEAN:
      return TDatumType::BOOL;
    case kTINYINT:
      return TDatumType::TINYINT;
    case kSMALLINT:
      return TDatumType::SMALLINT;
    case kINT:
      return TDatumType::INT;
    case kBIGINT:
      return TDatumType::BIGINT;
    case kFLOAT:
      return TDatumType::FLOAT;
    case kNUMERIC:
    case kDECIMAL:
      return TDatumType::DECIMAL;
    case kDOUBLE:
      return TDatumType::DOUBLE;
    case kTEXT:
    case kVARCHAR:
    case kCHAR:
      return TDatumType::STR;
    case kTIME:
      return TDatumType::TIME;
    case kTIMESTAMP:
      return TDatumType::TIMESTAMP;
    case kDATE:
      return TDatumType::DATE;
    case kINTERVAL_DAY_TIME:
      return TDatumType::INTERVAL_DAY_TIME;
    case kINTERVAL_YEAR_MONTH:
      return TDatumType::INTERVAL_YEAR_MONTH;
    case kPOINT:
      return TDatumType::POINT;
    case kMULTIPOINT:
      return TDatumType::MULTIPOINT;
    case kLINESTRING:
      return TDatumType::LINESTRING;
    case kMULTILINESTRING:
      return TDatumType::MULTILINESTRING;
    case kPOLYGON:
      return TDatumType::POLYGON;
    case kMULTIPOLYGON:
      return TDatumType::MULTIPOLYGON;
    case kGEOMETRY:
      return TDatumType::GEOMETRY;
    case kGEOGRAPHY:
      return TDatumType::GEOGRAPHY;
    default:
      break;
  }
  abort();
}

Here is the call graph for this function:

Here is the caller graph for this function: