基于Swinject为单个ViewController动态注入正确ViewModel的方法

我刚好处理过类似的场景，用Swinject + SwinjectStoryboard实现同一个ViewController动态注入不同ViewModel完全没问题，核心是避开默认按类型注册的冲突，用命名注册或者带参数的工厂方法来区分不同的ViewModel实例。下面给你两种具体的实现方案，你可以根据场景选择：

这种方式适合提前明确注入目标的场景，比如从不同入口跳转时指定对应的ViewModel标识。

1. 注册依赖 #

在你的Swinject容器里给两个ViewModel分别设置唯一命名，避免类型冲突：

let container = Container()

// 注册第一个ViewModel，命名为"first"
container.register(ViewModeling.self, name: "first") { _ in
    firstViewModel()
}

// 注册第二个ViewModel，命名为"second"
container.register(ViewModeling.self, name: "second") { _ in
    secondViewModel()
}

// 为MyViewController注册带参数的构造逻辑，通过传入的名称匹配ViewModel
container.register(MyViewController.self) { resolver, viewModelName in
    let vc = MyViewController()
    // 这里建议加错误处理，避免强制解包崩溃
    guard let viewModel = resolver.resolve(ViewModeling.self, name: viewModelName) else {
        fatalError("无法找到对应名称的ViewModel: \(viewModelName)")
    }
    vc.viewModel = viewModel
    return vc
}

2. 跳转时传入指定名称 #

跳转时不要直接用Storyboard默认的实例化方法，而是通过SwinjectStoryboard带参数的API传入ViewModel名称：

// 场景1：直接从容器解析
let firstVC = SwinjectStoryboard.defaultContainer.resolve(MyViewController.self, argument: "first")!

// 场景2：从指定Storyboard实例化
let storyboard = SwinjectStoryboard(name: "Main", bundle: nil)
let secondVC = storyboard.instantiateViewController(
    withIdentifier: "MyViewController", 
    arguments: "second"
) as! MyViewController

这种方式更灵活，适合根据运行时条件（比如用户权限、传入数据）动态选择ViewModel的场景。

1. 定义类型标识并注册工厂 #

先定义一个枚举（或其他标识类型）来区分ViewModel，再注册带参数的工厂逻辑：

// 定义ViewModel类型枚举
enum ViewModelType {
    case first, second
}

let container = Container()

// 注册带参数的ViewModel工厂，根据传入类型返回对应实例
container.register(ViewModeling.self) { _, type in
    switch type {
    case .first:
        return firstViewModel()
    case .second:
        return secondViewModel()
    }
}

// 为MyViewController注册带类型参数的构造逻辑
container.register(MyViewController.self) { resolver, type in
    let vc = MyViewController()
    guard let viewModel = resolver.resolve(ViewModeling.self, argument: type) else {
        fatalError("无法解析对应类型的ViewModel")
    }
    vc.viewModel = viewModel
    return vc
}

2. 跳转时传入类型参数 #

根据运行时条件选择对应的类型，再传入实例化方法：

// 模拟运行时条件判断
let selectedType: ViewModelType = isVipUser ? .second : .first

// 实例化ViewController并传入类型参数
let storyboard = SwinjectStoryboard(name: "Main", bundle: nil)
let targetVC = storyboard.instantiateViewController(
    withIdentifier: "MyViewController", 
    arguments: selectedType
) as! MyViewController

• 如果是通过Storyboard Segue跳转，可以在prepare(for:sender:)方法中获取目标MyViewController，手动设置ViewModel或者传入标识参数后触发注入。
• 不要过度依赖强制解包，建议在注入逻辑中加入错误处理，比如返回可选值或者抛出异常，提升代码健壮性。
• 如果你使用@Inject属性包装器，静态场景下可以用@Inject(name: "first")指定命名，但动态注入还是推荐上面的带参数注册方案。

内容的提问来源于stack exchange，提问作者goldengil