table_functions::anonymous_namespace{TableFunctionsFactory.cpp} Namespace Reference

Functions
SQLTypeInfo	ext_arg_pointer_type_to_type_info (const ExtArgumentType ext_arg_type)
SQLTypeInfo	ext_arg_type_to_type_info_output (const ExtArgumentType ext_arg_type)
template<size_t... I>
constexpr bool	string_view_array_sorted_and_distinct (std::string_view const *list, std::index_sequence< I...>)
bool	is_table_function_whitelisted (std::string_view const function_name)
std::string	drop_suffix_impl (const std::string &str)

Function Documentation

std::string table_functions::anonymous_namespace{TableFunctionsFactory.cpp}::drop_suffix_impl

(

const std::string &

str

)

Definition at line 552 of file TableFunctionsFactory.cpp.

References CHECK_GT.

                                                 {
  const auto idx = str.find("__");
  if (idx == std::string::npos) {
    return str;
  }
  CHECK_GT(idx, std::string::size_type(0));
  return str.substr(0, idx);
}

SQLTypeInfo table_functions::anonymous_namespace{TableFunctionsFactory.cpp}::ext_arg_pointer_type_to_type_info

(

const ExtArgumentType

ext_arg_type

)

Definition at line 34 of file TableFunctionsFactory.cpp.

References ext_arg_type_to_type_info(), kBIGINT, kBOOLEAN, kDOUBLE, kFLOAT, kINT, kNULLT, kSMALLINT, kTINYINT, PBool, PDouble, PFloat, PInt16, PInt32, PInt64, PInt8, and UNREACHABLE.

Referenced by table_functions::TableFunction::getInputSQLType().

                                                                                  {
  switch (ext_arg_type) {
    case ExtArgumentType::PInt8:
      return SQLTypeInfo(kTINYINT, false);
    case ExtArgumentType::PInt16:
      return SQLTypeInfo(kSMALLINT, false);
    case ExtArgumentType::PInt32:
      return SQLTypeInfo(kINT, false);
    case ExtArgumentType::PInt64:
      return SQLTypeInfo(kBIGINT, false);
    case ExtArgumentType::PFloat:
      return SQLTypeInfo(kFLOAT, false);
    case ExtArgumentType::PDouble:
      return SQLTypeInfo(kDOUBLE, false);
    case ExtArgumentType::PBool:
      return SQLTypeInfo(kBOOLEAN, false);
    default:
      return ext_arg_type_to_type_info(ext_arg_type);
  }
  UNREACHABLE();
  return SQLTypeInfo(kNULLT, false);
}

Here is the call graph for this function:

Here is the caller graph for this function:

SQLTypeInfo table_functions::anonymous_namespace{TableFunctionsFactory.cpp}::ext_arg_type_to_type_info_output

(

const ExtArgumentType

ext_arg_type

)

Definition at line 57 of file TableFunctionsFactory.cpp.

References ext_arg_type_to_type_info(), SQLTypeInfo::get_elem_type(), kBIGINT, kBOOLEAN, kDOUBLE, kFLOAT, kINT, kNULLT, kSMALLINT, kTINYINT, PBool, PDouble, PFloat, PInt16, PInt32, PInt64, PInt8, and UNREACHABLE.

Referenced by table_functions::TableFunction::getOutputSQLType().

                                                                                 {
  switch (ext_arg_type) {
    case ExtArgumentType::PInt8:
      return SQLTypeInfo(kTINYINT, false);
    case ExtArgumentType::PInt16:
      return SQLTypeInfo(kSMALLINT, false);
    case ExtArgumentType::PInt32:
      return SQLTypeInfo(kINT, false);
    case ExtArgumentType::PInt64:
      return SQLTypeInfo(kBIGINT, false);
    case ExtArgumentType::PFloat:
      return SQLTypeInfo(kFLOAT, false);
    case ExtArgumentType::PDouble:
      return SQLTypeInfo(kDOUBLE, false);
    case ExtArgumentType::PBool:
      return SQLTypeInfo(kBOOLEAN, false);
    default:
      return ext_arg_type_to_type_info(ext_arg_type).get_elem_type();
  }
  UNREACHABLE();
  return SQLTypeInfo(kNULLT, false);
}

