WordPress 使用钩子进行主题开发时避免死循环

2019年9月11日15:49:22WordPress 开发来源：wpzhiku评论99阅读模式

WordPress 开发时，我们经常会使用 save_post 或 create_post_tag 等钩子，以便在更新文章或分类时，增加一些附加数据，这些钩子触发的时机是发布或更新数据的时间，如果处理不好，就会出现死循环的情况，导致更新数据时出现死循环的情况。

举个例子，我们需要在保存文章的时候，挂在一个函数到 save_post 钩子上，这个函数的目的是修改文章标题，在文章标题前面加上作者名称，我们通过调用wp_update_post 函数来显示这个修改。示例代码如下。

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」这个钩子发生在 wp_update_post 函数的调用过程中，通过「save_post」钩子，我们又调用了 wp_update_post 这个函数，这个函数中，又运行了「save_post」钩子。于是，程序就这么不知疲倦的运行了下去。

有经验的程序员可能很容易就发现了这个问题，可是电脑并不知道这是个死循环，我们不能去责怪电脑，毕竟那只是个没有思想的机器。我们可以做的是在开发过程中尽量小心，避免死循环的出现。

那么，怎么解决 WordPress 中出现的这个死循环呢？其实方法很简单。

怎么避免 WordPress 中的死循环

还是上面的代码，我们稍作修改，就可以完美的避免这个死循环了。下面是修改后的代码，对比上面出现死循环的代码，我们只添加了两行。

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,
    ];
    
    // 更新文章
    remove_action('save_post', 'wprs_update_post');
    wp_update_post($args);
    add_action('save_post', 'wprs_update_post');
}