ControllerAware.swift

//
//  ControllerAware Protocol
//  Shows how to have view controller instances which implement 
//  a specific protocol have it's method called on viewDidLoad.

import UIKit

protocol ControllerAware {
    func onViewDidLoad()
}

class BaseViewController: UIViewController {
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        if let controllerAwareSelf = self as? ControllerAware {
            controllerAwareSelf.onViewDidLoad()
        }
    }
}

class MyCustomVC: BaseViewController, ControllerAware {
    
    func onViewDidLoad() {
        
    }
    
}