OmniSciDB  72c90bc290
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
com.mapd.parser.server.ExtensionFunctionSignatureParser Class Reference
+ Collaboration diagram for com.mapd.parser.server.ExtensionFunctionSignatureParser:

Static Package Functions

static Map< String,
ExtensionFunction
parse (final String file_path) throws IOException
 
static Map< String,
ExtensionFunction
parseUdfAst (final String file_path) throws IOException
 
static Map< String,
ExtensionFunction
parseFromString (final String udf_string) throws IOException
 
static Map< String,
ExtensionFunction
parseFromString (final String udf_string, final boolean is_row_func) throws IOException
 
static String signaturesToJson (final Map< String, ExtensionFunction > sigs)
 
static String join (final List< String > strs, final String sep)
 

Static Package Attributes

static final Logger HEAVYDBLOGGER
 

Static Private Member Functions

static boolean isWindows ()
 
static ExtensionFunction toSignature (final String ret, final String cs_param_list, final boolean has_variable_name)
 
static ExtensionFunction toSignature (final String ret, final String cs_param_list, final boolean has_variable_name, final boolean is_row_func)
 
static
ExtensionFunction.ExtArgumentType 
deserializeType (final String type_name)
 
static
ExtensionFunction.ExtArgumentType 
pointerType (final ExtensionFunction.ExtArgumentType targetType)
 

Static Private Attributes

static String OS = System.getProperty("os.name").toLowerCase()
 

Detailed Description

Definition at line 36 of file ExtensionFunctionSignatureParser.java.

Member Function Documentation

static ExtensionFunction.ExtArgumentType com.mapd.parser.server.ExtensionFunctionSignatureParser.deserializeType ( final String  type_name)
inlinestaticprivate

Definition at line 166 of file ExtensionFunctionSignatureParser.java.

References Double, com.mapd.parser.server.ExtensionFunctionSignatureParser.isWindows(), and com.mapd.parser.server.ExtensionFunctionSignatureParser.pointerType().

Referenced by com.mapd.parser.server.ExtensionFunctionSignatureParser.toSignature().

