WordPress的用户注册流程有固定的执行步骤，系统预留了多个动作钩子供开发者扩展功能，通过这些钩子可以在用户注册的不同阶段插入自定义PHP代码，实现个性化的业务需求。
WordPress用户注册相关的核心钩子
要实现用户注册时执行自定义代码，首先需要了解WordPress提供的两个核心注册相关钩子：
- user_register：当用户注册表单提交完成，数据成功写入数据库后触发，此时用户已经在系统中存在，是最常用的注册后执行逻辑的钩子。
- register_form：用于扩展注册表单，在注册表单末尾添加自定义的输入字段，适合需要在注册时收集额外用户信息的场景。
在user_register钩子中执行自定义逻辑
user_register钩子会接收一个参数，就是新注册用户的用户ID，我们可以通过这个ID获取用户的所有信息，执行后续操作。
示例1：给新注册用户自动分配自定义角色
假设我们站点有一个名为vip_member的自定义用户角色，希望所有新注册的用户自动获得这个角色，代码实现如下：
// 给新注册用户自动分配vip_member角色
add_action('user_register', 'assign_custom_role_to_new_user');
function assign_custom_role_to_new_user($user_id) {
// 检查用户ID是否有效
if (!$user_id) {
return;
}
// 获取用户对象
$user = new WP_User($user_id);
// 移除默认角色，添加自定义角色
$user->set_role('vip_member');
}
示例2：新用户注册后发送自定义欢迎邮件
默认的WordPress注册邮件内容比较简洁，我们可以自定义邮件内容，添加站点的专属信息：
// 新用户注册后发送自定义欢迎邮件
add_action('user_register', 'send_custom_welcome_email');
function send_custom_welcome_email($user_id) {
if (!$user_id) {
return;
}
// 获取用户信息
$user = get_user_by('id', $user_id);
$user_email = $user->user_email;
$user_login = $user->user_login;
// 邮件主题
$subject = '欢迎加入我们的站点';
// 邮件内容
$message = '尊敬的' . $user_login . '：<br/>';
$message .= '感谢您注册我们的站点，您现在可以登录系统使用所有功能。<br/>';
$message .= '如有任何问题，可以随时联系我们的客服团队。';
// 邮件头设置
$headers = array('Content-Type: text/html; charset=UTF-8');
// 发送邮件
wp_mail($user_email, $subject, $message, $headers);
}
扩展注册表单并收集自定义数据
如果需要在注册时收集额外的用户信息，比如用户的手机号，可以使用register_form钩子添加表单字段，再通过user_register钩子保存数据。
步骤1：添加自定义注册字段
// 在注册表单添加手机号字段
add_action('register_form', 'add_phone_field_to_register_form');
function add_phone_field_to_register_form() {
$phone = !empty($_POST['user_phone']) ? $_POST['user_phone'] : '';
?>
<p>
<label for="user_phone">手机号<br/>
<input type="text" name="user_phone" id="user_phone" class="input" value="<?php echo esc_attr($phone); ?>" size="25" />
</label>
</p>
<?php
}
步骤2：验证并保存自定义字段数据
需要先验证字段的合法性，再在用户注册成功后将数据存储到用户元数据表中：
// 验证手机号字段
add_filter('registration_errors', 'validate_phone_field', 10, 3);
function validate_phone_field($errors, $sanitized_user_login, $user_email) {
if (empty($_POST['user_phone'])) {
$errors->add('phone_error', '请输入您的手机号');
} elseif (!preg_match('/^1[3-9]d{9}$/', $_POST['user_phone'])) {
$errors->add('phone_error', '请输入正确的手机号格式');
}
return $errors;
}
// 保存手机号到用户元数据
add_action('user_register', 'save_phone_field_data');
function save_phone_field_data($user_id) {
if (!empty($_POST['user_phone'])) {
update_user_meta($user_id, 'user_phone', sanitize_text_field($_POST['user_phone']));
}
}
注意事项
- 自定义代码建议放在当前主题的
functions.php文件中，或者做成独立的插件，避免主题更新时代码丢失。 - 执行耗时较长的逻辑时，建议使用
wp_schedule_single_event函数设置为异步执行，避免影响用户注册的响应速度。 - 操作数据库或者发送邮件时，要做好异常处理，避免因为代码错误导致用户注册流程中断。
- 如果站点开启了注册审核功能，
user_register钩子会在用户通过审核后才触发，需要根据实际需求调整逻辑。
注意：修改WordPress核心文件会导致后续更新时自定义代码被覆盖，所有自定义逻辑都不要直接写在WordPress的核心程序文件中。