MySQL的多層SP中Cursor的m_max_cursor_index相關(guān)BUG分析

作者：wuyy 2024-04-03 08:33:31

在MySQL的sp里面使用cursor的話，因?yàn)閙_max_cursor_index?只用于統(tǒng)計(jì)，不用于實(shí)際賦值和計(jì)算過程，因此不影響使用。但是如果要用這個(gè)值用于二次開發(fā)，就要注意到這個(gè)問題。

一、問題發(fā)現(xiàn)

在一次開發(fā)中在sp中使用多層cursor的時(shí)候想知道每層的m_max_cursor_index值分別是多少，以用來做后續(xù)開發(fā)。于是做了以下的試驗(yàn)，但是發(fā)現(xiàn)第一個(gè)level=2那層的m_max_cursor_index的值有點(diǎn)問題。

注：本次使用的MySQL數(shù)據(jù)庫版本為最新的debug版本。

SQL語句示例：

greatsql> CREATE TABLE t1 (a INT, b VARCHAR(10));

以下注釋里面是該層sp_pcontext的參數(shù)值。
DELIMITER $$
CREATE PROCEDURE processnames() -- level=0，m_max_cursor_index=1+8+1
BEGIN
    DECLARE nameCursor0 CURSOR FOR SELECT * FROM t1; -- level=1，m_cursor_offset=0，m_max_cursor_index=1+8+1
    begin
    DECLARE nameCursor1 CURSOR FOR SELECT * FROM t1; -- level=2，m_cursor_offset=1，m_max_cursor_index=1+8 ☆問題點(diǎn)
       begin
   DECLARE nameCursor2 CURSOR FOR SELECT * FROM t1; -- level=3，m_cursor_offset=2，m_max_cursor_index=1
            DECLARE nameCursor3 CURSOR FOR SELECT * FROM t1; -- level=3，m_cursor_offset=2，m_max_cursor_index=2
            DECLARE nameCursor4 CURSOR FOR SELECT * FROM t1; -- level=3，m_cursor_offset=2，m_max_cursor_index=3
            DECLARE nameCursor5 CURSOR FOR SELECT * FROM t1; -- level=3，m_cursor_offset=2，m_max_cursor_index=4
        end;
    end;
    begin
  DECLARE nameCursor6 CURSOR FOR SELECT * FROM t1; -- level=2，m_cursor_offset=1，m_max_cursor_index=1
    end;
END $$
DELIMITER ;

首先查看上面的sp的code，可以發(fā)現(xiàn)nameCursor6和nameCursor1屬于同一層，因此他們的offset值一樣。

greatsql>  show procedure code processnames;
+-----+---------------------------------------+
| Pos | Instruction                           |
+-----+---------------------------------------+
|   0 | cpush nameCursor0@0: SELECT * FROM t1 |
|   1 | cpush nameCursor1@1: SELECT * FROM t1 |
|   2 | cpush nameCursor2@2: SELECT * FROM t1 |
|   3 | cpush nameCursor3@3: SELECT * FROM t1 |
|   4 | cpush nameCursor4@4: SELECT * FROM t1 |
|   5 | cpush nameCursor5@5: SELECT * FROM t1 |
|   6 | cpop 4                                |
|   7 | cpop 1                                |
|   8 | cpush nameCursor6@1: SELECT * FROM t1 |
|   9 | cpop 1                                |
|  10 | cpop 1                                |
+-----+---------------------------------------+
11 rows in set (6.02 sec)

然后通過debug查看每層sp_pcontext的參數(shù)值（相關(guān)參數(shù)值已經(jīng)在上面標(biāo)識(shí)出），發(fā)現(xiàn)第一個(gè)level=2的sp_pcontext的m_max_cursor_index值多了很多，預(yù)期值應(yīng)該是4+1，但是實(shí)際是8+1，而上面的層都沒錯(cuò)，這說明代碼最里面那層m_max_cursor_index賦值錯(cuò)了。

二、問題調(diào)查過程

1、發(fā)現(xiàn)了問題點(diǎn)就看看代碼里面對(duì)于每層的m_max_cursor_index是怎么賦值的。