167  {
168  final String const_prefix = "const ";
169  final String std_namespace_prefix = "std::";
170 
171  if (type_name.startsWith(const_prefix)) {
172  return deserializeType(type_name.substring(const_prefix.length()));
173  }
174  if (type_name.startsWith(std_namespace_prefix)) {
175  return deserializeType(type_name.substring(std_namespace_prefix.length()));
176  }
177 
178  if (type_name.equals("bool") || type_name.equals("_Bool")) {
179  return ExtensionFunction.ExtArgumentType.Bool;
180  }
181  if (type_name.equals("int8_t") || type_name.equals("char")
182  || type_name.equals("int8")) {
183  return ExtensionFunction.ExtArgumentType.Int8;
184  }
185  if (type_name.equals("int16_t") || type_name.equals("short")
186  || type_name.equals("int16")) {
187  return ExtensionFunction.ExtArgumentType.Int16;
188  }
189  if (type_name.equals("int32_t") || type_name.equals("int")
190  || type_name.equals("int32") || (isWindows() && type_name.equals("long"))) {
191  return ExtensionFunction.ExtArgumentType.Int32;
192  }
193  if (type_name.equals("int64_t") || type_name.equals("size_t")
194  || type_name.equals("int64")) {
195  return ExtensionFunction.ExtArgumentType.Int64;
196  }
197  if ((!isWindows() && type_name.equals("long"))
198  || (isWindows() && type_name.equals("long long"))) {
199  return ExtensionFunction.ExtArgumentType.Int64;
200  }
201  if (type_name.equals("float") || type_name.equals("float32")) {
202  return ExtensionFunction.ExtArgumentType.Float;
203  }
204  if (type_name.equals("double") || type_name.equals("float64")) {
206  }
207  if (type_name.isEmpty() || type_name.equals("void")) {
208  return ExtensionFunction.ExtArgumentType.Void;
209  }
210  if (type_name.endsWith(" *")) {
211  return pointerType(deserializeType(type_name.substring(0, type_name.length() - 2)));
212  }
213  if (type_name.endsWith("*")) {
214  return pointerType(deserializeType(type_name.substring(0, type_name.length() - 1)));
215  }
216  if (type_name.endsWith("&")) {
217  return deserializeType(type_name.substring(0, type_name.length() - 1).trim());
218  }
219  if (type_name.equals("Array<bool>")) {
220  return ExtensionFunction.ExtArgumentType.ArrayBool;
221  }
222  if (type_name.equals("Array<int8_t>") || type_name.equals("Array<char>")) {
223  return ExtensionFunction.ExtArgumentType.ArrayInt8;
224  }
225  if (type_name.equals("Array<int16_t>") || type_name.equals("Array<short>")) {
226  return ExtensionFunction.ExtArgumentType.ArrayInt16;
227  }
228  if (type_name.equals("Array<int32_t>") || type_name.equals("Array<int>")) {
229  return ExtensionFunction.ExtArgumentType.ArrayInt32;
230  }
231  if (type_name.equals("Array<int64_t>") || type_name.equals("Array<size_t>")
232  || type_name.equals("Array<long>")) {
233  return ExtensionFunction.ExtArgumentType.ArrayInt64;
234  }
235  if (type_name.equals("Array<float>")) {
236  return ExtensionFunction.ExtArgumentType.ArrayFloat;
237  }
238  if (type_name.equals("Array<double>")) {
239  return ExtensionFunction.ExtArgumentType.ArrayDouble;
240  }
241  if (type_name.equals("Array<bool>")) {
242  return ExtensionFunction.ExtArgumentType.ArrayBool;
243  }
244  if (type_name.equals("Array<TextEncodingDict>")) {
245  return ExtensionFunction.ExtArgumentType.ArrayTextEncodingDict;
246  }
247  if (type_name.equals("Array<TextEncodingNone>")) {
248  return ExtensionFunction.ExtArgumentType.ArrayTextEncodingNone;
249  }
250  if (type_name.equals("TextEncodingDict")) {
251  return ExtensionFunction.ExtArgumentType.TextEncodingDict;
252  }
253  if (type_name.equals("RowFunctionManager")) {
254  // RowFunctionManager is not actually materialized in udfs
255  // return void as a convenience here to not generate a warning
256  // in the line below
257  return ExtensionFunction.ExtArgumentType.Void;
258  }
259  if (type_name.equals("Timestamp")) {
260  return ExtensionFunction.ExtArgumentType.Timestamp;
261  }
262  if (type_name.equals("TextEncodingNone")) {
263  return ExtensionFunction.ExtArgumentType.TextEncodingNone;
264  }
265  if (type_name.equals("Column<int8_t>") || type_name.equals("Column<char>")) {
266  return ExtensionFunction.ExtArgumentType.ColumnInt8;
267  }
268  if (type_name.equals("Column<int16_t>") || type_name.equals("Column<short>")) {
269  return ExtensionFunction.ExtArgumentType.ColumnInt16;
270  }
271  if (type_name.equals("Column<int32_t>") || type_name.equals("Column<int>")) {
272  return ExtensionFunction.ExtArgumentType.ColumnInt32;
273  }
274  if (type_name.equals("Column<int64_t>") || type_name.equals("Column<size_t>")
275  || type_name.equals("Column<long>")) {
276  return ExtensionFunction.ExtArgumentType.ColumnInt64;
277  }
278  if (type_name.equals("Column<float>")) {
279  return ExtensionFunction.ExtArgumentType.ColumnFloat;
280  }
281  if (type_name.equals("Column<double>")) {
282  return ExtensionFunction.ExtArgumentType.ColumnDouble;
283  }
284  if (type_name.equals("Column<TextEncodingDict>")) {
285  return ExtensionFunction.ExtArgumentType.ColumnTextEncodingDict;
286  }
287  if (type_name.equals("Column<Timestamp>")) {
288  return ExtensionFunction.ExtArgumentType.ColumnTimestamp;
289  }
290  if (type_name.equals("Cursor")) {
291  return ExtensionFunction.ExtArgumentType.Cursor;
292  }
293  if (type_name.equals("GeoPoint")) {
295  }
296  if (type_name.equals("GeoMultiPoint")) {
298  }
299  if (type_name.equals("GeoLineString")) {
301  }
302  if (type_name.equals("GeoMultiLineString")) {
304  }
305  if (type_name.equals("GeoPolygon")) {
307  }
308  if (type_name.equals("GeoMultiPolygon")) {
310  }
311  if (type_name.equals("ColumnList<int8_t>") || type_name.equals("ColumnList<char>")) {
312  return ExtensionFunction.ExtArgumentType.ColumnListInt8;
313  }
314  if (type_name.equals("ColumnList<int16_t>")
315  || type_name.equals("ColumnList<short>")) {
316  return ExtensionFunction.ExtArgumentType.ColumnListInt16;
317  }
318  if (type_name.equals("ColumnList<int32_t>") || type_name.equals("ColumnList<int>")) {
319  return ExtensionFunction.ExtArgumentType.ColumnListInt32;
320  }
321  if (type_name.equals("ColumnList<int64_t>") || type_name.equals("ColumnList<size_t>")
322  || type_name.equals("ColumnList<long>")) {
323  return ExtensionFunction.ExtArgumentType.ColumnListInt64;
324  }
325  if (type_name.equals("ColumnList<float>")) {
326  return ExtensionFunction.ExtArgumentType.ColumnListFloat;
327  }
328  if (type_name.equals("ColumnList<double>")) {
329  return ExtensionFunction.ExtArgumentType.ColumnListDouble;
330  }
331  if (type_name.equals("ColumnList<TextEncodingDict>")) {
332  return ExtensionFunction.ExtArgumentType.ColumnListTextEncodingDict;
333  }
334  HEAVYDBLOGGER.info(
335  "ExtensionfunctionSignatureParser::deserializeType: unknown type_name=`"
336  + type_name + "`");
337  // TODO: Return void for convenience. Consider sanitizing functions for supported
338  // types before they reach Calcite
339  return ExtensionFunction.ExtArgumentType.Void;
340  }
struct GeoLineStringStruct GeoLineString
Definition: heavydbTypes.h:999
struct GeoPointStruct GeoPoint
Definition: heavydbTypes.h:963
static ExtensionFunction.ExtArgumentType pointerType(final ExtensionFunction.ExtArgumentType targetType)
struct GeoMultiPointStruct GeoMultiPoint
Definition: heavydbTypes.h:981
struct GeoMultiLineStringStruct GeoMultiLineString
static ExtensionFunction.ExtArgumentType deserializeType(final String type_name)
struct GeoMultiPolygonStruct GeoMultiPolygon
struct GeoPolygonStruct GeoPolygon
constexpr auto type_name() noexcept

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

