132 lines
4.3 KiB
Python
132 lines
4.3 KiB
Python
"""
|
|
This module is palo builtin functions
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
from string import Template
|
|
import palo_builtins_functions
|
|
|
|
java_registry_preamble = '\
|
|
// Modifications copyright (C) 2017, Baidu.com, Inc. \n\
|
|
// Copyright 2017 The Apache Software Foundation \n\
|
|
// \n\
|
|
// Licensed to the Apache Software Foundation (ASF) under one \n\
|
|
// or more contributor license agreements. See the NOTICE file \n\
|
|
// distributed with this work for additional information \n\
|
|
// regarding copyright ownership. The ASF licenses this file \n\
|
|
// to you under the Apache License, Version 2.0 (the \n\
|
|
// "License"); you may not use this file except in compliance \n\
|
|
// with the License. You may obtain a copy of the License at \n\
|
|
// \n\
|
|
// http://www.apache.org/licenses/LICENSE-2.0\n\
|
|
// \n\
|
|
// Unless required by applicable law or agreed to in writing, software\n\
|
|
// distributed under the License is distributed on an "AS IS" BASIS,\n\
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n\
|
|
// See the License for the specific language governing permissions and\n\
|
|
// limitations under the License.\n\
|
|
\n\
|
|
// This is a generated file, DO NOT EDIT.\n\
|
|
// To add new functions, see the generator at\n\
|
|
// common/function-registry/gen_builtins_catalog.py or the function list at\n\
|
|
// common/function-registry/palo_builtins_functions.py.\n\
|
|
\n\
|
|
package com.baidu.palo.builtins;\n\
|
|
\n\
|
|
import com.baidu.palo.catalog.PrimitiveType;\n\
|
|
import com.baidu.palo.catalog.FunctionSet;\n\
|
|
\n\
|
|
public class ScalarBuiltins { \n\
|
|
public static void initBuiltins(FunctionSet functionSet) { \
|
|
\n'
|
|
|
|
java_registry_epilogue = '\
|
|
}\n\
|
|
}\n'
|
|
|
|
FE_PATH = "../java/com/baidu/palo/builtins/"
|
|
|
|
# This contains all the metadata to describe all the builtins.
|
|
# Each meta data entry is itself a map to store all the meta data
|
|
# - fn_name, ret_type, args, symbol, sql_names
|
|
meta_data_entries = []
|
|
|
|
# Read in the function and add it to the meta_data_entries map
|
|
def add_function(fn_meta_data, user_visible):
|
|
"""add function
|
|
"""
|
|
assert 4 <= len(fn_meta_data) <= 6, \
|
|
"Invalid function entry in palo_builtins_functions.py:\n\t" + repr(fn_meta_data)
|
|
entry = {}
|
|
entry["sql_names"] = fn_meta_data[0]
|
|
entry["ret_type"] = fn_meta_data[1]
|
|
entry["args"] = fn_meta_data[2]
|
|
entry["symbol"] = fn_meta_data[3]
|
|
if len(fn_meta_data) >= 5:
|
|
entry["prepare"] = fn_meta_data[4]
|
|
if len(fn_meta_data) >= 6:
|
|
entry["close"] = fn_meta_data[5]
|
|
entry["user_visible"] = user_visible
|
|
meta_data_entries.append(entry)
|
|
|
|
|
|
def generate_fe_entry(entry, name):
|
|
"""add function
|
|
"""
|
|
java_output = ""
|
|
java_output += "\"" + name + "\""
|
|
java_output += ", \"" + entry["symbol"] + "\""
|
|
if entry["user_visible"]:
|
|
java_output += ", true"
|
|
else:
|
|
java_output += ", false"
|
|
|
|
if 'prepare' in entry:
|
|
java_output += ', "%s"' % entry["prepare"]
|
|
if 'close' in entry:
|
|
java_output += ', "%s"' % entry["close"]
|
|
else:
|
|
java_output += ', null'
|
|
|
|
# Check the last entry for varargs indicator.
|
|
if entry["args"] and entry["args"][-1] == "...":
|
|
entry["args"].pop()
|
|
java_output += ", true"
|
|
else:
|
|
java_output += ", false"
|
|
|
|
java_output += ", PrimitiveType." + entry["ret_type"]
|
|
for arg in entry["args"]:
|
|
java_output += ", PrimitiveType." + arg
|
|
return java_output
|
|
|
|
# Generates the FE builtins init file that registers all the builtins.
|
|
def generate_fe_registry_init(filename):
|
|
"""add function
|
|
"""
|
|
java_registry_file = open(filename, "w")
|
|
java_registry_file.write(java_registry_preamble)
|
|
|
|
for entry in meta_data_entries:
|
|
for name in entry["sql_names"]:
|
|
java_output = generate_fe_entry(entry, name)
|
|
java_registry_file.write(" functionSet.addScalarBuiltin(%s);\n" % java_output)
|
|
|
|
java_registry_file.write("\n")
|
|
java_registry_file.write(java_registry_epilogue)
|
|
java_registry_file.close()
|
|
|
|
# Read the function metadata inputs
|
|
for function in palo_builtins_functions.visible_functions:
|
|
add_function(function, True)
|
|
for function in palo_builtins_functions.invisible_functions:
|
|
add_function(function, False)
|
|
|
|
if not os.path.exists(FE_PATH):
|
|
os.makedirs(FE_PATH)
|
|
|
|
generate_fe_registry_init(FE_PATH + "ScalarBuiltins.java")
|
|
|
|
|