Wordpress在全局变量中设置自定义值

Question

如何在Wordpress中使用自定义值设置全局变量。我想用'wp_login'钩子登录时设置用户的branchId。

但是我无法让它发挥作用。这是代码：

function sp_setBranch() {
    // Set branch ID
    global $sp_branch, $wpdb, $current_user;
    if ($current_user->roles[0] == 'trainer'){
        $trainer_table=$wpdb->prefix.'sp_trainer';
        $branch=$wpdb->get_row("select branch_id from $trainer_table where wp_usr_id=$current_user->ID");
        $sp_branch = $branch->branch_id;
    } else {
         $sp_branch=58;
    }
}
add_action('wp_login','sp_setBranch');

后来我试图获得$sp_branch，但我没有在任何地方获得价值。我选择全球化，因为我在任何地方都需要这个。