用于协议的 Swift 扩展存储

Question

我制定了一个协议：

protocol SomeProtocol {
    func getData() -> String
}

我做了一个符合它的结构：

struct SomeStruct: SomeProtocol {
    func getData() -> String {
        return "Hello"
    }
}

现在我希望每个UIViewController 都有一个名为source 的属性，所以我可以做类似的事情……

class MyViewController : UIViewController {
    override func viewDidLoad() {
        self.title = source.getData()
    }
}

为此，我创建了一个协议来定义属性：

protocol SomeProtocolInjectable {
    var source: SomeProtocol! { get set }
}

现在我只需要使用这个属性来扩展视图控制器：

extension UIViewController: SomeProtocolInjectable {
    // ???
}

我怎样才能将可与协议类型一起使用的存储属性组合在一起？

什么没用：

• var source: SomeProtocol! 显然不起作用，因为扩展没有存储属性
• 我不能use Objective-C associated objects，因为协议不是对象
• 我不能wrap it in a class（这对其他值类型有效，但对协议无效）

还有其他建议吗？

Answer 1

任何协议对象都可以转换为类型擦除的类。构建一个 AnySomeProtocol 并存储它。

private var sourceKey: UInt8 = 0

final class AnySomeProtocol: SomeProtocol {
    func getData() -> String { return _getData() }
    init(_ someProtocol: SomeProtocol) { _getData = someProtocol.getData }
    private let _getData: () -> String
}

extension UIViewController: SomeProtocolInjectable {
    var source: SomeProtocol! {
        get {
            return objc_getAssociatedObject(self, &sourceKey) as? SomeProtocol
        }
        set(newValue) {
            objc_setAssociatedObject(self, &sourceKey, AnySomeProtocol(newValue), .OBJC_ASSOCIATION_RETAIN)
        }
    }
}

class MyViewController : UIViewController {
    override func viewDidLoad() {
        self.title = source.getData()
    }
}

调用者只能使用它来访问协议方法。您无法使用 as 将其强制恢复为原始类型，但无论如何您都应该避免这种情况。

作为旁注，我真的建议让source 返回SomeProtocol? 而不是SomeProtocol!。这里没有任何东西可以保证source 将被设置。在viewDidLoad 之前，您甚至都不会设置它。

Answer 2

您可以使用静态和视图控制器hash：

struct SomeProtocol {/*....*/}

struct DataProxy {
    var data: [Int: SomeProtocol]
}



protocol SomeProtocolInjectable {
    var source: SomeProtocol! { get set }
}

extension UIViewController: SomeProtocolInjectable {

    static var dataProxy = DataProxy(data: [:])

    var source: SomeProtocol! {
        get{
            return UIViewController.dataProxy.data[self.hashValue]
        }
        set{
            UIViewController.dataProxy.data[self.hashValue] = newValue
        }
    }

}

Answer 3

如何为getData() 添加一个默认实现，为协议提供一个虚拟结构实现，并将其用作source 变量的默认值：

protocol SomeProtocol {
    func getData() -> String
}

extension SomeProtocol {
    func getData() -> String {
        return "Hello"
    }
}

protocol SomeProtocolInjectable {
    var source: SomeProtocol { get set }
}

struct DummyProtocolImplementation: SomeProtocol {

}

class MyViewController : UIViewController {
    var _source: SomeProtocol = DummyProtocolImplementation()

    override func viewDidLoad() {
        self.title = source.getData()
    }
}

extension MyViewController: SomeProtocolInjectable {
    var source: SomeProtocol { get { return _source } set { _source = newValue } }
}

我冒昧地扩展了MyViewController 而不是UIViewController，因为后者根本不知道协议。