static boolean com.mapd.parser.server.ExtensionFunctionSignatureParser.isWindows ( )
inlinestaticprivate

Definition at line 41 of file ExtensionFunctionSignatureParser.java.

References com.mapd.parser.server.ExtensionFunctionSignatureParser.OS.

Referenced by com.mapd.parser.server.ExtensionFunctionSignatureParser.deserializeType().

41  {
42  return (OS.indexOf("win") >= 0);
43  }

+ Here is the caller graph for this function:

static String com.mapd.parser.server.ExtensionFunctionSignatureParser.join ( final List< String >  strs,
final String  sep 
)
inlinestaticpackage

Definition at line 365 of file ExtensionFunctionSignatureParser.java.

References File_Namespace.append().

Referenced by com.mapd.parser.server.ExtensionFunctionSignatureParser.signaturesToJson().

365  {
366  StringBuilder sb = new StringBuilder();
367  if (strs.isEmpty()) {
368  return "";
369  }
370  sb.append(strs.get(0));
371  for (int i = 1; i < strs.size(); ++i) {
372  sb.append(sep).append(strs.get(i));
373  }
374  return sb.toString();
375  }
size_t append(FILE *f, const size_t size, const int8_t *buf)
Appends the specified number of bytes to the end of the file f from buf.
Definition: File.cpp:158

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

static Map<String, ExtensionFunction> com.mapd.parser.server.ExtensionFunctionSignatureParser.parse ( final String  file_path) throws IOException
inlinestaticpackage

Definition at line 45 of file ExtensionFunctionSignatureParser.java.

References parse_ast.line, setup.name, and com.mapd.parser.server.ExtensionFunctionSignatureParser.toSignature().

45  {
46  File file = new File(file_path);
47  FileReader fileReader = new FileReader(file);
48  BufferedReader bufferedReader = new BufferedReader(fileReader);
49  String line;
50  Pattern s = Pattern.compile("\\| ([\\` ]|used)+ ([\\w]+) '([\\w<>]+) \\((.*)\\)'");
51  Map<String, ExtensionFunction> sigs = new HashMap<String, ExtensionFunction>();
52  while ((line = bufferedReader.readLine()) != null) {
53  Matcher m = s.matcher(line);
54  if (m.find()) {
55  final String name = m.group(2);
56  final String ret = m.group(3);
57  final String cs_param_list = m.group(4);
58  sigs.put(name, toSignature(ret, cs_param_list, false));
59  }
60  }
61  return sigs;
62  }
tuple line
Definition: parse_ast.py:10
static ExtensionFunction toSignature(final String ret, final String cs_param_list, final boolean has_variable_name)
string name
Definition: setup.in.py:72

