斯威夫特擴展存儲的協議

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() 
    } 
} 

爲了達到這個目的，我創建了一個協議到d efine屬性：

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：

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，而不是無論如何，後者並不知道該協議。