diff --git a/MVMCoreUI/Atomic/Protocols/VDSMoleculeViewProtocol.swift b/MVMCoreUI/Atomic/Protocols/VDSMoleculeViewProtocol.swift index 2bb95c43..8dfb9ae3 100644 --- a/MVMCoreUI/Atomic/Protocols/VDSMoleculeViewProtocol.swift +++ b/MVMCoreUI/Atomic/Protocols/VDSMoleculeViewProtocol.swift @@ -14,10 +14,12 @@ import MVMCore ///----------------------------------------------------------------------------- public protocol VDSMoleculeViewProtocol: MoleculeViewProtocol, MVMCoreViewProtocol { associatedtype ViewModel: MoleculeModelProtocol + var shouldUpdateView: Bool { get set } var viewModel: ViewModel! { get set } var delegateObject: MVMCoreUIDelegateObject? { get set } var additionalData: [AnyHashable: Any]? { get set } func viewModelDidUpdate() + func setNeedsUpdate() } extension VDSMoleculeViewProtocol { @@ -32,7 +34,10 @@ extension VDSMoleculeViewProtocol { self.delegateObject = delegateObject self.additionalData = additionalData viewModel = castedModel + shouldUpdateView = false viewModelDidUpdate() + shouldUpdateView = true + setNeedsUpdate() } public func update(viewModel: ViewModel){