+ Here is the call graph for this function:

static Map<String, ExtensionFunction> com.mapd.parser.server.ExtensionFunctionSignatureParser.parseFromString ( final String  udf_string) throws IOException
inlinestaticpackage

Definition at line 87 of file ExtensionFunctionSignatureParser.java.

88  {
89  return parseFromString(udf_string, true);
90  }
static Map< String, ExtensionFunction > parseFromString(final String udf_string)
static Map<String, ExtensionFunction> com.mapd.parser.server.ExtensionFunctionSignatureParser.parseFromString ( final String  udf_string,
final boolean  is_row_func 
) throws IOException
inlinestaticpackage

Definition at line 92 of file ExtensionFunctionSignatureParser.java.

References parse_ast.line, setup.name, and com.mapd.parser.server.ExtensionFunctionSignatureParser.toSignature().

93  {
94  StringReader stringReader = new StringReader(udf_string);
95  BufferedReader bufferedReader = new BufferedReader(stringReader);
96  String line;
97  Pattern r = Pattern.compile("([\\w]+)\\s+'([\\w]+)\\s*\\((.*)\\)'");
98  Map<String, ExtensionFunction> sigs = new HashMap<String, ExtensionFunction>();
99  while ((line = bufferedReader.readLine()) != null) {
100  Matcher m = r.matcher(line);
101  if (m.find()) {
102  final String name = m.group(1);
103  final String ret = m.group(2);
104  final String cs_param_list = m.group(3);
105  sigs.put(name, toSignature(ret, cs_param_list, is_row_func));
106  }
107  }
108  return sigs;
109  }
tuple line
Definition: parse_ast.py:10
static ExtensionFunction toSignature(final String ret, final String cs_param_list, final boolean has_variable_name)
string name
Definition: setup.in.py:72

+ Here is the call graph for this function:

static Map<String, ExtensionFunction> com.mapd.parser.server.ExtensionFunctionSignatureParser.parseUdfAst ( final String  file_path) throws IOException
inlinestaticpackage

Definition at line 64 of file ExtensionFunctionSignatureParser.java.

References parse_ast.line, setup.name, and com.mapd.parser.server.ExtensionFunctionSignatureParser.toSignature().

65  {
66  File file = new File(file_path);
67  FileReader fileReader = new FileReader(file);
68  BufferedReader bufferedReader = new BufferedReader(fileReader);
69  String line;
70  Pattern s = Pattern.compile("([<>:\\w]+) ([:\\w]+)(?:\\(\\))?\\((.*)\\)");
71  Map<String, ExtensionFunction> sigs = new HashMap<String, ExtensionFunction>();
72  while ((line = bufferedReader.readLine()) != null) {
73  Matcher m = s.matcher(line);
74  if (m.find()) {
75  final String name = m.group(2);
76  final String ret = m.group(1);
77  final String cs_param_list = m.group(3);
78  if (cs_param_list.isEmpty()) {
79  continue;
80  }
81  sigs.put(name, toSignature(ret, cs_param_list, true));
82  }
83  }
84  return sigs;
85  }
tuple line
Definition: parse_ast.py:10
static ExtensionFunction toSignature(final String ret, final String cs_param_list, final boolean has_variable_name)
string name
Definition: setup.in.py:72

+ Here is the call graph for this function:

static ExtensionFunction.ExtArgumentType com.mapd.parser.server.ExtensionFunctionSignatureParser.pointerType ( final ExtensionFunction.ExtArgumentType  targetType)
inlinestaticprivate

Definition at line 342 of file ExtensionFunctionSignatureParser.java.

References Bool, Double, Float, Int16, Int32, Int64, and Int8.

Referenced by com.mapd.parser.server.ExtensionFunctionSignatureParser.deserializeType().

343  {
344  switch (targetType) {
345  case Bool:
346  return ExtensionFunction.ExtArgumentType.PBool;
347  case Int8:
348  return ExtensionFunction.ExtArgumentType.PInt8;
349  case Int16:
350  return ExtensionFunction.ExtArgumentType.PInt16;
351  case Int32:
352  return ExtensionFunction.ExtArgumentType.PInt32;
353  case Int64:
354  return ExtensionFunction.ExtArgumentType.PInt64;
355  case Float:
356  return ExtensionFunction.ExtArgumentType.PFloat;
357  case Double:
358  return ExtensionFunction.ExtArgumentType.PDouble;
359  default:
360  assert false;
361  return null;
362  }
363  }

