從Join的實(shí)現(xiàn)窺探MySQL迭代器

作者：亮 2022-09-14 08:01:36

sql_select.cc?內(nèi)函數(shù)bool Sql_cmd_dml::execute(THD *thd)?函數(shù)內(nèi)主要操作為函數(shù)execute_inner?，在函數(shù)execute_inner內(nèi)首先會(huì)對(duì)當(dāng)前的執(zhí)行優(yōu)化操作。

以如下left join查詢(xún)語(yǔ)句為范例：

select * from t1 left join t2 on t1.c=t2.a ;

以下初始化數(shù)據(jù)：

1  DROP TABLE IF EXISTS `t1`;
2    CREATE TABLE `t1` (
3      `a` int DEFAULT NULL,
4      `b` varchar(20) DEFAULT NULL
5    )
6    INSERT INTO `t1` VALUES (1, 'a');
7    INSERT INTO `t1` VALUES (1, 'b');
8    INSERT INTO `t1` VALUES (4, 'a');
9    INSERT INTO `t1` VALUES (5, 'a');
10    
11    DROP TABLE IF EXISTS `t2`;
12    CREATE TABLE `t2` (
13      `c` int DEFAULT NULL,
14      `d` varchar(20) DEFAULT NULL
15    )
16    INSERT INTO `t2` VALUES (9, 'i');
17    INSERT INTO `t2` VALUES (1, 'i');
18    INSERT INTO `t2` VALUES (2, 'i');
19    INSERT INTO `t2` VALUES (3, 'i');

1.處理join的yacc入口

在sys_yacc.yy?文件內(nèi)解析t1 left join t2 on t1.c=t2.a;對(duì)應(yīng)處理位置。

1  table_reference outer_join_type table_reference ON_SYM expr
2    {
3  $$= NEW_PTN PT_joined_table_on($1, @2, $2, $3, $5);
4    }

其中outer_join_type對(duì)應(yīng)。

1  outer_join_type:
2    LEFT opt_outer JOIN_SYM          { $$= JTT_LEFT; }
3  | RIGHT opt_outer JOIN_SYM         { $$= JTT_RIGHT; }

入?yún)⑻幚碓诤瘮?shù)T_joined_table_on內(nèi)。

2.移步到函數(shù)PT_joined_table_on

從PT_joined_table_on?聲明可知其繼承PT_joined_table?函數(shù)，入?yún)⒆笥冶碣x值為PT_joined_table內(nèi)定義的tr1和tr2。

函數(shù)PT_joined_table_on?將輸入join的左右表加入context內(nèi)，并調(diào)用add_join_on將on內(nèi)的條件加入右表，記錄后續(xù)數(shù)據(jù)過(guò)濾條件。

3.執(zhí)行階段函數(shù)do_command(thd)

具體對(duì)應(yīng)執(zhí)行函數(shù)int mysql_execute_command(THD *thd, bool first_level)?，語(yǔ)句解析以及相應(yīng)參數(shù)保存完成后，進(jìn)入函數(shù)int mysql_execute_command(THD *thd, bool first_level)?，此函數(shù)內(nèi)根據(jù)前面解析到的命令類(lèi)型switch (lex->sql_command)?調(diào)用對(duì)應(yīng)的處理函數(shù)，如當(dāng)前語(yǔ)句為例查詢(xún)命令解析為lex->sql_command = SQLCOM_SELECT?則進(jìn)入函數(shù)lex->m_sql_cmd->execute(thd)?;其對(duì)應(yīng)為sql_select.cc?內(nèi)函數(shù)bool Sql_cmd_dml::execute(THD *thd)。

4.優(yōu)化器操作，生成access_paths

調(diào)用查詢(xún)表達(dá)式Query_expression?的優(yōu)化器unit->optimize?，此函數(shù)中會(huì)對(duì)該Query_expression?的內(nèi)的每個(gè)查詢(xún)塊query_block分別先進(jìn)行優(yōu)化操作。

查詢(xún)塊內(nèi)函數(shù)bool JOIN::optimize()?內(nèi)會(huì)將每個(gè)查詢(xún)塊優(yōu)化生成查詢(xún)執(zhí)行計(jì)劃，具體執(zhí)行函數(shù)為函數(shù)JOIN::create_access_paths()內(nèi)create_root_access_path_for_join()?函數(shù)，以當(dāng)前查詢(xún)?yōu)槔诤瘮?shù)create_root_access_path_for_join?內(nèi)根據(jù)參數(shù)條件主要調(diào)用ConnectJoins函數(shù)。

