遇到的问题

我在写程序的时候碰到这样一个简单的需求，用户点击“我的XX”这样的功能时候，需要判断当前用户是否已经登录，如果已经登录了，则显示该用户的相关信息并且可以切换到更多界面：

如果没有登录，则显示登录界面并且可以选择登录还是注册：

后来继续了解到其实不止一个功能需要有这样的需求，任何需要登录后才能进行的功能，当用户点击时，都需要做这样的判断。

那我怎样才能把上述两种情况有机地结合在一起，并且能在多个地方复用呢？

当时的第一想法是在代码里根据当前用户登录还是不登录手动替换NavigationController的RootViewController，但是我目前的程序完全是基于Storyboard来写的，因此界面的流转关系是完全体现在Storyboard上的。我想这个功能也能继续维持这样的状态，能在Storyboard上清晰地体现出来（这里不讨论Storyboard的优劣，只是想说明工程代码的表现形式要一致）。

由于年少不懂事，当时以为只要做出下图的这种关系就可以了，当然很快就发现这是不可能的

后来发现其实 NavigationController内部也是实现了类似于ContainerViewController的机制 （可能需FQ），所以只需要替换ContainerViewController的内容，自然也就替换了相应的界面。

多分支NavigationController的方法

根据上述的讨论，具体的方案已经呼之欲出了（关于具体的自定义ContainerViewController的技术细节就不再阐述了，上面的链接已经解释的很清楚了，这里主要是讨论如何使之应用到我们的场景中）。

我是先自己在Storyboard上建立一个RootViewController，名字暂且定为ConditionContainerViewController（具体代码后面会介绍到），如下图：

然后再继承UIStoryboardSegue，新建一个Segue，名字暂且定为ConditionShowSegue：

1 2 3 4 5 6 7 8 9

- (void)perform {    UIViewController *srcVC = self.sourceViewController;    UIViewController *destVC = self.destinationViewController;      [srcVC addChildViewController:destVC];    [srcVC.view addSubview:destVC.view];    destVC.view.frame = CGRectMake(0, 0, CGRectGetWidth(srcVC.view.frame), CGRectGetHeight(srcVC.view.frame));    [destVC didMoveToParentViewController:srcVC];    }

如果上面代码暂时不能理解，请回到之前发的链接再仔细理解里面的内容（我当时也是看了好几遍，呵呵）。

随后就可以在Storyboard中使用这个Segue了，可以看到菜单里多了一种“Condition Show”：

用新的Segue来连接之前在本文一开始就展示的两段分开的业务线，使之结合在一起

然后分别定义这两个Segue的Identifier为“ShowNeedLogin”以及“ShowLoggedIn”。

接下来我们再回过头来看ConditionContainerViewController的代码，其实主要代码非常简单就是重载viewWillAppear:这个方法

1 2 3 4 5 6 7 8 9 10

- (void)viewWillAppear:(BOOL)animated {      [ super  viewWillAppear:animated];        if  (isLogin) {        [self performSegueWithIdentifier:@ "ShowHasLogined"  sender:self];      }      else  {        [self performSegueWithIdentifier:@ "ShowLoggedIn"  sender:self];      }    }

isLogin这个你可以使用各种方法来实现，比如自定义一个变量或者全局有一个用户管理类来暴露一个isLogin属性等。

其实目前程序已经可以根据你当前登录的情况来自动切换展示的界面了，但是现在还有一个问题，登录成功之后又怎么跳回已经登录的界面的？只要在需要登录成功后调用如下代码：

1 2 3 4 5 6 7 8

// 普通情况下直接调用popToRootViewControllerAnimated即可    // 然后ConditionContainerViewController会通过viewWillAppear来判断    NSArray *poppedViewcontrollers = [self popToRootViewControllerAnimated:animated];    // 但是如果是从上面那个图的Need Login这个界面返回，这个时候已经在RootViewController了    // 因此需要手动调用viewWillAppear    if  (poppedViewcontrollers == nil) {       [[self.viewControllers firstObject] viewWillAppear:YES];    }

好了，接下来我们还需要一些清理代码，在ConditionContainerViewController写入以下函数：

1 2 3 4 5 6 7 8 9 10 11

@property (nonatomic, strong) UIViewController *lastViewController;    ...        - (void)prepareForSegue:(UIStoryboardSegue *)segue sender:(id)sender {      if  (segue.destinationViewController != self.lastViewController) {        [self.lastViewController willMoveToParentViewController:nil];        [self.lastViewController.view removeFromSuperview];        [self.lastViewController removeFromParentViewController];      }      self.lastViewController = segue.destinationViewController;    }

这个清理代码应该有更好地方来写，目前我只能想到放在这里，希望大家可以给一些建议。

当然，一开始我也提到这种方法是要可以复用的，且在Storyboard上能清晰地表示出来，最后我就展示一张简单的Storyboard来说明：