Here is the call graph for this function:

Here is the caller graph for this function:

bool table_functions::anonymous_namespace{TableFunctionsFactory.cpp}::is_table_function_whitelisted

(

std::string_view const

function_name

)

Definition at line 358 of file TableFunctionsFactory.cpp.

References g_enable_ml_functions, and string_view_array_sorted_and_distinct().

Referenced by table_functions::TableFunctionsFactory::add().

                                                                       {
  // All table functions that will be on by default (and not just for testing)
  // must be added to the whitelisted_table_functions list below.
  // These lists must be in alphabetical order (enforced at compile-time)
  constexpr std::string_view whitelisted_table_functions[]{
       "dbscan",
       "decision_tree_reg_fit",
       "gbt_reg_fit",
       "generate_random_strings",
       "generate_series",
       "get_decision_trees",
       "kmeans",
       "linear_reg_coefs",
       "linear_reg_fit",
       "ml_reg_predict",
       "pca_fit",
       "r2_score",
       "random_forest_reg_fit",
       "random_forest_reg_var_importance",
       "supported_ml_frameworks",
       "tf_compute_dwell_times",
       "tf_cross_section_1d",
       "tf_cross_section_2d",
       "tf_feature_self_similarity",
       "tf_feature_similarity",
       "tf_geo_multi_rasterize",
       "tf_geo_rasterize",
       "tf_geo_rasterize_slope",
       "tf_graph_shortest_path",
       "tf_graph_shortest_paths_distances",
       "tf_load_point_cloud",
       "tf_mandelbrot",
       "tf_mandelbrot_cuda",
       "tf_mandelbrot_cuda_float",
       "tf_mandelbrot_float",
       "tf_point_cloud_metadata",
       "tf_raster_contour_lines",
       "tf_raster_contour_polygons",
       "tf_raster_graph_shortest_slope_weighted_path"
    };
    constexpr auto whitelisted_table_functions_len =
       sizeof(whitelisted_table_functions) / sizeof(*whitelisted_table_functions);


  // Must be sorted.
  constexpr std::string_view ml_table_functions[]{"dbscan",
                                                  "decision_tree_reg_fit",
                                                  "gbt_reg_fit",
                                                  "get_decision_trees",
                                                  "kmeans",
                                                  "linear_reg_coefs",
                                                  "linear_reg_fit",
                                                  "ml_reg_predict",
                                                  "pca_fit"
                                                  "r2_score",
                                                  "random_forest_reg_fit",
                                                  "random_forest_reg_var_importance",
                                                  "supported_ml_frameworks"};
  constexpr auto ml_table_functions_len =
      sizeof(ml_table_functions) / sizeof(*ml_table_functions);

  // compile-time validation that above lists are sorted and distinct
  static_assert(string_view_array_sorted_and_distinct(
      whitelisted_table_functions,
      std::make_index_sequence<whitelisted_table_functions_len - 1>{}));
  static_assert(string_view_array_sorted_and_distinct(
      ml_table_functions, std::make_index_sequence<ml_table_functions_len - 1>{}));

  if (!std::binary_search(whitelisted_table_functions,
                          whitelisted_table_functions + whitelisted_table_functions_len,
                          function_name)) {
    return false;
  }
  return g_enable_ml_functions ||
         !std::binary_search(ml_table_functions,
                             ml_table_functions + ml_table_functions_len,
                             function_name);
}

Here is the call graph for this function:

Here is the caller graph for this function:

template<size_t... I>

constexpr bool table_functions::anonymous_namespace{TableFunctionsFactory.cpp}::string_view_array_sorted_and_distinct	(	std::string_view const *	list,
		std::index_sequence< I...>
	)

Definition at line 353 of file TableFunctionsFactory.cpp.

Referenced by is_table_function_whitelisted().

                                                                              {
  return ((list[I] < list[I + 1]) && ...);
}

Here is the caller graph for this function: