PostgreSQL外部模組載入過程
PostgreSQL如何載入extension,例如 PL/pgSQL、postgres_fdw,檢視它們的symbol:
quanzl-mac:postgresql.builtin_pool quanzl$ nm lib/plpgsql.so U _AllocSetContextCreateExtended U _BeginInternalSubTransaction U _BlessTupleDesc ... 0000000000007a50 T _Pg_magic_func 000000000001c86c s _Pg_magic_func.Pg_magic_data ... 0000000000007e10 T _pg_finfo_plpgsql_call_handler 000000000001c888 s _pg_finfo_plpgsql_call_handler.my_finfo ... 0000000000007e20 T _plpgsql_call_handler ...
T 表示 The symbol is in the text (code) section.
s 表示 The symbol is in an uninitialized data section for small objects.
本文關注模組載入時的動作,涉及上邊的兩個物件:_Pg_magic_func 和 _Pg_magic_func.Pg_magic_data
來看一個PG模組的程式碼(節選自 src/pl/plpgsql/src/pl_handler.c):
... static void plpgsql_extra_errors_assign_hook(const char *newvalue, void *extra); PG_MODULE_MAGIC; /* Custom GUC variable */ static const struct config_enum_entry variable_conflict_options[] = { {"error", PLPGSQL_RESOLVE_ERROR, false}, {"use_variable", PLPGSQL_RESOLVE_VARIABLE, false}, {"use_column", PLPGSQL_RESOLVE_COLUMN, false}, {NULL, 0, false} }; ...
這裡注意 PG_MODULE_MAGIC,它的定義(src/include/fmgr.h):
... /* The actual data block contents */ #define PG_MODULE_MAGIC_DATA \ { \ sizeof(Pg_magic_struct), \ PG_VERSION_NUM / 100, \ FUNC_MAX_ARGS, \ INDEX_MAX_KEYS, \ NAMEDATALEN, \ FLOAT4PASSBYVAL, \ FLOAT8PASSBYVAL \ } ... #define PG_MAGIC_FUNCTION_NAME Pg_magic_func #define PG_MAGIC_FUNCTION_NAME_STRING "Pg_magic_func" #define PG_MODULE_MAGIC \ extern PGDLLEXPORT const Pg_magic_struct *PG_MAGIC_FUNCTION_NAME(void); \ const Pg_magic_struct * \ PG_MAGIC_FUNCTION_NAME(void) \ { \ static const Pg_magic_struct Pg_magic_data = PG_MODULE_MAGIC_DATA; \ return &Pg_magic_data; \ } \ extern int no_such_variable
可以看出,這裡就是 Pg_magic_func 這個函式定義(symbol為什麼多一個下劃線請自行查資料),還有它的靜態變數 Pg_magic_data。
Pg_magic_struct 這個結構體作用是在載入時檢查模組是否與當前資料庫核心匹配,防止不同版本或者不是PG模組的庫的載入。
動態庫的載入函式 internal_load_library 程式碼節選(src/backend/utils/fmgr/dfmgr.c):
magic_func = (PGModuleMagicFunction)
pg_dlsym(file_scanner->handle, PG_MAGIC_FUNCTION_NAME_STRING);
if (magic_func)
{
const Pg_magic_struct *magic_data_ptr = (*magic_func) ();
...
如果找不到 Pg_magic_func,或者是它的返回值與當前資料庫伺服器不一致,拒絕載入。
用於結構體初始化的 PG_MODULE_MAGIC_DATA,包含
1、結構體尺寸
2、版本號
3、函式最大引數個數
4、索引最大列數
5、單精度浮點傳值
6、雙精度浮點傳值
(結束,如有疑問參考上述原始碼檔案 )
後續會寫函式載入的檢查,比如上邊特意列出來的 plpgsql_call_hand