本篇文章為大家展示了PostgreSQL中怎么載入外部模塊，內(nèi)容簡(jiǎn)明扼要并且容易理解，絕對(duì)能使你眼前一亮，通過這篇文章的詳細(xì)介紹希望你能有所收獲。

成都創(chuàng)新互聯(lián)堅(jiān)持“要么做到，要么別承諾”的工作理念，服務(wù)領(lǐng)域包括：成都網(wǎng)站建設(shè)、做網(wǎng)站、企業(yè)官網(wǎng)、英文網(wǎng)站、手機(jī)端網(wǎng)站、網(wǎng)站推廣等服務(wù)，滿足客戶于互聯(lián)網(wǎng)時(shí)代的郾城網(wǎng)站設(shè)計(jì)、移動(dòng)媒體設(shè)計(jì)的需求，幫助企業(yè)找到有效的互聯(lián)網(wǎng)解決方案。努力成為您成熟可靠的網(wǎng)絡(luò)建設(shè)合作伙伴！

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.

本文關(guān)注模塊加載時(shí)的動(dòng)作，涉及上邊的兩個(gè)對(duì)象：_Pg_magic_func 和 _Pg_magic_func.Pg_magic_data

來看一個(gè)PG模塊的代碼（節(jié)選自 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 這個(gè)函數(shù)定義（symbol為什么多一個(gè)下劃線請(qǐng)自行查資料），還有它的靜態(tài)變量 Pg_magic_data。

Pg_magic_struct 這個(gè)結(jié)構(gòu)體作用是在加載時(shí)檢查模塊是否與當(dāng)前數(shù)據(jù)庫(kù)內(nèi)核匹配，防止不同版本或者不是PG模塊的庫(kù)的加載。

動(dòng)態(tài)庫(kù)的加載函數(shù) internal_load_library 代碼節(jié)選（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，或者是它的返回值與當(dāng)前數(shù)據(jù)庫(kù)服務(wù)器不一致，拒絕加載。

用于結(jié)構(gòu)體初始化的 PG_MODULE_MAGIC_DATA，包含

1、結(jié)構(gòu)體尺寸
2、版本號(hào)
3、函數(shù)最大參數(shù)個(gè)數(shù)
4、索引最大列數(shù)
5、單精度浮點(diǎn)傳值
6、雙精度浮點(diǎn)傳值

上述內(nèi)容就是PostgreSQL中怎么載入外部模塊，你們學(xué)到知識(shí)或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識(shí)儲(chǔ)備，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道。