如何避免在子视图控制器中强制类型转换视图模型实例 - MVVM

Question

我是 MVVM 的新手，对如何避免在子视图控制器中对视图模型实例进行类型转换有点困惑。为了更好地解释它，我添加了下面的代码。

现在我知道每个 ViewController 都应该有一个视图模型，并且应该在 init 或 awakeFromNib 中进行初始化。所以我将协议声明为

public protocol ViewModelCreatorProtocol : class {
    var viewModel : BaseControllerViewModel! {get}
    func createViewModel()
}

我的BaseControllerViewModel 看起来像

public class BaseControllerViewModel {
    //some variables common to all child class
}

现在我有一个BaseViewController，它确认了这个协议，在我的案例中，所有其他 VC 都将从 BaseViewController 扩展

class BaseViewController : UIViewController, ViewModelCreatorProtocol {
    var viewModel: BaseControllerViewModel!

    lazy var disposeBag : DisposeBag = {
        return DisposeBag()
    }()

    override func awakeFromNib() {
        super.awakeFromNib()
        self.createViewModel()
    }
    override func viewDidLoad() {
        super.viewDidLoad()
    }

    func createViewModel() {
        viewModel = BaseControllerViewModel()
    }

    func hookupViewModelObservables() {
        //hookup observables
    }
}

如您所见，BaseViewController 为所有 childViewControllers 提供了一些公共属性，例如 disposeBag，并在 awakeFromNib 中调用 createViewModel，并确保即使在加载 ViewController 之前也始终实例化 viewModel。 （现在让我们假设我所有的 viewModel 都使用默认的 init 进行初始化，并且没有传递给它们的参数）

问题：

现在我创建一个 ChildViewController 让我们说AudioPlayerViewController

class AudioPlayerViewController: BaseViewController {
    //override createViewModel of parent to pass my ViewController specific view model

    override func createViewModel() {
        self.viewModel = AudioPlayerViewModel()
    }
}

显然我的AudioPlayerViewModel 是从BaseControllerViewModel 扩展而来的

class AudioPlayerViewModel : BaseControllerViewModel {
    var totalDuration : Int = 0
    var timeObserver : Any? = nil
    //some more variables and logics
}

一切正常，但现在如果我必须在我的AudioPlayerViewController 中访问totalDuration，我必须以身份访问它

(self.viewModel as! AudioPlayerViewModel).totalDuration

这使得我在 ViewController 中的整个代码库都被(self.viewModel as! AudioPlayerViewModel) 填充，我认为这提高了我的代码可读性。

有没有更好的方法来解决这个问题？

Answer 1

您是否尝试过使用泛型？所以BaseViewController 将有一个必须继承BaseControllerViewModel 的泛型类型。像这样的：

public class BaseControllerViewModel {
    required public init() {
        // needed to do T() in view model creation.
    }
}

class BaseViewController<T: BaseControllerViewModel> : UIViewController, ViewModelCreatorProtocol {
    var viewModel: T! // this will change based on the type you pass in subclass

    lazy var disposeBag : DisposeBag = {
        return DisposeBag()
    }()

    override func awakeFromNib() {
        super.awakeFromNib()
        self.createViewModel()
    }
    override func viewDidLoad() {
        super.viewDidLoad()
    }

    func createViewModel() {
        viewModel = T() // instantiate the view model
    }

    func hookupViewModelObservables() {
        //hookup observables
    }
}

例如，具体的类将是：

class ExampleVcViewModel: BaseControllerViewModel {
    var subclassVar: Int?
}

class EampleVc: BaseViewController<ExampleVcViewModel> {
    override func viewDidLoad() {
        super.viewDidLoad()
        print(viewModel.subclassVar) // no need to cast
    }
}

Answer 2

从BaseViewController中移除viewModel，每个视图控制器都应该有自己的viewModel，没有别的办法

Answer 3

对于这种情况，我个人使用另一个返回适当视图模型的属性。所以，它可能是：

var audioPlayerViewModel: AudioPlayerViewModel? {
    return viewModel as? AudioPlayerViewModel
}

func createViewModel() {
    viewModel = AudioPlayerViewModel()
}

或者您可以删除属性中的可选项，因为您知道它将完全是AudioPlayerViewModel。

var audioPlayerViewModel: AudioPlayerViewModel {
    return viewModel as! AudioPlayerViewModel
}