WordPress/PHP中如何为会员徽章函数传入size参数

我在WordPress中编写了get_membership_badge()函数，用于根据用户会员类型渲染对应徽章，原代码如下：

function get_membership_badge() {
  $user_id = bbp_get_user_id();
  $membership_level = rcp_get_subscription($user_id);

  if ($membership_level === 'Community') {
    echo '';
  }

  elseif ($membership_level === 'Supporter') {
    echo '<span class="member-badge supporter small">Supporter</span>';
  }

  elseif ($membership_level === 'Professional') {
    echo '<span class="member-badge pro small">Pro</span>';
  }
}

会员徽章的尺寸由CSS类控制，默认尺寸无需额外添加类。我希望在模板中调用函数时传入size参数，例如<?php get_membership_badge('small'); ?>，但尝试给函数添加$size参数后页面加载失败，修改后的代码如下：

function get_membership_badge($size) {
  $user_id = bbp_get_user_id();
  $membership_level = rcp_get_subscription($user_id);
  

  if ($membership_level === 'Community') {
    echo '';
  }

  elseif ($membership_level === 'Supporter') {
    echo '<span class="member-badge supporter ' . $size .'">Supporter</span>';
  }

  elseif ($membership_level === 'Professional') {
    echo '<span class="member-badge pro ' . $size .'">Pro</span>';
  }
}

问题原因与修复方案 #

页面加载失败的核心原因是未给参数设置默认值——如果调用函数时不传参数（比如之前的无参写法），PHP会因缺少必填参数抛出错误。同时可以优化几个细节提升函数健壮性：

1. 给$size设置默认空值，兼容无参数调用场景
2. 处理空尺寸情况，避免输出多余空格
3. 改用return替代echo，让函数更灵活（模板中需用echo输出）
4. 用esc_attr()转义参数，避免XSS风险

修复后的完整代码：

function get_membership_badge($size = '') {
  $user_id = bbp_get_user_id();
  $membership_level = rcp_get_subscription($user_id);
  
  // 处理尺寸类，空值时不输出额外内容
  $size_class = !empty($size) ? ' ' . esc_attr($size) : '';

  if ($membership_level === 'Community') {
    return '';
  } elseif ($membership_level === 'Supporter') {
    return '<span class="member-badge supporter' . $size_class . '">Supporter</span>';
  } elseif ($membership_level === 'Professional') {
    return '<span class="member-badge pro' . $size_class . '">Pro</span>';
  }
  
  // 处理未知会员类型的默认返回
  return '';
}

使用方式 #

• 默认尺寸（不传参数）：<?php echo get_membership_badge(); ?>
• 指定尺寸：<?php echo get_membership_badge('small'); ?> 或 <?php echo get_membership_badge('large'); ?>

内容的提问来源于stack exchange，提问作者cannydare