wordpress如何限制注册邮箱的后缀类型

wordpress建站如何某些邮箱后缀做限制，也就是说不允许部分后缀类型的邮箱注册！具体实现代码如下：

在functions.php中新增以下代码：

//设置邮箱白名单
function is_valid_email_domain($login, $email, $errors ){
  $valid_email_domains = array("gmail.com","qq.com");// 允许注册的邮箱信息
  $valid = false;
  foreach( $valid_email_domains as $d ){
    $d_length = strlen( $d );
    $current_email_domain = strtolower( substr( $email, -($d_length), $d_length));
    if( $current_email_domain == strtolower($d) ){
      $valid = true;
      break;
    }
  }
  
  if( $valid === false ){
    $errors->add('domain_whitelist_error',__( 'ERROR: 本站只支持gmail和QQ邮箱注册。' ));
  }
}
add_action('register_post', 'is_valid_email_domain',10,3 );