WordPress 使用鉤子進(jìn)行主題開發(fā)時避免死循環(huán)

LeexMuller 發(fā)布于2019-07-01 12:49 / 2343人閱讀

摘要：開發(fā)時，我們經(jīng)常會使用或等鉤子，以便在更新文章或分類時，增加一些附加數(shù)據(jù)，這些鉤子觸發(fā)的時機(jī)是發(fā)布或更新數(shù)據(jù)的時間，如果處理不好，就會出現(xiàn)死循環(huán)的情況，導(dǎo)致更新數(shù)據(jù)時出現(xiàn)死循環(huán)的情況。我們可以做的是在開發(fā)過程中盡量小心，避免死循環(huán)的出現(xiàn)。

WordPress 開發(fā)時，我們經(jīng)常會使用 save_post 或 create_post_tag 等鉤子，以便在更新文章或分類時，增加一些附加數(shù)據(jù)，這些鉤子觸發(fā)的時機(jī)是發(fā)布或更新數(shù)據(jù)的時間，如果處理不好，就會出現(xiàn)死循環(huán)的情況，導(dǎo)致更新數(shù)據(jù)時出現(xiàn)死循環(huán)的情況。

舉個例子，我們需要在保存文章的時候，掛在一個函數(shù)到 save_post 鉤子上，這個函數(shù)的目的是修改文章標(biāo)題，在文章標(biāo)題前面加上作者名稱，我們通過調(diào)用wp_update_post 函數(shù)來顯示這個修改。示例代碼如下。

add_action("save_post", "wprs_update_post");
function wprs_update_post($post_id)
{
    $title = get_the_title($post_id);
    $title = "一刀：" . $title;

    $args = [
        "ID"         => $post_id,
        "post_title" => $title,
    ];
    
    wp_update_post($args);
}

上面的代碼中，「save_post」這個鉤子發(fā)生在 wp_update_post 函數(shù)的調(diào)用過程中，通過「save_post」鉤子，我們又調(diào)用了 wp_update_post 這個函數(shù)，這個函數(shù)中，又運行了「save_post」鉤子。于是，程序就這么不知疲倦的運行了下去。

有經(jīng)驗的程序員可能很容易就發(fā)現(xiàn)了這個問題，可是電腦并不知道這是個死循環(huán)，我們不能去責(zé)怪電腦，畢竟那只是個沒有思想的機(jī)器。我們可以做的是在開發(fā)過程中盡量小心，避免死循環(huán)的出現(xiàn)。

那么，怎么解決 WordPress 中出現(xiàn)的這個死循環(huán)呢？其實方法很簡單。

怎么避免 WordPress 中的死循環(huán)

還是上面的代碼，我們稍作修改，就可以完美的避免這個死循環(huán)了。下面是修改后的代碼，對比上面出現(xiàn)死循環(huán)的代碼，我們只添加了兩行。

add_action("save_post", "wprs_update_post");
function wprs_update_post($post_id)
{
   // 獲取文章標(biāo)題并在文章標(biāo)題前面加上作者
    $title = get_the_title($post_id);
    $title = "一刀：" . $title;

    $args = [
        "ID"         => $post_id,
        "post_title" => $title,
    ];
    
    // 更新文章
    remove_action("save_post", "wprs_update_post");
    wp_update_post($args);
    add_action("save_post", "wprs_update_post");
}

原理很簡單，在運行「wp_update_post」函數(shù)之前，我們先卸載掉掛在掉「save_post」鉤子上導(dǎo)致死循環(huán)的函數(shù)，這樣就不會出現(xiàn)重復(fù)執(zhí)行「wprs_update_post」函數(shù)了，更新完文章后，我們再把這個鉤子掛載上，以便后續(xù)使用。

如果您在 WordPress開發(fā)過程中遇到了死循環(huán)問題，不妨參考上面的代碼進(jìn)行修改，這樣既能實現(xiàn)我們需要的功能，又不會導(dǎo)致死循環(huán)的問題。