相关 《Postgresql源码(78)plpgsql中调用call proc()时的参数传递和赋值(pl参数)》 《Postgresql源码(79)plpgsql中多层调用时参数传递关键点分析(pl参数)》 《Postgresql源码(84)语义分析——函数调用结构CallStmt的构造与函数多态的实现(pl参数)》
本文涉及模块:语法分析 、语义分析、查询重写
函数调用时在语义分析阶段,transform函数对函数入参进行分析,直观上需要完成几步工作:
PG对于上述问题都有了完善的处理逻辑,本篇尝试分析该过程的处理细节。
总入口:transformCallStmt
【1】transformCallStmt
{type=T_CallStmt,FuncCall,FuncExpr,List outargs}
)CallStmt->FuncExpr
、生成CallStmt->FuncExpr->args
(不包含指向参数和默认参数)CallStmt->FuncExpr->args
,加入指向参数和默认参数。【2】CallStmt是如何使用的
(《Postgresql源码(79)plpgsql中多层调用时参数传递关键点分析(pl参数)》问题四:内层ExecuteCallStmt如何构造fcinfo->args?)
CallStmt->FuncExpr->args
,把其中的值直接填入fcinfo->args[i].value
使用。CallStmt->FuncExpr->args
。【3】对比Oracle
Oracle的IN OUT类型不能有默认参数,PG可以。
Oracle的OUT参数必须给个变量,否则执行肯定报错。
【4】PG目前的多态逻辑总结
CREATE or replace PROCEDURE tp13(
a in integer,
b out integer,
c out integer,
d inout integer default 400,
e in integer default 500)
LANGUAGE plpgsql
AS $$
BEGIN
raise notice 'a: %', a;
raise notice 'b: %', b;
raise notice 'c: %', c;
raise notice 'd: %', d;
raise notice 'e: %', e;
END;
$$;
call tp13 (1,2,3,4,5);
call tp13 (1,2,3,e=>5);
transformCallStmt函数负责转换所有函数调用节点,例如:
call proc1();
select func1();
transformCallStmt函数负责生成CallStmt
结构:
typedef struct CallStmt
{
NodeTag type;
FuncCall *funccall; /* from the parser */
FuncExpr *funcexpr; /* transformed call, with only input args */
List *outargs; /* transformed output-argument expressions */
} CallStmt;
CallStmt结构在之前的函数参数分析文章中反复提到过:
截取一部分:Postgresql源码(79)plpgsql中多层调用时参数传递关键点分析(pl参数)
transformCallStmt内部有两个关键调用负责生成CallStmt->FuncExpr
结构:
ParseFuncOrColumn
func_get_detail // 从系统表中找到函数,多态实现在这里
FuncnameGetCandidates // 第一步:找候选者
【1】用名字匹配遍历每一个结果
【2】对于某个结果,拿到PG_PROC参数类型列表proallargtypes
【3】对于某个结果,检查参数数目够不够?
【3.1】对于全指向参数或混合型参数输入
如果 (proallargtypes个数) > (传入的全部参数个数):参数不够,需要补默认
如果 (传入的全部参数个数+默认参数个数) < (proallargtypes个数):补上默认参数就够用了!
如果 (传入的全部参数个数+默认参数个数) >=(proallargtypes个数):补上默认参数也不够,不使用当前函数。
如果 (proallargtypes个数) <= (传入的全部参数个数):参数够用
MatchNamedCall判断指向参数列表是否能匹配当前函数
例如:call tp13 (1,2,3,e=>5);
tp13(a in integer, b out integer,c out integer,d inout integer default 400,e in integer default 500)
MatchNamedCall返回argnumbers数组表示映射关系:
argnumbers = [0,1,2,4,3]
给的第一个参数对应当前函数的参数列表中的0位置:a
给的第二个参数对应当前函数的参数列表中的1位置:b
给的第三个参数对应当前函数的参数列表中的2位置:c
给的第四个参数对应当前函数的参数列表中的4位置:e
只给了4个参数进来,第五个位置补充一个需要默认参数的3位置:d
【3.2】对于全非指向参数输入
只需要判断参数个数就好了,和上面逻辑类似不在赘述
func_get_detail
【找到严格匹配候选者】遍历FuncnameGetCandidates返回结果,如果能和argtypes严格匹配,即找到best_candidate
PGPROC中拉出默认参数列表,删除掉没用的,结果放到*argdefaults返回
【没有严格匹配候选者】遍历FuncnameGetCandidates返回结果,没有候选者能和argtypes严格匹配
首先判断这是不是一个强制转换:例如 select int(3.1),如果是的可以当做强制转换返回
如果不是强制转换,这里肯定是参数类型对不上了,这里就开始进行【多态判断】
判断入参类型能不能通过转换 变成 候选者的参数类型:func_match_argtypes
如果只有一个候选者可以匹配, best_candidate = 当前候选者
如果有多个候选者经过转换可以匹配,选择一个:func_select_candidate
expand_function_arguments的逻辑就很简单了,只是把参数解析后拼接到FuncExpr->args中
(其实这件事情上面的函数已经做过了,但是只是用于参数类型匹配检测,并没有真正拼接到FuncExpr->args)
expand_function_arguments
...
/* If so, we must apply reorder_function_arguments */
if (has_named_args)
{
args = reorder_function_arguments(args, pronargs, func_tuple);
/* Recheck argument types and add casts if needed */
recheck_cast_function_args(args, result_type,
proargtypes, pronargs,
func_tuple);
}
else if (list_length(args) < pronargs)
{
/* No named args, but we seem to be short some defaults */
args = add_function_defaults(args, pronargs, func_tuple);
/* Recheck argument types and add casts if needed */
recheck_cast_function_args(args, result_type,
proargtypes, pronargs,
func_tuple);
}