重构 Switch 语句答案

【问题标题】：Refactoring Switch-Statement重构 Switch 语句
【发布时间】：2016-06-05 00:05:17
【问题描述】：

有没有办法重构这个函数，让我尽可能少地重复我自己？

我有四个 ViewController，每个都包含一个变量 eventData，这是一个自定义对象。

我想过将eventData 设为全局变量。这样做有什么缺点吗？

这是我现在使用的代码：

override func prepareForSegue(segue: UIStoryboardSegue, sender: AnyObject?)
{
    navigationItem.title = nil

    let segues = UIStoryboardSegue.Segue.self

    switch segue.identifier! {

    case segues.EventOverviewToGuestList.rawValue:

        let destinationVC = segue.destinationViewController as! GuestListVC

        destinationVC.eventData = eventData


    case segues.EventOverviewToAnalytics.rawValue:

        let destinationVC = segue.destinationViewController as! AnalyticsVC

        destinationVC.eventData = eventData


    case segues.EventOverviewToMessaging.rawValue:

        let destinationVC = segue.destinationViewController as! MessagingVC

        destinationVC.eventData = eventData


    case segues.EventOverviewToInformation.rawValue:

        let destinationVC = segue.destinationViewController as! InformationVC

        destinationVC.eventData = eventData

    default: break

    }

}

segues 是enum：

extension UIStoryboardSegue {

    enum Segue : String
    {

        case EventListToEventOverview
        case EventOverviewToGuestList
        case EventOverviewToAnalytics
        case EventOverviewToMessaging
        case EventOverviewToInformation

    }

}

【问题讨论】：

标签： swift switch-statement refactoring global-variables

【解决方案1】：

给定

struct EventData {}

定义以下协议

protocol Module0 { var eventData: EventData? { get set } }

并让你的视图控制器符合它

class GuestListVC: UIViewController, Module0 { var eventData: EventData? }
class AnalyticsVC: UIViewController, Module0 { var eventData: EventData? }
class MessagingVC: UIViewController, Module0 { var eventData: EventData? }
class InformationVC: UIViewController, Module0 { var eventData: EventData? }

给你的枚举添加一个属性

extension UIStoryboardSegue {

    enum Segue: String {
        case EventListToEventOverview
        case EventOverviewToGuestList
        case EventOverviewToAnalytics
        case EventOverviewToMessaging
        case EventOverviewToInformation

        static var allRawValues: [String] {
            return
                [
                    Segue.EventListToEventOverview.rawValue,
                    Segue.EventOverviewToGuestList.rawValue,
                    Segue.EventOverviewToAnalytics.rawValue,
                    Segue.EventOverviewToMessaging.rawValue,
                    Segue.EventOverviewToInformation.rawValue
            ]
        }
    }

}

现在很简单

class ViewController: UIViewController {
    var eventData = EventData()
    override func prepareForSegue(segue: UIStoryboardSegue, sender: AnyObject?) {
        navigationItem.title = nil

        if var destinationVC = segue.destinationViewController as? protocol<Module0> where UIStoryboardSegue.Segue.allRawValues.contains(segue.identifier!) {
            destinationVC.eventData = eventData
        }
    }
}

【讨论】：

【解决方案2】：

您可以在需要设置 eventData 的 ViewControllers 上使用 EventDataProtocol。任何使用 EventDataProtocol 的视图控制器将始终具有由 Segue 设置的 eventData。例如：

protocol EventDataProtocol {
    var eventData: Int? { get set } //Change to your actual type
}

class EventData: UIViewController, EventDataProtocol {
    var eventData: Int?
}

然后在你的 prepareForSegue 方法中：

override func prepareForSegue(segue: UIStoryboardSegue, sender: AnyObject?) {
    super.prepareForSegue(segue, sender: sender)

    if var eventDataVC = segue.destinationViewController as? EventDataProtocol {
        eventDataVC.eventData = 5
    }
}

【讨论】：

【解决方案3】：

是的，您可以拥有一个具有此 eventData 成员的基类。让我们说-

class BaseViewController : UIViewController{
    var eventData : SomeCustomClass?
}

并从中派生所有需要事件数据的视图控制器。

class InformationVC : BaseViewController{
}

class AnalyticsVC : BaseViewController{
}

class GuestListVC:BaseViewController{
}

现在您的案例可以合并：

switch segue.identifier! {

  case segues.EventOverviewToGuestList,
     segues.EventOverviewToAnalytics,
    segues.EventOverviewToMessaging,
     segues.EventOverviewToInformation:

  let destinationVC = segue.destinationViewController as! BaseViewController
  destinationVC.eventData = eventData

  default: break

}

【讨论】：