PostgreSQL支持函数获取主函数参数设置选择性问题排查

一、参数获取逻辑的正确性确认 #

你通过list_nth_cell(req->args, 4)获取主函数第5个参数的方式是正确的：PostgreSQL的List结构索引从0开始，第5个参数对应索引4，且req->args确实是主函数经过预简化后的参数列表。

但当前取值代码存在关键错误，这是导致rad始终为0.0的核心原因：

• Const结构体的constvalue本身就是Datum类型，不需要用PointerGetDatum()转换，错误的转换会导致获取无效值。
• 未处理参数为NULL的情况，可能引发未定义行为。

二、正确取值与选择性设置的修正方案 #

核心修正点：

1. 直接使用con->constvalue作为Datum值，无需额外转换
2. 增加constisnull检查，处理NULL参数场景
3. 对非常量参数返回默认选择性估算，避免优化器使用不合理的默认值
4. 替换硬编码的3.14为标准数学常量M_PI（需包含<math.h>头文件）

修正后的代码：

#include <math.h>
#include "postgres.h"
#include "fmgr.h"
#include "optimizer/optimizer.h"
#include "nodes/nodeFuncs.h"

PG_FUNCTION_INFO_V1(pgq3c_join_selectivity);
Datum pgq3c_join_selectivity(PG_FUNCTION_ARGS)
{
  Node    *rawreq = (Node *) PG_GETARG_POINTER(0);
  Node    *ret = NULL;
  Datum radDatum;
  double rad;
  double ratio;
  Node *dummy = NULL;
  
  if (IsA(rawreq, SupportRequestSelectivity))
    {
      SupportRequestSelectivity *req = (SupportRequestSelectivity *) rawreq;
      
      // 获取主函数第5个参数（索引4）
      dummy = (Node *) lfirst(list_nth_cell(req->args, 4));

      if (IsA(dummy, Const))
      {
        Const *con = (Const *) dummy;
        
        // 处理NULL参数，返回默认选择性
        if (con->constisnull)
        {
          req->selectivity = 0.5;
          ret = (Node *) req;
          PG_RETURN_POINTER(ret);
        }
        
        // 直接获取常量的Datum值
        radDatum = con->constvalue;
        rad = DatumGetFloat8(radDatum);     
        
        // 使用标准数学常量计算空间选择性
        ratio = M_PI * rad * rad / 41252.;
        // 确保选择性值在[0,1]范围内
        CLAMP_PROBABILITY(ratio);
        
        req->selectivity = ratio;
        ret = (Node *) req;
      }
      else
      {
        // 参数非常量时，返回合理的默认选择性估算
        req->selectivity = 0.1;
        ret = (Node *) req;
      }
    }
  if (IsA(rawreq, SupportRequestSimplify))
  {
    // 返回NULL表示无需简化主函数调用
    PG_RETURN_POINTER(NULL);
  }
  if (IsA(rawreq, SupportRequestCost))
  {
    SupportRequestCost *req = (SupportRequestCost *) rawreq;
    // 根据函数实际开销设置成本估算值
    req->cpu_cost = 1.2;
    req->io_cost = 0.0;
    ret = (Node *) req;
  }
  
  PG_RETURN_POINTER(ret);
}

三、额外注意事项 #

1. 支持函数注册：确保主函数pgq3c_sindist_5在创建时指定支持函数，示例：

CREATE FUNCTION pgq3c_sindist_5(...) RETURNS boolean
AS 'MODULE_PATHNAME'
LANGUAGE C
SUPPORT pgq3c_join_selectivity;

2. 非常量参数处理：如果需要对变量或表达式参数进行更准确的选择性估算，可以使用evaluate_expr()函数在合适的上下文环境中求值，但需注意权限与上下文依赖问题。
3. 编译依赖：使用M_PI时，编译需要链接数学库（如Makefile中添加LIBS += -lm）。

内容的提问来源于stack exchange，提问作者Greg Hennessy