在函數(shù)ConnectJoins?內(nèi)調(diào)用FindSubstructure判斷是join類(lèi)型內(nèi)連接、外連接、半鏈接等類(lèi)型。

根據(jù)FindSubstructure?返回join類(lèi)型調(diào)用相應(yīng)的函數(shù)生成path，當(dāng)前查詢(xún)?yōu)槔龍?zhí)行調(diào)用CreateHashJoinAccessPath生成path。

至此查詢(xún)塊query_block?的優(yōu)化操作和path生成完成，查詢(xún)塊優(yōu)化操作完成后再執(zhí)行整體表達(dá)式Query_expression的優(yōu)化和path的生成，因?yàn)槟壳胺独齼H為一個(gè)查詢(xún)塊，所以當(dāng)前無(wú)需再做整體表達(dá)式的優(yōu)化和path生成。

5.創(chuàng)建迭代器iterator

根據(jù)上一步生成的path調(diào)用CreateIteratorFromAccessPath函數(shù)生成迭代器，用于循環(huán)操作各表數(shù)據(jù)。

在此函數(shù)內(nèi)會(huì)根據(jù)path的類(lèi)型調(diào)用生成不同類(lèi)型的迭代器，以目前范例為例，會(huì)調(diào)用迭代器類(lèi)型為HashJoinIterator。

6.上述4、5步執(zhí)行完成后，執(zhí)行迭代器iterator

在函數(shù)execute_inner?內(nèi)執(zhí)行完成上述4、5步驟操作后主要繼續(xù)執(zhí)行unit->execute(thd)?函數(shù)，其對(duì)應(yīng)執(zhí)行查詢(xún)表達(dá)式函數(shù)bool Query_expression::ExecuteIteratorQuery(THD *thd)。

函數(shù)Query_expression::ExecuteIteratorQuery?內(nèi)主要執(zhí)行m_root_iterator->Init()?，迭代器iterator初始化，當(dāng)前范例為使用HashJoinIterator?類(lèi)型迭代器，因此對(duì)應(yīng)執(zhí)行迭代器函數(shù)HashJoinIterator::Init()。

執(zhí)行m_build_input->Init()?來(lái)初始右表table句柄,用于下面函數(shù)BuildHashTable()?內(nèi)讀取右表數(shù)據(jù)以便初始化返回?cái)?shù)據(jù)存儲(chǔ)表hashtable?，值得注意的是BuildHashTable?函數(shù)內(nèi)會(huì)根據(jù)處理流程調(diào)用SetReadingProbeRowState設(shè)置執(zhí)行狀態(tài)用于引導(dǎo)后續(xù)迭代器iterator執(zhí)行流程。

函數(shù)內(nèi)最后調(diào)用InitProbeIterator?執(zhí)行m_probe_input->Init()初始左表table句柄用于下面函數(shù)讀取左表數(shù)據(jù)。

上面操作完成后執(zhí)行m_root_iterator->Read()?函數(shù)，以當(dāng)前查詢(xún)?yōu)榉独鋵?duì)應(yīng)int HashJoinIterator::Read()?函數(shù)，執(zhí)行過(guò)程中根據(jù)前面SetReadingProbeRowState?設(shè)置的流程狀態(tài)再選擇對(duì)應(yīng)的操作函數(shù)，以當(dāng)前范例則會(huì)循環(huán)讀取左表數(shù)據(jù)，而在操作函數(shù)內(nèi)也會(huì)調(diào)用SetReadingProbeRowState?來(lái)設(shè)置迭代器iterator下一步操作，直至迭代器處理完成，其中在函數(shù)Query_expression::ExecuteIteratorQuery，每次讀取一條成功后就會(huì)調(diào)用send_data操作將結(jié)果發(fā)送至客戶(hù)端，直至所有查詢(xún)結(jié)果發(fā)送完成。

7.至此客戶(hù)端收到相應(yīng)顯示查詢(xún)結(jié)果。

責(zé)任編輯：武曉燕來(lái)源： GreatSQL社區(qū)