7b2主题vip跟普通用户怎么显示不同菜单！

function mrxu_vip_menu_args( $args = '' ) {
$user_id = get_current_user_id();
$vip = get_user_meta($user_id,'zrz_vip',true);
$array=array("vip0","vip1","vip2","vip3","vip4");
$vips=in_array($vip,$array);
if ($args['theme_location'] == 'post') {
if ($vips) {
$args['menu'] = '会员';
}
else {
$args['menu'] = '普通';
}
}
return $args;
}

add_filter( 'wp_nav_menu_args', 'mrxu_vip_menu_args' );