+ Here is the caller graph for this function:

static String com.mapd.parser.server.ExtensionFunctionSignatureParser.signaturesToJson ( final Map< String, ExtensionFunction sigs)
inlinestaticpackage

Definition at line 110 of file ExtensionFunctionSignatureParser.java.

References com.mapd.parser.server.ExtensionFunctionSignatureParser.join().

110  {
111  List<String> json_sigs = new ArrayList<String>();
112  if (sigs != null) {
113  for (Map.Entry<String, ExtensionFunction> sig : sigs.entrySet()) {
114  if (sig.getValue().isRowUdf()) {
115  json_sigs.add(sig.getValue().toJson(sig.getKey()));
116  }
117  }
118  }
119  return "[" + join(json_sigs, ",") + "]";
120  }
static String join(final List< String > strs, final String sep)

+ Here is the call graph for this function:

static ExtensionFunction com.mapd.parser.server.ExtensionFunctionSignatureParser.toSignature ( final String  ret,
final String  cs_param_list,
final boolean  has_variable_name 
)
inlinestaticprivate

Definition at line 122 of file ExtensionFunctionSignatureParser.java.

Referenced by com.mapd.parser.server.ExtensionFunctionSignatureParser.parse(), com.mapd.parser.server.ExtensionFunctionSignatureParser.parseFromString(), and com.mapd.parser.server.ExtensionFunctionSignatureParser.parseUdfAst().

123  {
124  return toSignature(ret, cs_param_list, has_variable_name, true);
125  }
static ExtensionFunction toSignature(final String ret, final String cs_param_list, final boolean has_variable_name)

+ Here is the caller graph for this function:

static ExtensionFunction com.mapd.parser.server.ExtensionFunctionSignatureParser.toSignature ( final String  ret,
final String  cs_param_list,
final boolean  has_variable_name,
final boolean  is_row_func 
)
inlinestaticprivate

Definition at line 127 of file ExtensionFunctionSignatureParser.java.

References run_benchmark_import.args, com.mapd.parser.server.ExtensionFunctionSignatureParser.deserializeType(), report.params, split(), and generate_TableFunctionsFactory_init.uses_manager().

130  {
131  String[] params = cs_param_list.split(",");
132  List<ExtensionFunction.ExtArgumentType> args =
133  new ArrayList<ExtensionFunction.ExtArgumentType>();
134  String uses_manager = "false";
135  List<Map<String, String>> annotations = new ArrayList<Map<String, String>>();
136  for (final String param : params) {
137  ExtensionFunction.ExtArgumentType arg_type;
138  if (param.contains("RowFunctionManager")) {
139  uses_manager = "true";
140  continue;
141  }
142  if (has_variable_name) {
143  String[] full_param = param.trim().split("\\s+");
144  if (full_param.length > 0) {
145  if (full_param[0].trim().compareTo("const") == 0) {
146  assert full_param.length > 1;
147  arg_type = deserializeType((full_param[1]).trim());
148  } else {
149  arg_type = deserializeType((full_param[0]).trim());
150  }
151  } else {
152  arg_type = deserializeType(full_param[0]);
153  }
154  } else {
155  arg_type = deserializeType(param.trim());
156  }
157  annotations.add(Collections.EMPTY_MAP);
158  if (arg_type != ExtensionFunction.ExtArgumentType.Void) {
159  args.add(arg_type);
160  }
161  }
162  assert is_row_func;
163  annotations.add(Collections.singletonMap("uses_manager", uses_manager));
164  return new ExtensionFunction(args, deserializeType(ret), annotations);
165  }
std::vector< std::string > split(std::string_view str, std::string_view delim, std::optional< size_t > maxsplit)
split apart a string into a vector of substrings
dictionary params
Definition: report.py:27
static ExtensionFunction.ExtArgumentType deserializeType(final String type_name)

+ Here is the call graph for this function:

Member Data Documentation

final Logger com.mapd.parser.server.ExtensionFunctionSignatureParser.HEAVYDBLOGGER
staticpackage
Initial value:
=
LoggerFactory.getLogger(ExtensionFunctionSignatureParser.class)

Definition at line 37 of file ExtensionFunctionSignatureParser.java.

String com.mapd.parser.server.ExtensionFunctionSignatureParser.OS = System.getProperty("os.name").toLowerCase()
staticprivate

The documentation for this class was generated from the following file: