关于登录用户显示不同的WordPress菜单

if( is_user_logged_in() ) { 
$args['menu'] = 'logged-in';
} else { 
$args['menu'] = 'logged-out';
} 
return $args;
}
add_filter( 'wp_nav_menu_args', 'wpc_wp_nav_menu_args' );

function wpc_wp_nav_menu_args( $args = '' ) {
if( is_user_logged_in()) { 
if( 'top-navigation' == $args['theme_location'] ) { // Change top-navigation to theme specific name
$args['menu'] = 'logged-in';
}
} else { 
if( 'top-navigation' == $args['theme_location'] ) { // Change top-navigation to theme specific name
$args['menu'] = 'logged-out';
}
}
return $args;
}
add_filter( 'wp_nav_menu_args', 'wpc_wp_nav_menu_args' );