1、初始化sp_pcontext的時(shí)候所有的參數(shù)都為0
sp_pcontext::sp_pcontext(THD *thd) 
    : m_level(0),
      m_max_var_index(0),
      m_max_cursor_index(0)...{init(0, 0, 0, 0);}

2、每加一層sp_pcontext，當(dāng)前的m_cursor_offset=上一層cursor個(gè)數(shù)
sp_pcontext::sp_pcontext(THD *thd, sp_pcontext *prev,  
                         sp_pcontext::enum_scope scope)
    : m_level(prev->m_level + 1),
      m_max_var_index(0),
      m_max_cursor_index(0)... {init(prev->current_cursor_count());}
void sp_pcontext::init(uint cursor_offset) {m_cursor_offset = cursor_offset;}
uint current_cursor_count() const {
    return m_cursor_offset + static_cast<uint>(m_cursors.size());
}

3、退出當(dāng)前sp_pcontext層，需要把當(dāng)前的max_cursor_index()信息值賦值給上一層的m_max_cursor_index，即當(dāng)前的cursor數(shù)量累加給上一層
sp_pcontext *sp_pcontext::pop_context() {
    uint submax = max_cursor_index();
    if (submax > m_parent->m_max_cursor_index)
      m_parent->m_max_cursor_index = submax;
}
uint max_cursor_index() const {
    return m_max_cursor_index + static_cast<uint>(m_cursors.size());
  }

4、每次增加一個(gè)cursor，m_max_cursor_index值遞增，m_max_cursor_index是計(jì)數(shù)器。
bool sp_pcontext::add_cursor(LEX_STRING name) {
  if (m_cursors.size() == m_max_cursor_index) ++m_max_cursor_index;

  return m_cursors.push_back(name);
}

2、根據(jù)第一步的分析，只在最里面那層的m_max_cursor_index累加出來計(jì)算錯(cuò)誤，看看上面的累加過程，是用max_cursor_index()值來累加的，于是查看max_cursor_index()函數(shù)的實(shí)現(xiàn)：

uint max_cursor_index() const {
    return m_max_cursor_index + static_cast<uint>(m_cursors.size());
  }

這里是把當(dāng)前層的m_max_cursor_index值加上m_cursors.size()，但是在函數(shù)add_cursor里面，m_cursors數(shù)組每增加一個(gè)cursor，m_max_cursor_index都要加1，也就是說在最里面那層sp_pcontext的計(jì)算重復(fù)了，計(jì)算了2遍m_cursors.size()，導(dǎo)致上面的level=2那層的m_max_cursor_index值變成2*4=8了。到這里問題點(diǎn)發(fā)現(xiàn)。

三、問題解決方案

通過以上代碼解析后，可以考慮只對(duì)最里面那層sp_pcontext的max_cursor_index()取值進(jìn)行修改，最里面那層的sp_pcontext沒有m_children，因此可以用這個(gè)數(shù)組值進(jìn)行判斷。代碼作如下修改：

uint max_cursor_index() const {
    if(m_children.size() == 0) -- 最里面那層sp_pcontext直接返回m_max_cursor_index的值。
     return m_max_cursor_index; -- 可以改為static_cast<uint>(m_cursors.size())，二者值一樣。
    else -- 上層sp_pcontext返回下層所有sp_pcontext的m_max_cursor_index的值，再加上當(dāng)前層的m_cursors.size()值。
        return m_max_cursor_index + static_cast<uint>(m_cursors.size());
}

四、問題總結(jié)

在MySQL的sp里面使用cursor的話，因?yàn)閙_max_cursor_index只用于統(tǒng)計(jì)，不用于實(shí)際賦值和計(jì)算過程，因此不影響使用。但是如果要用這個(gè)值用于二次開發(fā)，就要注意到這個(gè)問題。上面的修改方案只是其中一個(gè)解決方案，也可以根據(jù)自己的需要去改add_cursor的m_max_cursor_index的賦值過程。

這次發(fā)現(xiàn)的問題屬于不參與計(jì)算的bug，但卻影響開源代碼的后續(xù)開發(fā)，在實(shí)際開發(fā)應(yīng)用中類似的問題也要注意，一不小心就會(huì)踩坑。

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